1
0
mirror of https://github.com/mariadb-corporation/mariadb-columnstore-engine.git synced 2025-10-31 18:30:33 +03:00

Tracing fixes:

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
This commit is contained in:
Alexander Presnyakov
2025-09-04 17:21:25 +00:00
parent 194e7ccbdc
commit 151903cd18
13 changed files with 962 additions and 39 deletions

View File

@@ -2,37 +2,63 @@ 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("tracing")
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 trace_id=%s span_id=%s parent=%s attrs=%s",
"span_begin name='%s' kind=%s tid=%s sid=%s%s",
span.name, span.kind, span.trace_id, span.span_id,
span.parent_span_id, span.attributes,
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:
duration_ms = (time.time_ns() - span.start_ns) / 1_000_000
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 trace_id=%s span_id=%s parent=%s duration_ms=%.3f attrs=%s",
"span_end name='%s' kind=%s tid=%s sid=%s%s duration_ms=%.3f",
span.name, span.kind, span.trace_id, span.span_id,
span.parent_span_id, duration_ms, span.attributes,
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 trace_id=%s span_id=%s attrs=%s",
name, span.trace_id, span.span_id, attrs,
"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())