def handle(self, node):
try:
if self.framework == 'flask' or self.framework == 'bottle':
console.info("handle flask")
self.handle_flask(node)
elif self.framework == 'oslo_messaging':
console.info("handle oslo_messaging")
self.handle_rabbitmq(node)
elif self.framework == 'pika':
console.info("handle pika")
self.handle_pika(node)
else:
pass
except Exception as e:
console.exception("handle error")
def wrapper(wrapped, instance, args, kwargs):
console.info("args===>")
console.info(args)
console.info("args<===")
tracker = current_tracker()
console.info("wsgi_wrapper_inline>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>")
console.info("current_tracker:1: %s" % tracker)
if tracker:
console.info("current_tracker:2: None")
console.info("<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<")
return wrapped(*args, **kwargs)
environ, start_response = parse_wsgi_protocol(target, *args, **kwargs)
disable_agent = environ.get(AGENT_REQUEST_SWITCH, None)
if disable_agent:
console.debug("Current trace is disabled with http request environment. %s", environ)
return wrapped(*args, **kwargs)
tracker = Tracer(proxy_instance(), environ, framework)
tracker.generate_trace_id()
tracker.start_work()
def _start_response(status, response_headers, *args):
# deal the response header/data
process_header(tracker, response_headers)
tracker.deal_response(status, response_headers, *args)
_write = start_response(status, response_headers, *args)
return _write
result = []
try:
tracker.set_tracker_name(callable_name(wrapped), priority=1)
application = function_trace_wrapper(wrapped)
console.info("current_tracker:2: %s" % tracker)
with FunctionTracker(tracker, name='Application', group='Python.WSGI'):
result = TingYunWSGIBrowserRumMiddleware(tracker, application, _start_response, environ)() # sunyan: 1
# result = application(environ, start_response)
console.info("wsgi_wrapper_inline<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<")
except:
console.exception("wsgi_entrance")
tracker.finish_work(*sys.exc_info())
raise
return WSGIApplicationResponse(tracker, result)
def wrapper(wrapped, instance, args, kwargs):
incoming_message = args[0][0]
msg = incoming_message.message
console.info("msg:")
console.info(msg)
ctxt = incoming_message.ctxt
console.info("ctxt:")
console.info(ctxt)
oslo_messaging_trace_id = ctxt['oslo_messaging_trace_id'] if 'oslo_messaging_trace_id' in ctxt else None
tracker = current_tracker()
console.info("oslo_messaging_wrapper_inline>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>")
console.info("current_tracker:1: %s" % tracker)
if tracker:
console.info("current_tracker:2: None")
console.info("<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<")
return wrapped(*args, **kwargs)
environ = {
'k1': 'v1'
}
tracker = Tracer(proxy_instance(), environ, framework)
if oslo_messaging_trace_id is not None:
tracker._trace_id = oslo_messaging_trace_id
tracker.generate_trace_id()
tracker.start_work()
result = None
try:
tracker.set_tracker_name(callable_name(wrapped), priority=1)
with FunctionTracker(tracker, name='Application', group='Python.oslo_messaging', params=msg['args']):
result = wrapped(*args, **kwargs)
console.info("oslo_messaging_wrapper_inline<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<")
except:
console.exception("oslo_messaging_entrance")
tracker.finish_work(*sys.exc_info())
raise
else:
tracker.finish_work(None, None, None)
return result
def default_handler(self, encoded_span):
if self.app_name is None:
console.info("APP_NAME is none, error!")
return
try:
body = str.encode('\x0c\x00\x00\x00\x01') + encoded_span
console.info("url:{0}".format(self.zipkin_dsn))
rec = requests.post(
self.zipkin_dsn,
data=body,
headers={'Content-Type': 'application/x-thrift'},
timeout=1,
)
console.info("rec:{0}".format(rec))
return rec
except Exception as e:
console.exception("flask_zipkin")
if self._transport_exception_handler:
self._transport_exception_handler(e)
else:
self.default_exception_handler(e)
def handle_rabbitmq(self, node):
try:
console.info("environ>>>>>>")
print(self.environ)
console.info("environ<<<<<<")
if node.type == 'WebAction':
for child in node.children:
if 'oslo_messaging' in child.group:
transport_handler = self._transport_handler or self.default_handler
trace_id = node.trace_id
span_name = json.dumps(child.params)
start_time = child.start_time
end_time = child.end_time
parent_span_id = None
is_sampled = True
flags = '0'
zipkin_attrs = self.generate_zipkin_attrs(
trace_id=trace_id,
span_id=self._gen_random_id(),
parent_span_id=parent_span_id,
flags=flags,
is_sampled=is_sampled)
span2 = self.generate_span_2(
service_name=self.app_name,
span_name=span_name,
zipkin_attrs=zipkin_attrs,
transport_handler=transport_handler)
span2.start(start_time)
span2.stop(start_timestamp=start_time,
end_timestamp=end_time)
except Exception as e:
console.exception("handle_rabbitmq error")
def wrapper(wrapped, instance, args, kwargs):
console.info('args:')
console.info(args)
console.info('kwargs:')
console.info(kwargs)
method = args[0]
header = args[1]
body = None
pika_trace_id = None
try:
wrapped_message = json.loads(args[2])
if 'pika_trace_id' in wrapped_message:
pika_trace_id = wrapped_message['pika_trace_id']
body = wrapped_message['body']
else:
body = args[2]
except Exception:
body = args[2]
args = (method, header, body)
params = {
#'method': str(method),
#'header': str(header),
'body': body
}
tracker = current_tracker()
console.info(
"pika_wrapper_inline>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>")
console.info("current_tracker:1: %s" % tracker)
if tracker:
console.info("current_tracker:2: None")
console.info("<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<")
return wrapped(*args, **kwargs)
environ = {'k1': 'v1'}
tracker = Tracer(proxy_instance(), environ, framework)
if pika_trace_id is not None:
tracker._trace_id = pika_trace_id
tracker.generate_trace_id()
tracker.start_work()
result = None
try:
tracker.set_tracker_name(callable_name(wrapped), priority=1)
with FunctionTracker(tracker,
name='Application',
group='Python.pika',
params=params):
result = wrapped(*args, **kwargs)
console.info(
"pika_wrapper_inline<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<")
except:
console.exception("pika_wrapper_inline")
tracker.finish_work(*sys.exc_info())
raise
else:
tracker.finish_work(None, None, None)
return result
def handle_flask(self, node):
try:
console.info("environ>>>>>>")
print(self.environ)
console.info("environ<<<<<<")
"""
headers = self.environ['werkzeug.request'].headers
url = self.environ['werkzeug.request'].url
trace_id = headers.get('X-B3-TraceId') or self._gen_random_id()
parent_span_id = headers.get('X-B3-Parentspanid')
is_sampled = str(headers.get('X-B3-Sampled') or '0') == '1'
flags = headers.get('X-B3-Flags')
"""
trace_id = node.trace_id
#trace_id = self.environ.get('HTTP_X_B3_TRACEID') or self._gen_random_id()
start_time = node.start_time
end_time = node.end_time
url = "http://{0}{1}".format(self.environ.get('HTTP_HOST'),
self.environ.get('PATH_INFO'))
parent_span_id = self.environ.get('HTTP_X_B3_PARENTSPANID')
#is_sampled = str(self.environ.get('HTTP_X_B3_SAMPLED') or '0') == '1'
is_sampled = True
#flags = self.environ.get('HTTP_X_B3_FLAGS')
flags = '0'
# parent span
transport_handler = self._transport_handler or self.default_handler
zipkin_attrs = self.generate_zipkin_attrs(
trace_id=trace_id,
span_id=self._gen_random_id(),
parent_span_id=parent_span_id,
flags=flags,
is_sampled=is_sampled)
span2 = self.generate_span_2(service_name=self.app_name,
span_name=url,
zipkin_attrs=zipkin_attrs,
transport_handler=transport_handler)
span2.start(start_time)
if node.type == 'WebAction':
for child in node.children:
# flask
if child.name == 'Response':
for child2 in child.children:
# flask or bottle
if 'flask' in child2.name or 'bottle' in child2.name:
for child3 in child2.children:
if child3.group == 'Function':
for child4 in child3.children:
# mysql
if type(
child4
) == DatabaseNode and child4.sql != 'COMMIT':
sql = child4.sql
span_name = ''
execute_params = child4.execute_params
if sql.strip().upper(
).startswith("SELECT"):
sql = sql % execute_params[
0]
span_name = "SELECT"
elif sql.strip().upper(
).startswith("INSERT"):
sql = sql % execute_params[
0]
span_name = "INSERT"
elif sql.strip().upper(
).startswith("DELETE"):
sql = sql % execute_params[
0]
span_name = "DELETE"
elif sql.strip().upper(
).startswith("UPDATE"):
sql = sql % execute_params[
0][0]
span_name = "UPDATE"
else:
pass
# child span
span3 = self.generate_span_3(
service_name=self.app_name,
span_name=span_name,
binary_annotations={
'jdbc.query': sql
})
span3.start(child4.start_time)
span3.stop(
start_timestamp=child4.
start_time,
end_timestamp=child4.
end_time)
# requests
elif type(child4) == ExternalNode:
# TODO ?
pass
# rabbitmq, FIXME sunyan: right now, we don't need this branch, but may be useful in the future!
#elif type(child4) == FunctionNode and 'oslo_messaging' in child4.name:
# span3 = self.generate_span_3(service_name=self.app_name,
# span_name=json.dumps(child4.params),
# binary_annotations=child4.params)
# span3.start(child4.start_time)
# span3.stop(start_timestamp=child4.start_time,
# end_timestamp=child4.end_time)
else:
pass
else:
pass
# not flask and bottle
else:
pass
# others
else:
pass
span2.stop(start_timestamp=start_time, end_timestamp=end_time)
except Exception as e:
console.exception("handle_flask error")