def connect(self):
self.socket = TSocket.TSocket(self.host, self.port)
if self.timeout:
self.socket.setTimeout(self.timeout)
if self.use_plain_sasl:
self.transport = SaslClientTransport(self.host, self.socket, service=STORM_SASL_SERVICE, mechanism='PLAIN')
else:
self.transport = TTransport.TFramedTransport(self.socket)
self.protocol = TBinaryProtocol.TBinaryProtocol(self.transport)
self.transport.open()
self.client = Client(self.protocol)
