示例#1
0
 def __init__(self, host, port, username="******"):
     self.host = host
     self.port = port
     self.username = username
     self.sessionId = None
     self.channel = TajoRpcChannel(self.host, self.port)
     self.service = RpcService(tajo_pb2.TajoMasterClientProtocolService_Stub,
                               self.channel)
示例#2
0
class TajoSessionConnection():
    def __init__(self, host, port, username="******"):
        self.host = host
        self.port = port
        self.username = username
        self.sessionId = None
        self.channel = TajoRpcChannel(self.host, self.port)
        self.service = RpcService(tajo_pb2.TajoMasterClientProtocolService_Stub,
                                  self.channel)

    def createNewSessionId(self):
        request = ClientProtos_pb2.CreateSessionRequest()
        request.username = self.username
        response = self.service.createSession(request)
        self.sessionId = response.sessionId.id
        return self.sessionId

    def checkSessionAndGet(self):
        if self.sessionId is None:
            self.sessionId = self.createNewSessionId()

        return self.sessionId