fastapi - 如何为 FastAPI 路由设置 OpenTelemetry 跨度属性？

Question

将标签添加到跟踪的跨度对于稍后分析跟踪数据并按所需标签对其进行切片和切块非常有用。

阅读OpenTelemetry 文档后，我想不出一种将自定义标签添加到跨度的方法。

这是我的示例 FastAPI 应用程序，已经使用 OpenTelemetry 进行了检测：

"""main.py"""
from typing import Dict

import fastapi

from opentelemetry import trace
from opentelemetry.sdk.trace.export import (
    ConsoleSpanExporter,
    SimpleExportSpanProcessor,
)
from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
from opentelemetry.sdk.trace import TracerProvider


trace.set_tracer_provider(TracerProvider())
trace.get_tracer_provider().add_span_processor(SimpleExportSpanProcessor(ConsoleSpanExporter()))


app = fastapi.FastAPI()


@app.get("/user/{id}")
async def get_user(id: int) -> Dict[str, str]:
    """Test endpoint."""
    return {"message": "hello user!"}


FastAPIInstrumentor.instrument_app(app)

你可以运行它uvicorn main:app --reload

如何将用户 ID 添加到跨度？

Answer 1

在阅读了 ASGI 工具的 OpenTelemetryMiddleware 的源代码（此处）后，我意识到您可以简单地获取当前跨度，设置标签（或属性），然后将返回当前跨度及其所有属性。

@app.get("/user/{id}")
async def get_user(id: int) -> Dict[str, str]:
    """Test endpoint."""
    # Instrument the user id as a span tag
    current_span = trace.get_current_span()
    if current_span:
        current_span.set_attribute("user_id", id)

    return {"message": "hello user!"}