654a271b82
Zipkin is a trend distributed tracing framewrok developed at Twitter. Such tracing is useful for both developers and operatos to understand the behavior of complex distributed systems and find performance bottlenecks. This patch provides a WSGI application using eventlet with tracing facility that complies with Zipkin. Signed-off-by: Yuichi Bando <bando.yuichi@lab.ntt.co.jp> Original commit modified for PEP-8 fixes. https://github.com/eventlet/eventlet/pull/218
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
import base64
|
|
import warnings
|
|
|
|
from scribe import scribe
|
|
from thrift.transport import TTransport, TSocket
|
|
from thrift.protocol import TBinaryProtocol
|
|
|
|
from eventlet import GreenPile
|
|
|
|
|
|
CATEGORY = 'zipkin'
|
|
|
|
|
|
class ZipkinClient(object):
|
|
|
|
def __init__(self, host='127.0.0.1', port=9410):
|
|
"""
|
|
:param host: zipkin collector IP addoress (default '127.0.0.1')
|
|
:param port: zipkin collector port (default 9410)
|
|
"""
|
|
self.host = host
|
|
self.port = port
|
|
self.pile = GreenPile(1)
|
|
self._connect()
|
|
|
|
def _connect(self):
|
|
socket = TSocket.TSocket(self.host, self.port)
|
|
self.transport = TTransport.TFramedTransport(socket)
|
|
protocol = TBinaryProtocol.TBinaryProtocol(self.transport,
|
|
False, False)
|
|
self.scribe_client = scribe.Client(protocol)
|
|
try:
|
|
self.transport.open()
|
|
except TTransport.TTransportException as e:
|
|
warnings.warn(e.message)
|
|
|
|
def _build_message(self, thrift_obj):
|
|
trans = TTransport.TMemoryBuffer()
|
|
protocol = TBinaryProtocol.TBinaryProtocolAccelerated(trans=trans)
|
|
thrift_obj.write(protocol)
|
|
return base64.b64encode(trans.getvalue())
|
|
|
|
def send_to_collector(self, span):
|
|
self.pile.spawn(self._send, span)
|
|
|
|
def _send(self, span):
|
|
log_entry = scribe.LogEntry(CATEGORY, self._build_message(span))
|
|
try:
|
|
self.scribe_client.Log([log_entry])
|
|
except Exception as e:
|
|
msg = 'ZipkinClient send error %s' % str(e)
|
|
warnings.warn(msg)
|
|
self._connect()
|
|
|
|
def close(self):
|
|
self.transport.close()
|