thrift hive 笔记

• Z时代
• 2024-01-10
• 分类：综合

import org.apache.hadoop.hive.common.auth.HiveAuthUtils;
import org.apache.hive.service.rpc.thrift.TCLIService;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.server.TThreadPoolServer;
import org.apache.thrift.transport.*;
import java.util.concurrent.TimeUnit;
public class MyHiveServer {
    public static void main(String[] args) throws Exception{
//         PlainSaslHelper.getPlainTransportFactory
//        TSaslServerTransport.Factory saslFactory = new TSaslServerTransport.Factory();
//
//        saslFactory.addServerDefinition("PLAIN", "CUSTOM", null, new HashMap<String, String>(),new PlainSaslHelper.PlainServerCallbackHandler("CUSTOM"))
//    源码    HiveAuthFactory 第160行
        try {
            System.out.println("服务端开启");
//            transportFactory = new TTransportFactory();
            TCLIService.Processor<ServiceFaceImpl> processor = new TCLIService.Processor<>(new ServiceFaceImpl());
            TServerSocket serverSocket = HiveAuthUtils.getServerSocket("127.0.0.1", 10000);
            TThreadPoolServer.Args sargs = new TThreadPoolServer.Args(serverSocket).processor(processor)
                    .transportFactory(new TTransportFactory()).protocolFactory(new TBinaryProtocol.Factory())
                    .inputProtocolFactory(new TBinaryProtocol.Factory(true, true, 100*1024*1024, 100*1024*1024))
                    .requestTimeout(100000).requestTimeoutUnit(TimeUnit.SECONDS);
            TThreadPoolServer server = new TThreadPoolServer(sargs);
            // TCP Server
            server.serve();
        } catch (TTransportException e) {
            e.printStackTrace();
        }
        System.out.println("服务端已启动  ：10000");
    }
}

public class ServiceFaceImpl implements TCLIService.Iface {
    private static final TStatus OK_STATUS = new TStatus(TStatusCode.SUCCESS_STATUS);
    @Override
    public TOpenSessionResp OpenSession(TOpenSessionReq req) throws TException {
        TOpenSessionResp resp = new TOpenSessionResp();
        TSessionHandle tSessionHandle = new TSessionHandle(new HandleIdentifier().toTHandleIdentifier());
        resp.setSessionHandle(tSessionHandle);
        System.out.println("OpenSession");
        HashMap<String, String> configurationMap = new HashMap<>();
        configurationMap.put("","");
        resp.setConfiguration(configurationMap);
//        resp.setServerProtocolVersion(TProtocolVersion.HIVE_CLI_SERVICE_PROTOCOL_V9);
        resp.setStatus(OK_STATUS);
        return resp;
    }
    @Override
    public TCloseSessionResp CloseSession(TCloseSessionReq req) throws TException {
        System.out.println("CloseSession");
        return null;
    }
    @Override
    public TGetInfoResp GetInfo(TGetInfoReq req) throws TException {
        System.out.println("GetInfo");
        return null;
    }
    @Override
    public TExecuteStatementResp ExecuteStatement(TExecuteStatementReq req) throws TException {
        String statement = req.getStatement();
        System.out.println(statement);
        System.out.println("ExecuteStatement");
        return null;
    }
    @Override
    public TGetTypeInfoResp GetTypeInfo(TGetTypeInfoReq req) throws TException {
        return null;
    }
    @Override
    public TGetCatalogsResp GetCatalogs(TGetCatalogsReq req) throws TException {
        return null;
    }
    @Override
    public TGetSchemasResp GetSchemas(TGetSchemasReq req) throws TException {
        return null;
    }
    @Override
    public TGetTablesResp GetTables(TGetTablesReq req) throws TException {
        return null;
    }
    @Override
    public TGetTableTypesResp GetTableTypes(TGetTableTypesReq req) throws TException {
        return null;
    }
    @Override
    public TGetColumnsResp GetColumns(TGetColumnsReq req) throws TException {
        return null;
    }
    @Override
    public TGetFunctionsResp GetFunctions(TGetFunctionsReq req) throws TException {
        return null;
    }
    @Override
    public TGetPrimaryKeysResp GetPrimaryKeys(TGetPrimaryKeysReq req) throws TException {
        return null;
    }
    @Override
    public TGetCrossReferenceResp GetCrossReference(TGetCrossReferenceReq req) throws TException {
        return null;
    }
    @Override
    public TGetOperationStatusResp GetOperationStatus(TGetOperationStatusReq req) throws TException {
        return null;
    }
    @Override
    public TCancelOperationResp CancelOperation(TCancelOperationReq req) throws TException {
        return null;
    }
    @Override
    public TCloseOperationResp CloseOperation(TCloseOperationReq req) throws TException {
        return null;
    }
    @Override
    public TGetResultSetMetadataResp GetResultSetMetadata(TGetResultSetMetadataReq req) throws TException {
        return null;
    }
    @Override
    public TFetchResultsResp FetchResults(TFetchResultsReq req) throws TException {
        return null;
    }
    @Override
    public TGetDelegationTokenResp GetDelegationToken(TGetDelegationTokenReq req) throws TException {
        return null;
    }
    @Override
    public TCancelDelegationTokenResp CancelDelegationToken(TCancelDelegationTokenReq req) throws TException {
        return null;
    }
    @Override
    public TRenewDelegationTokenResp RenewDelegationToken(TRenewDelegationTokenReq req) throws TException {
        return null;
    }
}

以上是 thrift hive 笔记 的全部内容， 来源链接： utcz.com/z/267426.html