def from_header(self, binary):
"""Generate a SpanContext object using the trace context header.
The value of enabled parsed from header is int. Need to convert to
bool.
:type binary: bytes
:param binary: Trace context header which was extracted from the
request headers.
:rtype: :class:`~opencensus.trace.span_context.SpanContext`
:returns: SpanContext generated from the trace context header.
"""
# If cannot parse, return a new SpanContext and ignore the context
# from binary.
try:
data = Header._make(struct.unpack(BINARY_FORMAT, binary))
except struct.error:
logging.warn(
'Cannot parse the incoming binary data {}, '
'wrong format. Total bytes length should be {}.'.format(
binary, FORMAT_LENGTH))
return SpanContext(from_header=False)
# data.trace_id is in bytes with length 16, hexlify it to hex bytes
# with length 32, then decode it to hex string using utf-8.
trace_id = str(binascii.hexlify(data.trace_id).decode(UTF8))
span_id = data.span_id
trace_options = TraceOptions(data.trace_option)
span_context = SpanContext(trace_id=trace_id,
span_id=span_id,
trace_options=trace_options,
from_header=True)
return span_context
def from_headers(self, headers):
"""Generate a SpanContext object using the W3C Distributed Tracing headers.
:type headers: dict
:param headers: HTTP request headers.
:rtype: :class:`~opencensus.trace.span_context.SpanContext`
:returns: SpanContext generated from the trace context header.
"""
if headers is None:
return SpanContext()
if _TRACE_PARENT_HEADER_NAME not in headers:
return SpanContext()
return self.from_header(headers[_TRACE_PARENT_HEADER_NAME])
def from_header(self, header):
"""Generate a SpanContext object using the trace context header.
:type header: str
:param header: Trace context header which was extracted from the HTTP
request headers.
:rtype: :class:`~opencensus.trace.span_context.SpanContext`
:returns: SpanContext generated from the trace context header.
"""
if header is None:
return SpanContext()
try:
match = re.search(_TRACE_CONTEXT_HEADER_RE, header)
except TypeError:
logging.warning(
'Header should be str, got {}. Cannot parse the header.'
.format(header.__class__.__name__))
raise
if match:
version = match.group(1)
if version == '00':
trace_id = match.group(3)
span_id = match.group(5)
trace_options = match.group(7)
if trace_options is None:
trace_options = 1
# Need to convert span_id from hex string to int
span_context = SpanContext(
trace_id=trace_id,
span_id=span_id,
trace_options=TraceOptions(trace_options),
from_header=True)
return span_context
else:
logging.warning(
'Header format version {} is not supported, generate a new'
'context instead.'.format(version))
else:
logging.warning(
'Cannot parse the header {}, generate a new context instead.'
.format(header))
return SpanContext()
def from_headers(self, headers):
"""Generate a SpanContext object using the trace context header.
:type headers: dict
:param headers: HTTP request headers.
:rtype: :class:`~opencensus.trace.span_context.SpanContext`
:returns: SpanContext generated from the trace context header.
"""
if headers is None:
return SpanContext()
header = headers.get(_TRACE_CONTEXT_HEADER_NAME)
if header is None:
return SpanContext()
return self.from_header(header)
def from_carrier(self, carrier):
"""Generate a SpanContext object using the information in the carrier.
:type carrier: dict
:param carrier: The carrier which has the trace_id, span_id, options
information for creating a SpanContext.
:rtype: :class:`~opencensus.trace.span_context.SpanContext`
:returns: SpanContext generated from the carrier.
"""
trace_id = None
span_id = None
trace_options = None
for key in carrier:
key = key.lower()
if key == _TRACE_ID_KEY:
trace_id = carrier[key]
if key == _SPAN_ID_KEY:
span_id = carrier[key]
if key == _TRACE_OPTIONS_KEY:
trace_options = bool(carrier[key])
if trace_options is None:
trace_options = DEFAULT_TRACE_OPTIONS
return SpanContext(trace_id=trace_id,
span_id=span_id,
trace_options=TraceOptions(trace_options),
from_header=True)
def from_carrier(self, carrier):
"""Generate a SpanContext object using the information in the carrier.
:type carrier: dict
:param carrier: The carrier which has the trace_id, span_id, options
information for creating a SpanContext.
:rtype: :class:`~opencensus.trace.span_context.SpanContext`
:returns: SpanContext generated from the carrier.
"""
trace_id = None
span_id = None
enabled = 1
for key in carrier:
key = key.lower()
if key == _TRACE_ID_KEY:
trace_id = carrier[key]
if key == _SPAN_ID_KEY:
span_id = carrier[key]
if key == _ENABLED_TRACE_KEY:
enabled = bool(carrier[key])
return SpanContext(
trace_id=trace_id,
span_id=span_id,
enabled=enabled,
from_header=True)
def from_headers(self, headers=None):
"""Generate a SpanContext object using the trace context header.
:type headers: dict
:param headers: HTTP request headers.
:rtype: :class:`~opencensus.trace.span_context.SpanContext`
:returns: SpanContext generated from the trace context header.
"""
if headers is None:
return SpanContext()
for name in headers:
if name.lower() == _UBER_HEADER_NAME.lower():
return self.from_header(headers[name])
return SpanContext()
def from_headers(self, headers):
"""Generate a SpanContext object using the W3C Distributed Tracing headers.
:type headers: dict
:param headers: HTTP request headers.
:rtype: :class:`~opencensus.trace.span_context.SpanContext`
:returns: SpanContext generated from the trace context header.
"""
if headers is None:
return SpanContext()
header = headers.get(_TRACEPARENT_HEADER_NAME)
if header is None:
return SpanContext()
match = re.search(_TRACEPARENT_HEADER_FORMAT_RE, header)
if not match:
return SpanContext()
version = match.group(1)
trace_id = match.group(2)
span_id = match.group(3)
trace_options = match.group(4)
if trace_id == '0' * 32 or span_id == '0' * 16:
return SpanContext()
if version == '00':
if match.group(5):
return SpanContext()
if version == 'ff':
return SpanContext()
span_context = SpanContext(
trace_id=trace_id,
span_id=span_id,
trace_options=TraceOptions(trace_options),
from_header=True)
header = headers.get(_TRACESTATE_HEADER_NAME)
if header is None:
return span_context
try:
tracestate = TracestateStringFormatter().from_string(header)
if tracestate.is_valid():
span_context.tracestate = \
TracestateStringFormatter().from_string(header)
except ValueError:
pass
return span_context
def __init__(self, exporter=None, span_context=None):
if span_context is None:
span_context = SpanContext()
self.tracer = tracer.Tracer(
exporter=exporter,
span_context=span_context,
)
def from_header(self, header):
"""Generate a SpanContext object using the trace context header.
The value of enabled parsed from header is int. Need to convert to
bool.
:type header: str
:param header: Trace context header which was extracted from the HTTP
request headers.
:rtype: :class:`~opencensus.trace.span_context.SpanContext`
:returns: SpanContext generated from the trace context header.
"""
if header is None:
return SpanContext()
try:
match = re.search(_TRACE_CONTEXT_HEADER_RE, header)
except TypeError:
logging.warning(
'Header should be str, got %s. Cannot parse the header.',
header.__class__.__name__)
raise
if match:
trace_id = match.group(1)
span_id = match.group(3)
trace_options = match.group(5)
if trace_options is None:
trace_options = 1
if span_id:
span_id = '{:016x}'.format(int(span_id))
span_context = SpanContext(
trace_id=trace_id,
span_id=span_id,
trace_options=TraceOptions(trace_options),
from_header=True)
return span_context
else:
logging.warning(
'Cannot parse the header %s, generate a new context instead.',
header)
return SpanContext()
def __init__(self, span_context=None):
if span_context is None:
span_context = SpanContext()
self.span_context = span_context
self.trace_id = span_context.trace_id
self.cur_trace = self.trace()
self._span_stack = []
self.root_span_id = span_context.span_id
def stop_trace():
"""Stop the current trace."""
if not execution_context:
return
execution_context.set_current_span(None)
tracer = execution_context.get_opencensus_tracer()
tracer.tracer = noop_tracer.NoopTracer()
tracer.span_context = SpanContext(
trace_options=trace_options.TraceOptions(0))
tracer.sampler = always_off.AlwaysOffSampler()
def from_header(self, header=None):
"""Generate a SpanContext object using the trace context header.
The value of enabled parsed from header is int. Need to convert to
bool.
:type header: str
:param header: Trace context header which was extracted from the HTTP
request headers.
:rtype: :class:`~opencensus.trace.span_context.SpanContext`
:returns: SpanContext generated from the trace context header.
"""
if header is None:
return SpanContext()
try:
match = re.match(_UBER_HEADER_RE, header)
except TypeError:
logging.warning(
'Header should be str, got {}. Cannot parse the header.'.
format(header.__class__.__name__))
raise
if match is not None:
trace_id = match.group(1)
span_id = match.group(2)
trace_options = match.group(4)
span_context = SpanContext(trace_id=trace_id.zfill(32),
span_id=span_id.zfill(16),
trace_options=TraceOptions(
str(int('0x' + trace_options, 16))),
from_header=True)
return span_context
else:
logging.warning(
'Cannot parse the header {}, generate a new context instead.'.
format(header))
return SpanContext()
def from_carrier(cls, carrier):
if cls.trace_id_key not in carrier or \
cls.span_id_key not in carrier:
return None
trace_id = carrier[cls.trace_id_key]
span_id = carrier[cls.span_id_key]
return SpanContext(
trace_id=trace_id,
span_id=span_id,
from_header=True,
)
def __init__(self, exporter=None, span_context=None):
if exporter is None:
exporter = print_exporter.PrintExporter()
if span_context is None:
span_context = SpanContext()
self.exporter = exporter
self.span_context = span_context
self.trace_id = span_context.trace_id
self.root_span_id = span_context.span_id
# List of spans to report
self._spans_list = []
def Echo(self, request, context):
metadata = context.invocation_metadata()
logging.debug("Echo metadata: " + str(metadata))
metadata_dict = GetTracingMetadata(context)
logging.debug("Metadata dict: " + str(metadata_dict))
if 'x-b3-traceid' in metadata_dict:
trace_id = metadata_dict['x-b3-traceid']
logging.debug("Trace ID: " + trace_id)
span_context = SpanContext(trace_id=trace_id,
span_id=metadata_dict['x-b3-spanid'])
tracer = Tracer(span_context=span_context,
exporter=exporter,
sampler=always_on.AlwaysOnSampler())
with tracer.span(name='echo') as span:
logging.debug("Processing Echo: " + str(request))
span.add_attribute("message", request.message)
time.sleep(0.2)
return echo_pb2.EchoResponse(message=request.message)
def __init__(self,
span_context=None,
sampler=None,
exporter=None,
propagator=None):
if span_context is None:
span_context = SpanContext()
if sampler is None:
sampler = always_on.AlwaysOnSampler()
if exporter is None:
exporter = print_exporter.PrintExporter()
if propagator is None:
propagator = google_cloud_format.GoogleCloudFormatPropagator()
self.span_context = span_context
self.sampler = sampler
self.exporter = exporter
self.propagator = propagator
self.tracer = self.get_tracer()
self.store_tracer()
def __init__(self,
span_context=None,
sampler=None,
exporter=None,
propagator=None):
if span_context is None:
span_context = SpanContext()
if sampler is None:
sampler = samplers.ProbabilitySampler()
if exporter is None:
exporter = print_exporter.PrintExporter()
if propagator is None:
propagator = \
trace_context_http_header_format.TraceContextPropagator()
self.span_context = span_context
self.sampler = sampler
self.exporter = exporter
self.propagator = propagator
self.tracer = self.get_tracer()
self.store_tracer()
def test_span_data_to_envelope(self):
from opencensus.trace.span import SpanKind
from opencensus.trace.span_context import SpanContext
from opencensus.trace.span_data import SpanData
from opencensus.trace.status import Status
from opencensus.trace.trace_options import TraceOptions
from opencensus.trace.tracestate import Tracestate
exporter = trace_exporter.AzureExporter(
instrumentation_key='12345678-1234-5678-abcd-12345678abcd',
storage_path=os.path.join(TEST_FOLDER, self.id()),
)
# SpanKind.CLIENT HTTP
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={
'component': 'HTTP',
'http.method': 'GET',
'http.url': 'https://www.wikipedia.org/wiki/Rabbit',
'http.status_code': 200,
},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=[
Link('6e0c63257de34c90bf9efcd03927272e', '6e0c63257de34c91')
],
status=Status(0),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.CLIENT,
))
self.assertEqual(
envelope.iKey,
'12345678-1234-5678-abcd-12345678abcd')
self.assertEqual(
envelope.name,
'Microsoft.ApplicationInsights.RemoteDependency')
self.assertEqual(
envelope.tags['ai.operation.parentId'],
'6e0c63257de34c93')
self.assertEqual(
envelope.tags['ai.operation.id'],
'6e0c63257de34c90bf9efcd03927272e')
self.assertEqual(
envelope.time,
'2010-10-24T07:28:38.123456Z')
self.assertEqual(
envelope.data.baseData.name,
'GET /wiki/Rabbit')
self.assertEqual(
envelope.data.baseData.data,
'https://www.wikipedia.org/wiki/Rabbit')
self.assertEqual(
envelope.data.baseData.target,
'www.wikipedia.org')
self.assertEqual(
envelope.data.baseData.id,
'6e0c63257de34c92')
self.assertEqual(
envelope.data.baseData.resultCode,
'200')
self.assertEqual(
envelope.data.baseData.duration,
'0.00:00:00.111')
self.assertEqual(
envelope.data.baseData.type,
'HTTP')
self.assertEqual(
envelope.data.baseType,
'RemoteDependencyData')
json_dict = json.loads(
envelope.data.baseData.properties["_MS.links"]
)[0]
self.assertEqual(
json_dict["id"],
"6e0c63257de34c91",
)
self.assertEqual(
json_dict["operation_Id"],
"6e0c63257de34c90bf9efcd03927272e",
)
# SpanKind.CLIENT unknown type
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=Status(0),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.CLIENT,
))
self.assertEqual(
envelope.iKey,
'12345678-1234-5678-abcd-12345678abcd')
self.assertEqual(
envelope.name,
'Microsoft.ApplicationInsights.RemoteDependency')
self.assertEqual(
envelope.tags['ai.operation.parentId'],
'6e0c63257de34c93')
self.assertEqual(
envelope.tags['ai.operation.id'],
'6e0c63257de34c90bf9efcd03927272e')
self.assertEqual(
envelope.time,
'2010-10-24T07:28:38.123456Z')
self.assertEqual(
envelope.data.baseData.name,
'test')
self.assertEqual(
envelope.data.baseData.id,
'6e0c63257de34c92')
self.assertEqual(
envelope.data.baseData.duration,
'0.00:00:00.111')
self.assertEqual(
envelope.data.baseData.type,
None)
self.assertEqual(
envelope.data.baseType,
'RemoteDependencyData')
# SpanKind.CLIENT missing method
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={
'component': 'HTTP',
'http.url': 'https://www.wikipedia.org/wiki/Rabbit',
'http.status_code': 200,
},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=Status(0),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.CLIENT,
))
self.assertEqual(
envelope.iKey,
'12345678-1234-5678-abcd-12345678abcd')
self.assertEqual(
envelope.name,
'Microsoft.ApplicationInsights.RemoteDependency')
self.assertEqual(
envelope.tags['ai.operation.parentId'],
'6e0c63257de34c93')
self.assertEqual(
envelope.tags['ai.operation.id'],
'6e0c63257de34c90bf9efcd03927272e')
self.assertEqual(
envelope.time,
'2010-10-24T07:28:38.123456Z')
self.assertEqual(
envelope.data.baseData.name,
'test')
self.assertEqual(
envelope.data.baseData.data,
'https://www.wikipedia.org/wiki/Rabbit')
self.assertEqual(
envelope.data.baseData.target,
'www.wikipedia.org')
self.assertEqual(
envelope.data.baseData.id,
'6e0c63257de34c92')
self.assertEqual(
envelope.data.baseData.resultCode,
'200')
self.assertEqual(
envelope.data.baseData.duration,
'0.00:00:00.111')
self.assertEqual(
envelope.data.baseData.type,
'HTTP')
self.assertEqual(
envelope.data.baseType,
'RemoteDependencyData')
# SpanKind.SERVER HTTP - 200 request
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={
'component': 'HTTP',
'http.method': 'GET',
'http.path': '/wiki/Rabbit',
'http.route': '/wiki/Rabbit',
'http.url': 'https://www.wikipedia.org/wiki/Rabbit',
'http.status_code': 200,
},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=Status(0),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.SERVER,
))
self.assertEqual(
envelope.iKey,
'12345678-1234-5678-abcd-12345678abcd')
self.assertEqual(
envelope.name,
'Microsoft.ApplicationInsights.Request')
self.assertEqual(
envelope.tags['ai.operation.parentId'],
'6e0c63257de34c93')
self.assertEqual(
envelope.tags['ai.operation.id'],
'6e0c63257de34c90bf9efcd03927272e')
self.assertEqual(
envelope.tags['ai.operation.name'],
'GET /wiki/Rabbit')
self.assertEqual(
envelope.time,
'2010-10-24T07:28:38.123456Z')
self.assertEqual(
envelope.data.baseData.id,
'6e0c63257de34c92')
self.assertEqual(
envelope.data.baseData.duration,
'0.00:00:00.111')
self.assertEqual(
envelope.data.baseData.responseCode,
'200')
self.assertEqual(
envelope.data.baseData.name,
'GET /wiki/Rabbit')
self.assertEqual(
envelope.data.baseData.properties['request.name'],
'GET /wiki/Rabbit')
self.assertEqual(
envelope.data.baseData.success,
True)
self.assertEqual(
envelope.data.baseData.url,
'https://www.wikipedia.org/wiki/Rabbit')
self.assertEqual(
envelope.data.baseData.properties['request.url'],
'https://www.wikipedia.org/wiki/Rabbit')
self.assertEqual(
envelope.data.baseType,
'RequestData')
# SpanKind.SERVER HTTP - Failed request
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={
'component': 'HTTP',
'http.method': 'GET',
'http.path': '/wiki/Rabbit',
'http.route': '/wiki/Rabbit',
'http.url': 'https://www.wikipedia.org/wiki/Rabbit',
'http.status_code': 400,
},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=Status(0),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.SERVER,
))
self.assertEqual(
envelope.iKey,
'12345678-1234-5678-abcd-12345678abcd')
self.assertEqual(
envelope.name,
'Microsoft.ApplicationInsights.Request')
self.assertEqual(
envelope.tags['ai.operation.parentId'],
'6e0c63257de34c93')
self.assertEqual(
envelope.tags['ai.operation.id'],
'6e0c63257de34c90bf9efcd03927272e')
self.assertEqual(
envelope.tags['ai.operation.name'],
'GET /wiki/Rabbit')
self.assertEqual(
envelope.time,
'2010-10-24T07:28:38.123456Z')
self.assertEqual(
envelope.data.baseData.id,
'6e0c63257de34c92')
self.assertEqual(
envelope.data.baseData.duration,
'0.00:00:00.111')
self.assertEqual(
envelope.data.baseData.responseCode,
'400')
self.assertEqual(
envelope.data.baseData.name,
'GET /wiki/Rabbit')
self.assertEqual(
envelope.data.baseData.success,
False)
self.assertEqual(
envelope.data.baseData.url,
'https://www.wikipedia.org/wiki/Rabbit')
self.assertEqual(
envelope.data.baseType,
'RequestData')
# SpanKind.SERVER unknown type
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=Status(0),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.SERVER,
))
self.assertEqual(
envelope.iKey,
'12345678-1234-5678-abcd-12345678abcd')
self.assertEqual(
envelope.name,
'Microsoft.ApplicationInsights.Request')
self.assertEqual(
envelope.tags['ai.operation.parentId'],
'6e0c63257de34c93')
self.assertEqual(
envelope.tags['ai.operation.id'],
'6e0c63257de34c90bf9efcd03927272e')
self.assertEqual(
envelope.time,
'2010-10-24T07:28:38.123456Z')
self.assertEqual(
envelope.data.baseData.id,
'6e0c63257de34c92')
self.assertEqual(
envelope.data.baseData.duration,
'0.00:00:00.111')
self.assertEqual(
envelope.data.baseType,
'RequestData')
# SpanKind.UNSPECIFIED
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id=None,
attributes={'key1': 'value1'},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=Status(0),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.UNSPECIFIED,
))
self.assertEqual(
envelope.iKey,
'12345678-1234-5678-abcd-12345678abcd')
self.assertEqual(
envelope.name,
'Microsoft.ApplicationInsights.RemoteDependency')
self.assertRaises(
KeyError,
lambda: envelope.tags['ai.operation.parentId'])
self.assertEqual(
envelope.tags['ai.operation.id'],
'6e0c63257de34c90bf9efcd03927272e')
self.assertEqual(
envelope.time,
'2010-10-24T07:28:38.123456Z')
self.assertEqual(
envelope.data.baseData.name,
'test')
self.assertEqual(
envelope.data.baseData.duration,
'0.00:00:00.111')
self.assertEqual(
envelope.data.baseData.id,
'6e0c63257de34c92')
self.assertEqual(
envelope.data.baseData.type,
'INPROC')
self.assertEqual(
envelope.data.baseData.success,
True
)
self.assertEqual(
envelope.data.baseType,
'RemoteDependencyData')
# Status server status code attribute
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={
'http.status_code': 201
},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=Status(0),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.SERVER,
))
self.assertEqual(envelope.data.baseData.responseCode, "201")
self.assertTrue(envelope.data.baseData.success)
# Status server status code attribute missing
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=Status(1),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.SERVER,
))
self.assertFalse(envelope.data.baseData.success)
# Server route attribute missing
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={
'component': 'HTTP',
'http.method': 'GET',
'http.path': '/wiki/Rabbitz',
'http.url': 'https://www.wikipedia.org/wiki/Rabbit',
'http.status_code': 400,
},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=Status(1),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.SERVER,
))
self.assertEqual(envelope.data.baseData.properties['request.name'],
'GET /wiki/Rabbitz')
# Server route and path attribute missing
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={
'component': 'HTTP',
'http.method': 'GET',
'http.url': 'https://www.wikipedia.org/wiki/Rabbit',
'http.status_code': 400,
},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=Status(1),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.SERVER,
))
self.assertIsNone(
envelope.data.baseData.properties.get('request.name'))
# Status client status code attribute
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={
'http.status_code': 201
},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=Status(0),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.CLIENT,
))
self.assertEqual(envelope.data.baseData.resultCode, "201")
self.assertTrue(envelope.data.baseData.success)
# Status client status code attributes missing
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=Status(1),
annotations=None,
message_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.CLIENT,
))
self.assertFalse(envelope.data.baseData.success)
exporter._stop()
def new_span_context(trace_id, span_id):
return SpanContext(trace_id=trace_id, span_id=span_id)
def __init__(self):
self.span_context = SpanContext(
trace_options=trace_options.TraceOptions(0))
def test_span_data_to_envelope(self):
from opencensus.ext.azure.common import Options
from opencensus.trace.span import SpanKind
from opencensus.trace.span_context import SpanContext
from opencensus.trace.span_data import SpanData
from opencensus.trace.trace_options import TraceOptions
from opencensus.trace.tracestate import Tracestate
exporter = trace_exporter.AzureExporter(
Options(
instrumentation_key='12345678-1234-5678-abcd-12345678abcd',
storage_path=os.path.join(TEST_FOLDER, 'bar'),
),
)
# SpanKind.CLIENT HTTP
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={
'http.method': 'GET',
'http.url': 'https://www.wikipedia.org/wiki/Rabbit',
'http.status_code': 200,
},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=None,
time_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.CLIENT,
))
self.assertEqual(
envelope.iKey,
'12345678-1234-5678-abcd-12345678abcd')
self.assertEqual(
envelope.name,
'Microsoft.ApplicationInsights.RemoteDependency')
self.assertEqual(
envelope.tags['ai.operation.parentId'],
'|6e0c63257de34c90bf9efcd03927272e.6e0c63257de34c93.')
self.assertEqual(
envelope.tags['ai.operation.id'],
'6e0c63257de34c90bf9efcd03927272e')
self.assertEqual(
envelope.time,
'2010-10-24T07:28:38.123456Z')
self.assertEqual(
envelope.data.baseData.name,
'www.wikipedia.org')
self.assertEqual(
envelope.data.baseData.id,
'|6e0c63257de34c90bf9efcd03927272e.6e0c63257de34c92.')
self.assertEqual(
envelope.data.baseData.resultCode,
'200')
self.assertEqual(
envelope.data.baseData.duration,
'0.00:00:00.111')
self.assertEqual(
envelope.data.baseData.type,
'HTTP')
self.assertEqual(
envelope.data.baseType,
'RemoteDependencyData')
# SpanKind.CLIENT unknown type
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=None,
time_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.CLIENT,
))
self.assertEqual(
envelope.iKey,
'12345678-1234-5678-abcd-12345678abcd')
self.assertEqual(
envelope.name,
'Microsoft.ApplicationInsights.RemoteDependency')
self.assertEqual(
envelope.tags['ai.operation.parentId'],
'|6e0c63257de34c90bf9efcd03927272e.6e0c63257de34c93.')
self.assertEqual(
envelope.tags['ai.operation.id'],
'6e0c63257de34c90bf9efcd03927272e')
self.assertEqual(
envelope.time,
'2010-10-24T07:28:38.123456Z')
self.assertEqual(
envelope.data.baseData.name,
'test')
self.assertEqual(
envelope.data.baseData.id,
'|6e0c63257de34c90bf9efcd03927272e.6e0c63257de34c92.')
self.assertEqual(
envelope.data.baseData.duration,
'0.00:00:00.111')
self.assertEqual(
envelope.data.baseData.type,
'HTTP')
self.assertEqual(
envelope.data.baseType,
'RemoteDependencyData')
# SpanKind.SERVER HTTP
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={
'http.method': 'GET',
'http.url': 'https://www.wikipedia.org/wiki/Rabbit',
'http.status_code': 200,
},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=None,
time_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.SERVER,
))
self.assertEqual(
envelope.iKey,
'12345678-1234-5678-abcd-12345678abcd')
self.assertEqual(
envelope.name,
'Microsoft.ApplicationInsights.Request')
self.assertEqual(
envelope.tags['ai.operation.parentId'],
'|6e0c63257de34c90bf9efcd03927272e.6e0c63257de34c93.')
self.assertEqual(
envelope.tags['ai.operation.id'],
'6e0c63257de34c90bf9efcd03927272e')
self.assertEqual(
envelope.time,
'2010-10-24T07:28:38.123456Z')
self.assertEqual(
envelope.data.baseData.id,
'|6e0c63257de34c90bf9efcd03927272e.6e0c63257de34c92.')
self.assertEqual(
envelope.data.baseData.duration,
'0.00:00:00.111')
self.assertEqual(
envelope.data.baseData.responseCode,
'200')
self.assertEqual(
envelope.data.baseData.name,
'GET https://www.wikipedia.org/wiki/Rabbit')
self.assertEqual(
envelope.data.baseData.url,
'https://www.wikipedia.org/wiki/Rabbit')
self.assertEqual(
envelope.data.baseType,
'RequestData')
# SpanKind.SERVER unknown type
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id='6e0c63257de34c93',
attributes={},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=None,
time_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.SERVER,
))
self.assertEqual(
envelope.iKey,
'12345678-1234-5678-abcd-12345678abcd')
self.assertEqual(
envelope.name,
'Microsoft.ApplicationInsights.Request')
self.assertEqual(
envelope.tags['ai.operation.parentId'],
'|6e0c63257de34c90bf9efcd03927272e.6e0c63257de34c93.')
self.assertEqual(
envelope.tags['ai.operation.id'],
'6e0c63257de34c90bf9efcd03927272e')
self.assertEqual(
envelope.time,
'2010-10-24T07:28:38.123456Z')
self.assertEqual(
envelope.data.baseData.id,
'|6e0c63257de34c90bf9efcd03927272e.6e0c63257de34c92.')
self.assertEqual(
envelope.data.baseData.duration,
'0.00:00:00.111')
self.assertEqual(
envelope.data.baseType,
'RequestData')
# SpanKind.UNSPECIFIED
envelope = exporter.span_data_to_envelope(SpanData(
name='test',
context=SpanContext(
trace_id='6e0c63257de34c90bf9efcd03927272e',
span_id='6e0c63257de34c91',
trace_options=TraceOptions('1'),
tracestate=Tracestate(),
from_header=False,
),
span_id='6e0c63257de34c92',
parent_span_id=None,
attributes={'key1': 'value1'},
start_time='2010-10-24T07:28:38.123456Z',
end_time='2010-10-24T07:28:38.234567Z',
stack_trace=None,
links=None,
status=None,
time_events=None,
same_process_as_parent_span=None,
child_span_count=None,
span_kind=SpanKind.UNSPECIFIED,
))
self.assertEqual(
envelope.iKey,
'12345678-1234-5678-abcd-12345678abcd')
self.assertEqual(
envelope.name,
'Microsoft.ApplicationInsights.RemoteDependency')
self.assertRaises(
KeyError,
lambda: envelope.tags['ai.operation.parentId'])
self.assertEqual(
envelope.tags['ai.operation.id'],
'6e0c63257de34c90bf9efcd03927272e')
self.assertEqual(
envelope.time,
'2010-10-24T07:28:38.123456Z')
self.assertEqual(
envelope.data.baseData.name,
'test')
self.assertEqual(
envelope.data.baseData.duration,
'0.00:00:00.111')
self.assertEqual(
envelope.data.baseData.id,
'|6e0c63257de34c90bf9efcd03927272e.6e0c63257de34c92.')
self.assertEqual(
envelope.data.baseData.type,
'INPROC')
self.assertEqual(
envelope.data.baseType,
'RemoteDependencyData')
def from_headers(self, headers):
"""Generate a SpanContext object from B3 propagation headers.
:type headers: dict
:param headers: HTTP request headers.
:rtype: :class:`~opencensus.trace.span_context.SpanContext`
:returns: SpanContext generated from B3 propagation headers.
"""
if headers is None:
return SpanContext(from_header=False)
trace_id, span_id, sampled = None, None, None
state = headers.get(_STATE_HEADER_KEY)
if state:
fields = state.split('-', 4)
if len(fields) == 1:
sampled = fields[0]
elif len(fields) == 2:
trace_id, span_id = fields
elif len(fields) == 3:
trace_id, span_id, sampled = fields
elif len(fields) == 4:
trace_id, span_id, sampled, _parent_span_id = fields
else:
return SpanContext(from_header=False)
else:
trace_id = headers.get(_TRACE_ID_KEY)
span_id = headers.get(_SPAN_ID_KEY)
sampled = headers.get(_SAMPLED_KEY)
if sampled is not None:
# The specification encodes an enabled tracing decision as "1".
# In the wild pre-standard implementations might still send "true".
# "d" is set in the single header case when debugging is enabled.
sampled = sampled.lower() in ('1', 'd', 'true')
else:
# If there's no incoming sampling decision, it was deferred to us.
# Even though we set it to False here, we might still sample
# depending on the tracer configuration.
sampled = False
trace_options = TraceOptions()
trace_options.set_enabled(sampled)
# TraceId and SpanId headers both have to exist
if not trace_id or not span_id:
return SpanContext(trace_options=trace_options)
# Convert 64-bit trace ids to 128-bit
if len(trace_id) == 16:
trace_id = '0' * 16 + trace_id
span_context = SpanContext(trace_id=trace_id,
span_id=span_id,
trace_options=trace_options,
from_header=True)
return span_context
def nope(*args, **kwargs):
trace_options = TraceOptions()
trace_options.set_enabled(False)
return SpanContext(trace_options=trace_options)
from opencensus.trace.tracer import Tracer
RANGE = 10
FAILURE_NUMBER = 5
APP = 'edgelhd'
properties = {
'custom_dimensions': {
'job_id': '100',
'appName': APP,
'nodeId': '123'
}
}
tracer = Tracer(
exporter=AzureExporter(),
sampler=AlwaysOnSampler(),
span_context=SpanContext(),
)
logger = logging.getLogger(APP)
logger.setLevel(logging.INFO)
logger.addHandler(AzureLogHandler())
def something_happen(span, error=False):
time.sleep(randrange(RANGE))
if error:
# Cause an exception
try:
1 / 0
except Exception as e:
logger.exception(str(e), exc_info=True, extra=properties)