mirror of
https://github.com/hanzo-ds/driver.git
synced 2026-10-30 20:32:40 +00:00
zeekay
eb74c996a6
Imports read `datastore_driver`, distribution `datastore-driver`, so nothing has to import another product's name to talk to Hanzo Datastore. Cut from upstream 0.2.9 — the version this stack already ran — rather than from upstream HEAD, so it is a PURE RENAME with no behaviour change. Forking from HEAD would have dragged versions along with the name (connect alone jumped 0.8.17 -> 1.6.0, a major), which is a different and much riskier change wearing a rename's clothes. X-ClickHouse-* headers are untouched: they are wire protocol the server sends and expects, and a different token shape from the module name.
44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
|
|
class OpenTelemetryTraceContext(object):
|
|
traceparent_tpl = 'xx-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx-xxxxxxxxxxxxxxxx-xx'
|
|
translation = str.maketrans('1234567890abcdef', 'xxxxxxxxxxxxxxxx')
|
|
|
|
def __init__(self, traceparent, tracestate):
|
|
# xx-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx-xxxxxxxxxxxxxxxx-xx
|
|
# ^ ^ ^ ^
|
|
# version trace_id span_id flags
|
|
|
|
self.trace_id = None # UUID
|
|
self.span_id = None # UInt64
|
|
self.tracestate = tracestate # String
|
|
self.trace_flags = None # UInt8
|
|
|
|
if traceparent is not None:
|
|
self.parse_traceparent(traceparent)
|
|
|
|
super(OpenTelemetryTraceContext, self).__init__()
|
|
|
|
def parse_traceparent(self, traceparent):
|
|
traceparent = traceparent.lower()
|
|
|
|
if len(traceparent) != len(self.traceparent_tpl):
|
|
raise ValueError('unexpected length {}, expected {}'.format(
|
|
len(traceparent), len(self.traceparent_tpl)
|
|
))
|
|
|
|
if traceparent.translate(self.translation) != self.traceparent_tpl:
|
|
raise ValueError(
|
|
'Malformed traceparant header: {}'.format(traceparent)
|
|
)
|
|
|
|
parts = traceparent.split('-')
|
|
version = int(parts[0], 16)
|
|
if version != 0:
|
|
raise ValueError(
|
|
'unexpected version {}, expected 00'.format(parts[0])
|
|
)
|
|
|
|
self.trace_id = (int(parts[1][16:], 16) << 64) + int(parts[1][:16], 16)
|
|
self.span_id = int(parts[2], 16)
|
|
self.trace_flags = int(parts[3], 16)
|