You've already forked mariadb-columnstore-engine
mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-10-31 18:30:33 +03:00
151903cd18
Don't log trace_params in tracing logger, because it already has all this data Don't print span attrs, it can contain lots of headers Save small part of the response into the span, if the response was a JSON string Added JSON logging of trace details into a separate file (to not spam the main log with machine readable stuff) Record part of the response into the span Set duration attribute in server spans Log 404 errors Colorize the traces (each span slightly changes the color of the parent span) Improve trace visualization with duration formatting and notes for request/response pairs
65 lines
2.4 KiB
Python
65 lines
2.4 KiB
Python
import logging
|
|
import time
|
|
from typing import Any, Dict, Optional
|
|
|
|
import cherrypy
|
|
from mcs_node_control.models.node_config import NodeConfig
|
|
from tracing.tracer import TracerBackend, TraceSpan
|
|
from tracing.utils import swallow_exceptions
|
|
|
|
logger = logging.getLogger("tracer")
|
|
json_logger = logging.getLogger("json_trace")
|
|
|
|
|
|
class TraceparentBackend(TracerBackend):
|
|
"""Default backend that logs span lifecycle and mirrors events/status."""
|
|
def __init__(self):
|
|
my_addresses = list(NodeConfig().get_network_addresses())
|
|
logger.info("My addresses: %s", my_addresses)
|
|
json_logger.info("my_addresses", extra={"my_addresses": my_addresses})
|
|
|
|
@swallow_exceptions
|
|
def on_span_start(self, span: TraceSpan) -> None:
|
|
logger.info(
|
|
"span_begin name='%s' kind=%s tid=%s sid=%s%s",
|
|
span.name, span.kind, span.trace_id, span.span_id,
|
|
f' psid={span.parent_span_id}' if span.parent_span_id else '',
|
|
)
|
|
json_logger.info("span_begin", extra=span.to_flat_dict())
|
|
|
|
@swallow_exceptions
|
|
def on_span_end(self, span: TraceSpan, exc: Optional[BaseException]) -> None:
|
|
end_ns = time.time_ns()
|
|
duration_ms = (end_ns - span.start_ns) / 1_000_000
|
|
span.set_attribute("duration_ms", duration_ms)
|
|
span.set_attribute("end_ns", end_ns)
|
|
|
|
# Try to set status code if not already set
|
|
if span.kind == "SERVER" and "http.status_code" not in span.attributes:
|
|
try:
|
|
status_str = str(cherrypy.response.status)
|
|
code = int(status_str.split()[0])
|
|
span.set_attribute("http.status_code", code)
|
|
except Exception:
|
|
pass
|
|
|
|
logger.info(
|
|
"span_end name='%s' kind=%s tid=%s sid=%s%s duration_ms=%.3f",
|
|
span.name, span.kind, span.trace_id, span.span_id,
|
|
f' psid={span.parent_span_id}' if span.parent_span_id else '',
|
|
duration_ms,
|
|
)
|
|
json_logger.info("span_end", extra=span.to_flat_dict())
|
|
|
|
@swallow_exceptions
|
|
def on_span_event(self, span: TraceSpan, name: str, attrs: Dict[str, Any]) -> None:
|
|
logger.info(
|
|
"span_event name='%s' tid=%s sid=%s%s attrs=%s",
|
|
name, span.trace_id, span.span_id,
|
|
f' psid={span.parent_span_id}' if span.parent_span_id else '',
|
|
attrs,
|
|
)
|
|
json_logger.info("span_event", extra=span.to_flat_dict())
|
|
|
|
|