You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-09-11 08:50:45 +03:00
9b98c5c20a
Added mirroring of spans into Sentry Tracer is a facade that redirects actions to tracing backends
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
"""
|
|
CherryPy tool that uses the tracer to start a span for each request.
|
|
"""
|
|
import socket
|
|
from typing import Dict
|
|
|
|
import cherrypy
|
|
|
|
from tracing.tracer import get_tracer
|
|
|
|
|
|
def _on_request_start() -> None:
|
|
req = cherrypy.request
|
|
tracer = get_tracer()
|
|
|
|
headers: Dict[str, str] = dict(req.headers or {})
|
|
tracer.notify_incoming_request(
|
|
headers=headers,
|
|
method=getattr(req, 'method', ''),
|
|
path=getattr(req, 'path_info', '')
|
|
)
|
|
trace_id, parent_span_id = tracer.extract_traceparent(headers)
|
|
tracer.set_incoming_context(trace_id, parent_span_id)
|
|
|
|
span_name = f"{getattr(req, 'method', 'HTTP')} {getattr(req, 'path_info', '/')}"
|
|
|
|
ctx = tracer.start_as_current_span(span_name, kind="SERVER")
|
|
span = ctx.__enter__()
|
|
span.set_attribute('http.method', getattr(req, 'method', ''))
|
|
span.set_attribute('http.path', getattr(req, 'path_info', ''))
|
|
span.set_attribute('client.ip', getattr(getattr(req, 'remote', None), 'ip', ''))
|
|
span.set_attribute('instance.hostname', socket.gethostname())
|
|
safe_headers = {k: v for k, v in headers.items() if k.lower() not in {'authorization', 'x-api-key'}}
|
|
span.set_attribute('sentry.incoming_headers', safe_headers)
|
|
req._trace_span_ctx = ctx
|
|
req._trace_span = span
|
|
|
|
tracer.inject_traceparent(cherrypy.response.headers) # type: ignore[arg-type]
|
|
|
|
|
|
def _on_request_end() -> None:
|
|
req = cherrypy.request
|
|
try:
|
|
status_str = str(cherrypy.response.status)
|
|
status_code = int(status_str.split()[0])
|
|
except Exception:
|
|
status_code = None
|
|
tracer = get_tracer()
|
|
tracer.notify_request_finished(status_code)
|
|
span = getattr(req, "_trace_span", None)
|
|
if span is not None and status_code is not None:
|
|
span.set_attribute('http.status_code', status_code)
|
|
ctx = getattr(req, "_trace_span_ctx", None)
|
|
if ctx is not None:
|
|
try:
|
|
ctx.__exit__(None, None, None)
|
|
finally:
|
|
req._trace_span_ctx = None
|
|
req._trace_span = None
|
|
|
|
|
|
def register_tracing_tools() -> None:
|
|
cherrypy.tools.trace = cherrypy.Tool("on_start_resource", _on_request_start, priority=10)
|
|
cherrypy.tools.trace_end = cherrypy.Tool("on_end_resource", _on_request_end, priority=80)
|
|
|
|
|
|
|