Добавил просмотр trace логов полностью

This commit is contained in:
2026-08-03 12:40:31 +03:00
parent e6509ee0cd
commit d2f5a22501
8 changed files with 433 additions and 81 deletions
+6
View File
@@ -2,6 +2,12 @@
## Установка
Быстрая установка через SSH-доступ к Git:
```bash
pip install "plba @ git+ssh://git@git.lesha.spb.ru/alex/plba.git@main"
```
Установка пакета напрямую из Git-репозитория через `pip`:
```bash
+1 -1
View File
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
[project]
name = "plba"
version = "0.4.0"
version = "0.5.0"
description = "Platform runtime for business applications"
readme = "README.md"
requires-python = ">=3.11"
+5
View File
@@ -90,9 +90,12 @@ class TraceLogRecord:
class TraceLogView:
trace_id: str
parent_id: str | None
alias: str = ""
child_ids: tuple[str, ...] = ()
records: tuple[TraceLogRecord, ...] = ()
ancestors: tuple[TraceLogView, ...] = ()
descendants: tuple[TraceLogView, ...] = ()
requested_trace_id: str | None = None
class TraceLogReader(Protocol):
@@ -101,5 +104,7 @@ class TraceLogReader(Protocol):
trace_id: str,
levels: tuple[TraceLevel, ...],
ancestor_depth: int | None = 0,
*,
view: Literal["tree", "single"] = "tree",
) -> TraceLogView | None:
"""Load trace context and filtered log records."""
+3 -1
View File
@@ -20,6 +20,7 @@ ActionResult = str | dict[str, object]
ActionHandler = Callable[[ControlActionRequest], Awaitable[ActionResult]]
HealthHandler = Callable[[], Awaitable[HealthPayload]]
TraceResponseFormat = Literal["json", "text", "html"]
TraceViewMode = Literal["tree", "single"]
@dataclass(slots=True)
@@ -27,7 +28,8 @@ class TraceQueryRequest:
levels: tuple[TraceLevel, ...] = ("ERROR", "WARNING", "INFO")
include_attrs_json: bool = False
response_format: TraceResponseFormat = "html"
ancestor_depth: int | None = 0
ancestor_depth: int | None = None
view: TraceViewMode = "tree"
TraceLookupHandler = Callable[[str, TraceQueryRequest], Awaitable[TraceLogView]]
+124 -28
View File
@@ -21,11 +21,16 @@ class TraceRequestParser:
response_format = raw_format.strip().lower()
if response_format not in {"json", "text", "html"}:
raise ValueError(f"unsupported trace format: {raw_format}")
raw_view = request.query_params.get("view", "tree")
view = raw_view.strip().lower()
if view not in {"tree", "single"}:
raise ValueError(f"unsupported trace view: {raw_view}")
return TraceQueryRequest(
levels=self._trace_levels(raw_levels),
include_attrs_json=self._bool_param(request, "attrs_json") or False,
response_format=response_format,
ancestor_depth=self._ancestor_depth(request),
ancestor_depth=self._ancestor_depth(request, default_all=view == "tree"),
view=view,
)
def _trace_levels(self, raw_levels: str | None) -> tuple[TraceLevel, ...]:
@@ -51,10 +56,10 @@ class TraceRequestParser:
return False
raise ValueError(f"invalid boolean query parameter: {name}={raw_value}")
def _ancestor_depth(self, request: Request) -> int | None:
def _ancestor_depth(self, request: Request, *, default_all: bool) -> int | None:
raw_value = request.query_params.get("ancestor_depth")
if raw_value is None:
return 0
return None if default_all else 0
normalized = raw_value.strip().lower()
if normalized == "all":
return None
@@ -69,26 +74,35 @@ class TraceRequestParser:
class TraceResponseRenderer:
def render(self, trace_view: TraceLogView, request: TraceQueryRequest) -> Response:
if request.view == "single":
if request.response_format == "json":
return self._render_json_single(trace_view, request)
if request.response_format == "html":
return self._render_html_single(trace_view, request)
return self._render_text_single(trace_view, request)
if request.response_format == "json":
return self._render_json(trace_view, request)
return self._render_json_tree(trace_view, request)
if request.response_format == "html":
return self._render_html(trace_view, request)
return self._render_text(trace_view, request)
return self._render_html_tree(trace_view, request)
return self._render_text_tree(trace_view, request)
def _render_json(self, trace_view: TraceLogView, request: TraceQueryRequest) -> JSONResponse:
def _render_json_single(self, trace_view: TraceLogView, request: TraceQueryRequest) -> JSONResponse:
return JSONResponse(
content={
"trace_id": trace_view.trace_id,
"parent_id": trace_view.parent_id or "",
"child_ids": list(trace_view.child_ids),
"messages": [record.as_dict(include_attrs_json=request.include_attrs_json) for record in trace_view.records],
"ancestors": [self._trace_payload(view, request) for view in trace_view.ancestors],
"ancestors": [self._trace_payload_single(view, request) for view in trace_view.ancestors],
}
)
def _render_text(self, trace_view: TraceLogView, request: TraceQueryRequest) -> PlainTextResponse:
def _render_json_tree(self, trace_view: TraceLogView, request: TraceQueryRequest) -> JSONResponse:
return JSONResponse(content=self._trace_payload_tree(trace_view, request))
def _render_text_single(self, trace_view: TraceLogView, request: TraceQueryRequest) -> PlainTextResponse:
lineage = [*trace_view.ancestors, trace_view]
lines = self._text_trace_summary_lines(trace_view)
lines = self._text_trace_summary_lines_single(trace_view)
for index, entry in enumerate(lineage):
if index == 0:
lines.append("")
@@ -97,10 +111,29 @@ class TraceResponseRenderer:
lines.extend(self._text_trace_log_lines(entry, request))
return PlainTextResponse(content="\n".join(lines))
def _render_html(self, trace_view: TraceLogView, request: TraceQueryRequest) -> HTMLResponse:
def _render_text_tree(self, trace_view: TraceLogView, request: TraceQueryRequest) -> PlainTextResponse:
lines = self._text_trace_summary_lines_tree(trace_view)
for index, entry in enumerate(self._tree_walk(trace_view)):
if index == 0:
lines.append("")
else:
lines.extend(["", ""])
lines.extend(self._text_trace_log_lines(entry, request))
return PlainTextResponse(content="\n".join(lines))
def _render_html_single(self, trace_view: TraceLogView, request: TraceQueryRequest) -> HTMLResponse:
title = escape(f"Trace {trace_view.trace_id}")
lines = self._html_lines(trace_view, request)
html = f"""<!DOCTYPE html>
lines = self._html_lines_single(trace_view, request)
return HTMLResponse(content=self._html_document(title, lines))
def _render_html_tree(self, trace_view: TraceLogView, request: TraceQueryRequest) -> HTMLResponse:
requested = trace_view.requested_trace_id or trace_view.trace_id
title = escape(f"Trace {requested}")
lines = self._html_lines_tree(trace_view, request)
return HTMLResponse(content=self._html_document(title, lines))
def _html_document(self, title: str, lines: str) -> str:
return f'''<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="utf-8">
@@ -160,8 +193,7 @@ class TraceResponseRenderer:
{lines}
</div>
</body>
</html>"""
return HTMLResponse(content=html)
</html>'''
def _child_id_lines(self, child_ids: tuple[str, ...]) -> list[str]:
lines = ["child_ids:"]
@@ -178,6 +210,7 @@ class TraceResponseRenderer:
"format": "html",
"levels": ",".join(request.levels),
"attrs_json": "true" if request.include_attrs_json else "false",
"view": request.view,
}
if request.ancestor_depth is None:
params["ancestor_depth"] = "all"
@@ -186,9 +219,9 @@ class TraceResponseRenderer:
query = urlencode(params)
return f"/traces/{trace_id}?{query}"
def _html_lines(self, trace_view: TraceLogView, request: TraceQueryRequest) -> str:
def _html_lines_single(self, trace_view: TraceLogView, request: TraceQueryRequest) -> str:
lineage = [*trace_view.ancestors, trace_view]
lines = self._html_trace_summary_lines(trace_view, request)
lines = self._html_trace_summary_lines_single(trace_view, request)
for index, entry in enumerate(lineage):
if index == 0:
lines.append(self._html_plain_line(""))
@@ -197,7 +230,17 @@ class TraceResponseRenderer:
lines.extend(self._html_trace_log_lines(entry, request))
return "".join(lines)
def _trace_payload(self, trace_view: TraceLogView, request: TraceQueryRequest) -> dict[str, object]:
def _html_lines_tree(self, trace_view: TraceLogView, request: TraceQueryRequest) -> str:
lines = self._html_trace_summary_lines_tree(trace_view, request)
for index, entry in enumerate(self._tree_walk(trace_view)):
if index == 0:
lines.append(self._html_plain_line(""))
else:
lines.extend([self._html_plain_line(""), self._html_plain_line("")])
lines.extend(self._html_trace_log_lines(entry, request))
return "".join(lines)
def _trace_payload_single(self, trace_view: TraceLogView, request: TraceQueryRequest) -> dict[str, object]:
return {
"trace_id": trace_view.trace_id,
"parent_id": trace_view.parent_id or "",
@@ -205,17 +248,38 @@ class TraceResponseRenderer:
"messages": [record.as_dict(include_attrs_json=request.include_attrs_json) for record in trace_view.records],
}
def _text_trace_summary_lines(self, trace_view: TraceLogView) -> list[str]:
def _trace_payload_tree(self, trace_view: TraceLogView, request: TraceQueryRequest) -> dict[str, object]:
payload: dict[str, object] = {
"trace_id": trace_view.trace_id,
"parent_id": trace_view.parent_id or "",
"child_ids": list(trace_view.child_ids),
"messages": [record.as_dict(include_attrs_json=request.include_attrs_json) for record in trace_view.records],
"descendants": [self._trace_payload_tree(view, request) for view in trace_view.descendants],
}
if trace_view.requested_trace_id and trace_view.requested_trace_id != trace_view.trace_id:
payload["requested_trace_id"] = trace_view.requested_trace_id
return payload
def _text_trace_summary_lines_single(self, trace_view: TraceLogView) -> list[str]:
return [
f"trace_id: {trace_view.trace_id}",
f"parent_id: {trace_view.parent_id or ''}",
*self._child_id_lines(trace_view.child_ids),
]
def _text_trace_summary_lines_tree(self, trace_view: TraceLogView) -> list[str]:
lines = [f"trace_id: {trace_view.trace_id}"]
requested = trace_view.requested_trace_id or trace_view.trace_id
if requested != trace_view.trace_id:
lines.append(f"requested_trace_id: {requested}")
lines.append(f"parent_id: {trace_view.parent_id or ''}")
lines.extend(self._child_id_lines(trace_view.child_ids))
return lines
def _text_trace_log_lines(self, trace_view: TraceLogView, request: TraceQueryRequest) -> list[str]:
lines = [
TRACE_SECTION_SEPARATOR,
f"trace_id: {trace_view.trace_id}",
self._trace_context_label(trace_view),
"",
]
previous_step: str | None = None
@@ -230,7 +294,7 @@ class TraceResponseRenderer:
lines.append(self._text_message(record, request.include_attrs_json))
return lines
def _html_trace_summary_lines(self, trace_view: TraceLogView, request: TraceQueryRequest) -> list[str]:
def _html_trace_summary_lines_single(self, trace_view: TraceLogView, request: TraceQueryRequest) -> list[str]:
return [
self._html_plain_line(f"trace_id: {self._trace_link(trace_view.trace_id, request)}"),
self._html_plain_line(f"parent_id: {self._optional_trace_link(trace_view.parent_id, request)}"),
@@ -238,10 +302,24 @@ class TraceResponseRenderer:
*(self._html_plain_line(f" - {self._trace_link(child_id, request)}") for child_id in trace_view.child_ids),
]
def _html_trace_summary_lines_tree(self, trace_view: TraceLogView, request: TraceQueryRequest) -> list[str]:
lines = [self._html_plain_line(f"trace_id: {self._trace_link(trace_view.trace_id, request)}")]
requested = trace_view.requested_trace_id or trace_view.trace_id
if requested != trace_view.trace_id:
lines.append(self._html_plain_line(f"requested_trace_id: {self._trace_link(requested, request)}"))
lines.extend(
[
self._html_plain_line(f"parent_id: {self._optional_trace_link(trace_view.parent_id, request)}"),
self._html_plain_line("child_ids:"),
*(self._html_plain_line(f" - {self._trace_link(child_id, request)}") for child_id in trace_view.child_ids),
]
)
return lines
def _html_trace_log_lines(self, trace_view: TraceLogView, request: TraceQueryRequest) -> list[str]:
lines = [
self._html_plain_line(TRACE_SECTION_SEPARATOR),
self._html_plain_line(f"trace_id: {self._trace_link(trace_view.trace_id, request)}"),
self._html_plain_line(self._html_trace_context_label(trace_view, request)),
self._html_plain_line(""),
]
previous_step: str | None = None
@@ -260,29 +338,36 @@ class TraceResponseRenderer:
return lines
def _html_message_lines(self, record: TraceLogRecord, include_attrs_json: bool) -> list[str]:
lines = [self._html_colored_line(self._text_message(record, include_attrs_json), record.level)]
return lines
return [self._html_colored_line(self._text_message(record, include_attrs_json), record.level)]
def _html_plain_line(self, content: str) -> str:
return f"<div class=\"line\">{content or '&nbsp;'}</div>"
return f'<div class="line">{content or "&nbsp;"}</div>'
def _html_step_line(self, content: str) -> str:
return f"<div class=\"line\" style=\"color: var(--step);\">{escape(content) or '&nbsp;'}</div>"
return f'<div class="line" style="color: var(--step);">{escape(content) or "&nbsp;"}</div>'
def _html_colored_line(self, content: str, level: str) -> str:
level_class = self._level_class(level)
return f"<div class=\"line {level_class}\">{escape(content)}</div>"
return f'<div class="line {level_class}">{escape(content)}</div>'
def _trace_link(self, trace_id: str, request: TraceQueryRequest) -> str:
href = escape(self._trace_href(trace_id, request), quote=True)
text = escape(trace_id)
return f"<a href=\"{href}\">{text}</a>"
return f'<a href="{href}">{text}</a>'
def _optional_trace_link(self, trace_id: str | None, request: TraceQueryRequest) -> str:
if not trace_id:
return ""
return self._trace_link(trace_id, request)
def _trace_context_label(self, trace_view: TraceLogView) -> str:
return f"trace context: {trace_view.alias}, {trace_view.trace_id}"
def _html_trace_context_label(self, trace_view: TraceLogView, request: TraceQueryRequest) -> str:
alias = escape(trace_view.alias)
trace_link = self._trace_link(trace_view.trace_id, request)
return f"trace context: {alias}, {trace_link}"
def _level_class(self, level: str) -> str:
if level == "ERROR":
return "msg-error"
@@ -291,3 +376,14 @@ class TraceResponseRenderer:
if level == "INFO":
return "msg-info"
return "msg-debug"
def _tree_walk(self, trace_view: TraceLogView) -> list[TraceLogView]:
ordered: list[TraceLogView] = []
def _visit(node: TraceLogView) -> None:
ordered.append(node)
for child in node.descendants:
_visit(child)
_visit(trace_view)
return ordered
+1 -1
View File
@@ -141,7 +141,7 @@ class RuntimeManager:
reader = build_trace_log_reader(self.traces.transport)
if reader is None:
raise RuntimeError("trace log reader is not configured")
trace_view = reader.read_trace(trace_id, request.levels, request.ancestor_depth)
trace_view = reader.read_trace(trace_id, request.levels, request.ancestor_depth, view=request.view)
if trace_view is None:
raise KeyError(trace_id)
return trace_view
+70 -18
View File
@@ -1,11 +1,13 @@
from __future__ import annotations
import json
from typing import Any
from typing import Any, Literal
from app_runtime.contracts.trace import TraceLevel, TraceLogReader, TraceLogRecord, TraceLogView, TraceTransport
from app_runtime.tracing.transport import MySqlTraceConnectionFactory, MySqlTraceTransport
TraceViewMode = Literal["tree", "single"]
class MySqlTraceLogReader(TraceLogReader):
def __init__(self, connection_factory: MySqlTraceConnectionFactory) -> None:
@@ -16,19 +18,72 @@ class MySqlTraceLogReader(TraceLogReader):
trace_id: str,
levels: tuple[TraceLevel, ...],
ancestor_depth: int | None = 0,
*,
view: TraceViewMode = "tree",
) -> TraceLogView | None:
parent_id = self._read_parent_id(trace_id)
if parent_id is None and not self._trace_exists(trace_id):
context = self._read_context(trace_id)
if context is None:
return None
ancestors = self._read_ancestors(parent_id, levels, ancestor_depth)
parent_id, alias = context
if view == "single":
ancestors = self._read_ancestors(parent_id, levels, ancestor_depth)
child_ids = self._read_child_ids(trace_id)
records = self._read_records(trace_id, levels)
return TraceLogView(
trace_id=trace_id,
parent_id=parent_id,
alias=alias,
child_ids=tuple(child_ids),
records=tuple(records),
ancestors=tuple(ancestors),
)
root_trace_id = self._resolve_tree_root(trace_id, ancestor_depth)
root_view = self._read_tree(root_trace_id, levels)
if root_view is None:
return None
if root_trace_id == trace_id:
return root_view
return TraceLogView(
trace_id=root_view.trace_id,
parent_id=root_view.parent_id,
alias=root_view.alias,
child_ids=root_view.child_ids,
records=root_view.records,
ancestors=root_view.ancestors,
descendants=root_view.descendants,
requested_trace_id=trace_id,
)
def _resolve_tree_root(self, trace_id: str, ancestor_depth: int | None) -> str:
current_trace_id = trace_id
remaining_depth = ancestor_depth
while True:
context = self._read_context(current_trace_id)
if context is None:
return current_trace_id
parent_id, _alias = context
if parent_id is None:
return current_trace_id
if remaining_depth == 0:
return current_trace_id
current_trace_id = parent_id
if remaining_depth is not None:
remaining_depth -= 1
def _read_tree(self, trace_id: str, levels: tuple[TraceLevel, ...]) -> TraceLogView | None:
context = self._read_context(trace_id)
if context is None:
return None
parent_id, alias = context
child_ids = self._read_child_ids(trace_id)
records = self._read_records(trace_id, levels)
descendants = [self._read_tree(child_id, levels) for child_id in child_ids]
return TraceLogView(
trace_id=trace_id,
parent_id=parent_id,
alias=alias,
child_ids=tuple(child_ids),
records=tuple(records),
ancestors=tuple(ancestors),
records=tuple(self._read_records(trace_id, levels)),
descendants=tuple(view for view in descendants if view is not None),
)
def _read_ancestors(
@@ -43,11 +98,15 @@ class MySqlTraceLogReader(TraceLogReader):
ancestors: list[TraceLogView] = []
current_trace_id = parent_id
while current_trace_id is not None and (remaining_depth is None or remaining_depth > 0):
current_parent_id = self._read_parent_id(current_trace_id)
context = self._read_context(current_trace_id)
if context is None:
break
current_parent_id, alias = context
ancestors.append(
TraceLogView(
trace_id=current_trace_id,
parent_id=current_parent_id,
alias=alias,
child_ids=tuple(self._read_child_ids(current_trace_id)),
records=tuple(self._read_records(current_trace_id, levels)),
)
@@ -58,22 +117,15 @@ class MySqlTraceLogReader(TraceLogReader):
ancestors.reverse()
return ancestors
def _trace_exists(self, trace_id: str) -> bool:
query = "SELECT 1 FROM trace_contexts WHERE trace_id = %s"
with self._connection_factory.connect() as connection:
with connection.cursor() as cursor:
cursor.execute(query, (trace_id,))
return cursor.fetchone() is not None
def _read_parent_id(self, trace_id: str) -> str | None:
query = "SELECT parent_id FROM trace_contexts WHERE trace_id = %s"
def _read_context(self, trace_id: str) -> tuple[str | None, str] | None:
query = "SELECT parent_id, alias FROM trace_contexts WHERE trace_id = %s"
with self._connection_factory.connect() as connection:
with connection.cursor() as cursor:
cursor.execute(query, (trace_id,))
row = cursor.fetchone()
if row is None:
return None
return self._string_or_none(row.get("parent_id"))
return self._string_or_none(row.get("parent_id")), self._string_or_none(row.get("alias")) or ""
def _read_records(self, trace_id: str, levels: tuple[TraceLevel, ...]) -> list[TraceLogRecord]:
placeholders = ", ".join(["%s"] * len(levels))
+223 -32
View File
@@ -54,6 +54,7 @@ def test_trace_endpoint_returns_html_by_default() -> None:
return TraceLogView(
trace_id="trace-1",
parent_id="root-trace",
alias="mailbox_sync",
child_ids=("child-1", "child-2"),
records=(
_trace_record(row_id=1, level="ERROR", message="first error"),
@@ -69,7 +70,7 @@ def test_trace_endpoint_returns_html_by_default() -> None:
assert response.status_code == 200
assert response.headers["content-type"].startswith("text/html")
assert "trace_id:" in response.text
assert "trace context: mailbox_sync, " in response.text
assert "first error" in response.text
assert "second warning" in response.text
assert captured == [
@@ -79,7 +80,8 @@ def test_trace_endpoint_returns_html_by_default() -> None:
levels=("ERROR", "WARNING", "INFO"),
include_attrs_json=False,
response_format="html",
ancestor_depth=0,
ancestor_depth=None,
view="tree",
),
)
]
@@ -90,6 +92,7 @@ def test_trace_endpoint_returns_text_when_requested() -> None:
return TraceLogView(
trace_id="trace-1",
parent_id="root-trace",
alias="mailbox_sync",
child_ids=("child-1", "child-2"),
records=(
_trace_record(row_id=1, level="ERROR", message="first error"),
@@ -112,7 +115,7 @@ def test_trace_endpoint_returns_text_when_requested() -> None:
" - child-2\n"
"\n"
"==============================\n"
"trace_id: trace-1\n"
"trace context: mailbox_sync, trace-1\n"
"\n"
"step: process\n"
"first error\n"
@@ -125,6 +128,7 @@ def test_trace_endpoint_appends_attrs_json_in_text_mode() -> None:
return TraceLogView(
trace_id="trace-1",
parent_id=None,
alias="mailbox_sync",
child_ids=(),
records=(
_trace_record(row_id=1, level="ERROR", message="failure", attrs_json={"attempt": 2, "source": "crm"}),
@@ -144,7 +148,7 @@ def test_trace_endpoint_appends_attrs_json_in_text_mode() -> None:
"child_ids:\n"
"\n"
"==============================\n"
"trace_id: trace-1\n"
"trace context: mailbox_sync, trace-1\n"
"\n"
"step: process\n"
'failure, {"attempt":2,"source":"crm"}'
@@ -156,6 +160,7 @@ def test_trace_endpoint_separates_messages_by_step_in_text_mode() -> None:
return TraceLogView(
trace_id="trace-1",
parent_id=None,
alias="mailbox_sync",
child_ids=(),
records=(
_trace_record(row_id=1, level="INFO", message="load first", step="load_stocks"),
@@ -177,7 +182,7 @@ def test_trace_endpoint_separates_messages_by_step_in_text_mode() -> None:
"child_ids:\n"
"\n"
"==============================\n"
"trace_id: trace-1\n"
"trace context: mailbox_sync, trace-1\n"
"\n"
"step: load_stocks\n"
"load first\n"
@@ -201,7 +206,7 @@ def test_trace_endpoint_returns_json_payload() -> None:
client = _build_client(trace_provider)
try:
response = client.get("/traces/trace-1?format=json&attrs_json=true&levels=info")
response = client.get("/traces/trace-1?format=json&attrs_json=true&levels=info&view=single")
finally:
client.close()
@@ -257,7 +262,7 @@ def test_trace_endpoint_returns_json_payload_with_ancestors() -> None:
client = _build_client(trace_provider)
try:
response = client.get("/traces/trace-1?format=json&ancestor_depth=1")
response = client.get("/traces/trace-1?format=json&ancestor_depth=1&view=single")
finally:
client.close()
@@ -303,6 +308,7 @@ def test_trace_endpoint_returns_html_page_with_related_links() -> None:
return TraceLogView(
trace_id="trace-1",
parent_id="parent-1",
alias="order_confirmation",
child_ids=("child-1", "child-2"),
records=(
_trace_record(row_id=1, level="INFO", message="loaded prices", step="load_stocks", status="ok"),
@@ -327,14 +333,14 @@ def test_trace_endpoint_returns_html_page_with_related_links() -> None:
assert "--error: #ff817d;" in response.text
assert "--other: #ececec;" in response.text
assert 'font: 13px/1.1 "SFMono-Regular", monospace;' in response.text
assert '<div class="line">trace_id: <a href="/traces/trace-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true">trace-1</a></div>' in response.text
assert '<div class="line">parent_id: <a href="/traces/parent-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true">parent-1</a></div>' in response.text
assert '<div class="line">trace_id: <a href="/traces/trace-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true&amp;view=tree&amp;ancestor_depth=all">trace-1</a></div>' in response.text
assert '<div class="line">parent_id: <a href="/traces/parent-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true&amp;view=tree&amp;ancestor_depth=all">parent-1</a></div>' in response.text
assert '<div class="line">child_ids:</div>' in response.text
assert '<div class="line"> - <a href="/traces/child-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true">child-1</a></div>' in response.text
assert '<div class="line"> - <a href="/traces/child-2?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true">child-2</a></div>' in response.text
assert '<div class="line"> - <a href="/traces/child-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true&amp;view=tree&amp;ancestor_depth=all">child-1</a></div>' in response.text
assert '<div class="line"> - <a href="/traces/child-2?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true&amp;view=tree&amp;ancestor_depth=all">child-2</a></div>' in response.text
assert '<div class="line">==============================</div>' in response.text
assert '<div class="line">trace context: order_confirmation, <a href="/traces/trace-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true&amp;view=tree&amp;ancestor_depth=all">trace-1</a></div>' in response.text
assert '<div class="line" style="color: var(--step);">load_stocks</div>' in response.text
assert '<div class="line">trace_id: <a href="/traces/trace-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true">trace-1</a></div>' in response.text
assert '<div class="line" style="color: var(--step);">filter_stocks</div>' in response.text
assert "loaded prices" in response.text
assert "filtered suspicious ticker" in response.text
@@ -348,18 +354,21 @@ def test_trace_endpoint_renders_ancestors_in_text_mode() -> None:
return TraceLogView(
trace_id="trace-1",
parent_id="parent-1",
alias="child",
child_ids=(),
records=(_trace_record(row_id=1, level="INFO", message="child message"),),
ancestors=(
TraceLogView(
trace_id="root-1",
parent_id=None,
alias="root",
child_ids=("parent-1",),
records=(_trace_record(row_id=2, level="INFO", message="root message"),),
),
TraceLogView(
trace_id="parent-1",
parent_id="root-1",
alias="parent",
child_ids=("trace-1",),
records=(_trace_record(row_id=3, level="WARNING", message="parent message"),),
),
@@ -368,7 +377,7 @@ def test_trace_endpoint_renders_ancestors_in_text_mode() -> None:
client = _build_client(trace_provider)
try:
response = client.get("/traces/trace-1?format=text&ancestor_depth=1")
response = client.get("/traces/trace-1?format=text&ancestor_depth=1&view=single")
finally:
client.close()
@@ -379,21 +388,21 @@ def test_trace_endpoint_renders_ancestors_in_text_mode() -> None:
"child_ids:\n"
"\n"
"==============================\n"
"trace_id: root-1\n"
"trace context: root, root-1\n"
"\n"
"step: process\n"
"root message\n"
"\n"
"\n"
"==============================\n"
"trace_id: parent-1\n"
"trace context: parent, parent-1\n"
"\n"
"step: process\n"
"parent message\n"
"\n"
"\n"
"==============================\n"
"trace_id: trace-1\n"
"trace context: child, trace-1\n"
"\n"
"step: process\n"
"child message"
@@ -405,18 +414,21 @@ def test_trace_endpoint_preserves_ancestor_depth_in_html_links() -> None:
return TraceLogView(
trace_id="trace-1",
parent_id="parent-1",
alias="child",
child_ids=("child-1",),
records=(_trace_record(row_id=1, level="INFO", message="loaded prices"),),
ancestors=(
TraceLogView(
trace_id="root-1",
parent_id=None,
alias="root",
child_ids=("parent-1",),
records=(_trace_record(row_id=2, level="INFO", message="root info"),),
),
TraceLogView(
trace_id="parent-1",
parent_id="root-1",
alias="parent",
child_ids=("trace-1",),
records=(_trace_record(row_id=3, level="WARNING", message="parent warning"),),
),
@@ -425,19 +437,102 @@ def test_trace_endpoint_preserves_ancestor_depth_in_html_links() -> None:
client = _build_client(trace_provider)
try:
response = client.get("/traces/trace-1?format=html&attrs_json=true&ancestor_depth=all")
response = client.get("/traces/trace-1?format=html&attrs_json=true&ancestor_depth=all&view=single")
finally:
client.close()
assert response.status_code == 200
assert 'href="/traces/trace-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true&amp;ancestor_depth=all"' in response.text
assert 'href="/traces/root-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true&amp;ancestor_depth=all"' in response.text
assert 'href="/traces/parent-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true&amp;ancestor_depth=all"' in response.text
assert 'href="/traces/trace-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true&amp;view=single&amp;ancestor_depth=all"' in response.text
assert 'href="/traces/root-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true&amp;view=single&amp;ancestor_depth=all"' in response.text
assert 'href="/traces/parent-1?format=html&amp;levels=ERROR%2CWARNING%2CINFO&amp;attrs_json=true&amp;view=single&amp;ancestor_depth=all"' in response.text
assert response.text.index("root info") < response.text.index("parent warning") < response.text.index("loaded prices")
assert "root info" in response.text
assert "parent warning" in response.text
def test_trace_endpoint_returns_full_tree_by_default() -> None:
async def trace_provider(trace_id: str, request: TraceQueryRequest) -> TraceLogView:
assert trace_id == "child-1"
assert request.view == "tree"
assert request.ancestor_depth is None
return TraceLogView(
trace_id="root-1",
parent_id=None,
alias="root",
child_ids=("parent-1",),
records=(_trace_record(row_id=1, level="INFO", message="root message"),),
descendants=(
TraceLogView(
trace_id="parent-1",
parent_id="root-1",
alias="parent",
child_ids=("child-1", "sibling-1"),
records=(_trace_record(row_id=2, level="WARNING", message="parent message"),),
descendants=(
TraceLogView(
trace_id="child-1",
parent_id="parent-1",
alias="child",
child_ids=(),
records=(_trace_record(row_id=3, level="ERROR", message="child message"),),
),
TraceLogView(
trace_id="sibling-1",
parent_id="parent-1",
alias="sibling",
child_ids=(),
records=(_trace_record(row_id=4, level="INFO", message="sibling message"),),
),
),
),
),
requested_trace_id="child-1",
)
client = _build_client(trace_provider)
try:
response = client.get("/traces/child-1?format=text")
finally:
client.close()
assert response.status_code == 200
assert "requested_trace_id: child-1" in response.text
assert response.text.index("root message") < response.text.index("parent message") < response.text.index("child message") < response.text.index("sibling message")
def test_trace_endpoint_supports_single_view_opt_in() -> None:
async def trace_provider(_trace_id: str, request: TraceQueryRequest) -> TraceLogView:
assert request.view == "single"
assert request.ancestor_depth == 0
return TraceLogView(
trace_id="trace-1",
parent_id="root-trace",
alias="single_trace",
child_ids=("child-1",),
records=(_trace_record(row_id=1, level="ERROR", message="only current trace"),),
)
client = _build_client(trace_provider)
try:
response = client.get("/traces/trace-1?format=text&view=single")
finally:
client.close()
assert response.status_code == 200
assert response.text == (
"trace_id: trace-1\n"
"parent_id: root-trace\n"
"child_ids:\n"
" - child-1\n"
"\n"
"==============================\n"
"trace context: single_trace, trace-1\n"
"\n"
"step: process\n"
"only current trace"
)
def test_trace_endpoint_validates_query_params() -> None:
client = _build_client(lambda _trace_id, _request: None)
try:
@@ -445,6 +540,7 @@ def test_trace_endpoint_validates_query_params() -> None:
invalid_format = client.get("/traces/trace-1?format=xml")
invalid_ancestor_depth = client.get("/traces/trace-1?ancestor_depth=-1")
invalid_ancestor_type = client.get("/traces/trace-1?ancestor_depth=up")
invalid_view = client.get("/traces/trace-1?view=graph")
finally:
client.close()
@@ -462,12 +558,18 @@ def test_trace_endpoint_validates_query_params() -> None:
"status": "error",
"detail": "invalid ancestor depth query parameter: ancestor_depth=up",
}
assert invalid_view.status_code == 400
assert invalid_view.json() == {
"status": "error",
"detail": "unsupported trace view: graph",
}
def test_runtime_trace_logs_uses_configured_reader(monkeypatch) -> None:
expected = TraceLogView(
trace_id="trace-1",
parent_id="root",
alias="runtime",
child_ids=("child-1",),
records=(_trace_record(row_id=1, level="ERROR", message="boom"),),
)
@@ -478,16 +580,19 @@ def test_runtime_trace_logs_uses_configured_reader(monkeypatch) -> None:
trace_id: str,
levels: tuple[str, ...],
ancestor_depth: int | None = 0,
*,
view: str = "tree",
) -> TraceLogView | None:
assert trace_id == "trace-1"
assert levels == ("ERROR",)
assert ancestor_depth is None
assert view == "tree"
return expected
monkeypatch.setattr(runtime_module, "build_trace_log_reader", lambda _transport: StubReader())
runtime = RuntimeManager()
result = asyncio.run(runtime.trace_logs("trace-1", TraceQueryRequest(levels=("ERROR",), ancestor_depth=None)))
result = asyncio.run(runtime.trace_logs("trace-1", TraceQueryRequest(levels=("ERROR",), ancestor_depth=None, view="tree")))
assert result == expected
@@ -504,9 +609,9 @@ def test_mysql_trace_log_reader_maps_db_rows() -> None:
def fetchone(self) -> dict[str, object] | None:
if self.executed[-1][1] == ("trace-1",):
return {"parent_id": "root-77"}
return {"parent_id": "root-77", "alias": "single"}
if self.executed[-1][1] == ("root-77",):
return {"parent_id": None}
return {"parent_id": None, "alias": "root"}
return None
def fetchall(self) -> list[dict[str, object]]:
@@ -554,11 +659,12 @@ def test_mysql_trace_log_reader_maps_db_rows() -> None:
factory = FakeConnectionFactory()
reader = MySqlTraceLogReader(factory) # type: ignore[arg-type]
view = reader.read_trace("trace-1", ("ERROR", "WARNING"))
view = reader.read_trace("trace-1", ("ERROR", "WARNING"), view="single")
assert view == TraceLogView(
trace_id="trace-1",
parent_id="root-77",
alias="single",
child_ids=("child-1", "child-2"),
records=(
TraceLogRecord(
@@ -591,11 +697,11 @@ def test_mysql_trace_log_reader_loads_requested_ancestors() -> None:
def fetchone(self) -> dict[str, object] | None:
if self.executed[-1][1] == ("trace-1",):
return {"parent_id": "parent-1"}
return {"parent_id": "parent-1", "alias": "child"}
if self.executed[-1][1] == ("parent-1",):
return {"parent_id": "root-1"}
return {"parent_id": "root-1", "alias": "parent"}
if self.executed[-1][1] == ("root-1",):
return {"parent_id": None}
return {"parent_id": None, "alias": "root"}
return None
def fetchall(self) -> list[dict[str, object]]:
@@ -651,17 +757,101 @@ def test_mysql_trace_log_reader_loads_requested_ancestors() -> None:
factory = FakeConnectionFactory()
reader = MySqlTraceLogReader(factory) # type: ignore[arg-type]
view = reader.read_trace("trace-1", ("ERROR",), 1)
view = reader.read_trace("trace-1", ("ERROR",), 1, view="single")
assert view is not None
assert view.trace_id == "trace-1"
assert view.parent_id == "parent-1"
assert view.alias == "child"
assert len(view.ancestors) == 1
assert view.ancestors[0].trace_id == "parent-1"
assert view.ancestors[0].parent_id == "root-1"
assert view.ancestors[0].alias == "parent"
assert view.ancestors[0].child_ids == ("trace-1",)
def test_mysql_trace_log_reader_tree_view_returns_root_subtree() -> None:
class FakeCursor:
def __init__(self) -> None:
self.executed: list[tuple[str, tuple[object, ...]]] = []
self._current_query = ""
def execute(self, query: str, params: tuple[object, ...]) -> None:
self.executed.append((query, params))
self._current_query = query
def fetchone(self) -> dict[str, object] | None:
trace_id = self.executed[-1][1][0]
mapping = {
"child-1": {"parent_id": "parent-1", "alias": "child"},
"parent-1": {"parent_id": "root-1", "alias": "parent"},
"root-1": {"parent_id": None, "alias": "root"},
"sibling-1": {"parent_id": "parent-1", "alias": "sibling"},
}
return mapping.get(trace_id)
def fetchall(self) -> list[dict[str, object]]:
if "WHERE parent_id = %s" in self._current_query:
parent_id = self.executed[-1][1][0]
if parent_id == "root-1":
return [{"trace_id": "parent-1"}]
if parent_id == "parent-1":
return [{"trace_id": "child-1"}, {"trace_id": "sibling-1"}]
return []
trace_id = self.executed[-1][1][0]
return [
{
"id": 8,
"trace_id": trace_id,
"event_time": datetime(2026, 4, 28, 10, 11, 12, tzinfo=timezone.utc),
"step": "parse",
"status": "failed",
"level": "ERROR",
"message": f"broken:{trace_id}",
"attrs_json": '{"attempt":1}',
}
]
def __enter__(self) -> FakeCursor:
return self
def __exit__(self, exc_type, exc, tb) -> None:
return None
class FakeConnection:
def __init__(self, cursor: FakeCursor) -> None:
self._cursor = cursor
def cursor(self) -> FakeCursor:
return self._cursor
def __enter__(self) -> FakeConnection:
return self
def __exit__(self, exc_type, exc, tb) -> None:
return None
class FakeConnectionFactory:
def __init__(self) -> None:
self.cursor = FakeCursor()
def connect(self) -> FakeConnection:
return FakeConnection(self.cursor)
factory = FakeConnectionFactory()
reader = MySqlTraceLogReader(factory) # type: ignore[arg-type]
view = reader.read_trace("child-1", ("ERROR",), None, view="tree")
assert view is not None
assert view.trace_id == "root-1"
assert view.alias == "root"
assert view.requested_trace_id == "child-1"
assert tuple(child.trace_id for child in view.descendants) == ("parent-1",)
assert view.descendants[0].alias == "parent"
assert tuple(child.trace_id for child in view.descendants[0].descendants) == ("child-1", "sibling-1")
def test_mysql_trace_log_reader_orders_ancestors_root_first() -> None:
class FakeCursor:
def __init__(self) -> None:
@@ -674,11 +864,11 @@ def test_mysql_trace_log_reader_orders_ancestors_root_first() -> None:
def fetchone(self) -> dict[str, object] | None:
if self.executed[-1][1] == ("trace-1",):
return {"parent_id": "parent-1"}
return {"parent_id": "parent-1", "alias": "child"}
if self.executed[-1][1] == ("parent-1",):
return {"parent_id": "root-1"}
return {"parent_id": "root-1", "alias": "parent"}
if self.executed[-1][1] == ("root-1",):
return {"parent_id": None}
return {"parent_id": None, "alias": "root"}
return None
def fetchall(self) -> list[dict[str, object]]:
@@ -732,7 +922,8 @@ def test_mysql_trace_log_reader_orders_ancestors_root_first() -> None:
factory = FakeConnectionFactory()
reader = MySqlTraceLogReader(factory) # type: ignore[arg-type]
view = reader.read_trace("trace-1", ("ERROR",), 2)
view = reader.read_trace("trace-1", ("ERROR",), 2, view="single")
assert view is not None
assert tuple(ancestor.trace_id for ancestor in view.ancestors) == ("root-1", "parent-1")
assert tuple(ancestor.alias for ancestor in view.ancestors) == ("root", "parent")