java-classic-code/hadoop-auth/HiveConnectionFactory.java

package com.primeton.dsp.datarelease.data.bdata;

import com.primeton.dsp.datarelease.server.model.DspHiveResource;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;

/**
 *
 * 根据参数不同，生成不动的测试连接实例
 *
 * <pre>
 *
 * Created by zhaopx.
 * User: zhaopx
 * Date: 2020/4/21
 * Time: 18:02
 *
 * </pre>
 *
 * @author zhaopx
 */
@Slf4j
public class HiveConnectionFactory {


    /**
     * 生成不同的测试实例
     * @return
     */
    public static HiveConnectionService getHiveInstance(DspHiveResource hiveResource) {
	    String type = hiveResource.getCollectionType();
        String authType = hiveResource.getAuthType();
        String hiveDbUser = hiveResource.getHiveDbUser();
        if(StringUtils.isBlank(hiveDbUser) || "noauth".equalsIgnoreCase(authType)) {
            // 默认无认证的方式测试
            return new SimpleHiveConnectionServiceImpl(hiveResource);
        } else if ("kerberos".equalsIgnoreCase(authType) && StringUtils.isNotBlank(hiveDbUser)) {
            return new Krb5HiveConnectionServiceImpl(hiveResource);
        } else if ("fi".equalsIgnoreCase(authType) && StringUtils.isNotBlank(hiveDbUser)) {
            return new FIHiveConnectionServiceImpl(hiveResource);
        }

        // 未知的方式，只能通过简单的来处理一下了。
        log.info("未知的Hive认证方式：" + authType);
        return new SimpleHiveConnectionServiceImpl(hiveResource);
    }
}