mirror of
https://github.com/fastapi-practices/fastapi_best_architecture.git
synced 2026-03-13 09:31:31 +08:00
4bc5ba53e6
* Add the Grafana observability suite * Update configs * Update docker script directory structure * Fix the otel trace id * Add grafana ini * Fix some configs and loguru integration * Add the celery grafana * Update Grafana dashboards * Update configs * Fix issues with the panel * Update grafana configs * Update grafana dashboards * Optimized panel styles * Add sqlalchemy traces * Fix the CORS * Update the grafana query and config * Update grafana status is off by default
32 lines
931 B
Python
32 lines
931 B
Python
from typing import Any
|
|
|
|
from opentelemetry import trace
|
|
from starlette.requests import Request
|
|
from starlette_context.plugins import Plugin
|
|
|
|
from backend.common.context import ctx
|
|
from backend.core.conf import settings
|
|
|
|
|
|
def get_request_trace_id() -> str:
|
|
"""从上下文中获取追踪 ID"""
|
|
if ctx.exists():
|
|
return ctx.get(settings.TRACE_ID_REQUEST_HEADER_KEY, settings.TRACE_ID_LOG_DEFAULT_VALUE)
|
|
return settings.TRACE_ID_LOG_DEFAULT_VALUE
|
|
|
|
|
|
class OtelTraceIdPlugin(Plugin):
|
|
"""OpenTelemetry Trace ID 插件"""
|
|
|
|
key = settings.TRACE_ID_REQUEST_HEADER_KEY
|
|
|
|
async def process_request(self, request: Request) -> Any:
|
|
"""从 OpenTelemetry span 中提取 trace_id"""
|
|
span = trace.get_current_span()
|
|
span_ctx = span.get_span_context()
|
|
|
|
if span_ctx.is_valid:
|
|
return trace.format_trace_id(span_ctx.trace_id)
|
|
|
|
return settings.TRACE_ID_LOG_DEFAULT_VALUE
|