1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
from __future__ import print_function
from opentracing.ext import tags
from ..utils import get_logger
logger = get_logger(__name__)
class RequestHandler(object):
def __init__(self, tracer, context=None, ignore_active_span=True):
self.tracer = tracer
self.context = context
self.ignore_active_span = ignore_active_span
def before_request(self, request, request_context):
logger.info('Before request %s' % request)
# If we should ignore the active Span, use any passed SpanContext
# as the parent. Else, use the active one.
if self.ignore_active_span:
span = self.tracer.start_span('send',
child_of=self.context,
ignore_active_span=True)
else:
span = self.tracer.start_span('send')
span.set_tag(tags.SPAN_KIND, tags.SPAN_KIND_RPC_CLIENT)
request_context['span'] = span
def after_request(self, request, request_context):
logger.info('After request %s' % request)
span = request_context.get('span')
if span is not None:
span.finish()
|
