def __init__(self, *args, **kwargs): ThriftClient.__init__(self, *args, **kwargs) self.trace_context = thread_storage.getTraceContext() self.client_ip = socket.gethostbyname(socket.gethostname()) self.protocol = TMiloProtocol(self.transport) self.client_ = self.service.Client(self.protocol) self.client_.transport = self.transport
def handle(self, client): itrans = self.inputTransportFactory.getTransport(client) otrans = self.outputTransportFactory.getTransport(client) iprot = self.inputProtocolFactory.getProtocol(itrans) oprot = self.outputProtocolFactory.getProtocol(otrans) try: while True: start_time = time.time() * 1000000 self.processor.process(iprot, oprot) end_time = time.time() * 1000000 tctx = thread_storage.getTraceContext() if tctx.isEnabled(): name = span_util.createSpanName(self.service_name, \ tctx.getRPCMethod()) span = span_util.createSpan(name, tctx.getTraceID(), \ tctx.getSpanID(), tctx.getSpanParentID()) span.server_host = self.server_ip span.events = [span_util.createEvent(start_time, EventType.SERVER_RECV), \ span_util.createEvent(end_time, EventType.SERVER_SEND)] if self.debug: logging.debug(span) tctx.addSpan(span) self.logSpans(tctx) except TTransport.TTransportException, tx: pass
def __init__(self, trans): TBinaryProtocol.__init__(self, trans) self.strictRead = False self.strictWrite = True self.trace_context = getTraceContext()