mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-21 21:15:13 +00:00
63 lines
2.5 KiB
Python
63 lines
2.5 KiB
Python
from fastapi import FastAPI
|
|
from opentelemetry import trace
|
|
from opentelemetry._logs import set_logger_provider
|
|
from opentelemetry.exporter.otlp.proto.grpc._log_exporter import OTLPLogExporter
|
|
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
|
|
from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
|
|
from opentelemetry.instrumentation.logging import LoggingInstrumentor
|
|
from opentelemetry.instrumentation.redis import RedisInstrumentor
|
|
from opentelemetry.instrumentation.sqlalchemy import SQLAlchemyInstrumentor
|
|
from opentelemetry.sdk._logs import LoggerProvider, LoggingHandler
|
|
from opentelemetry.sdk._logs._internal.export import BatchLogRecordProcessor
|
|
from opentelemetry.sdk.resources import Resource
|
|
from opentelemetry.sdk.trace import TracerProvider
|
|
from opentelemetry.sdk.trace.export import BatchSpanProcessor
|
|
|
|
from backend.common.log import log, request_id_filter
|
|
from backend.core.conf import settings
|
|
from backend.database.db import async_engine
|
|
from backend.database.redis import redis_client
|
|
|
|
|
|
def init_otel(app: FastAPI) -> None:
|
|
"""
|
|
初始化 OpenTelemetry
|
|
|
|
:param app: FastAPI 应用实例
|
|
:return:
|
|
"""
|
|
from backend import __version__
|
|
|
|
resource = Resource(
|
|
attributes={
|
|
'service.name': settings.GRAFANA_APP_NAME,
|
|
'service.version': __version__,
|
|
'deployment.environment': settings.ENVIRONMENT,
|
|
},
|
|
)
|
|
|
|
tracer_provider = TracerProvider(resource=resource)
|
|
trace.set_tracer_provider(tracer_provider)
|
|
|
|
span_exporter = OTLPSpanExporter(endpoint=settings.GRAFANA_OTLP_GRPC_ENDPOINT, insecure=True)
|
|
tracer_provider.add_span_processor(BatchSpanProcessor(span_exporter))
|
|
|
|
logger_provider = LoggerProvider(resource=resource)
|
|
set_logger_provider(logger_provider)
|
|
|
|
log_exporter = OTLPLogExporter(endpoint=settings.GRAFANA_OTLP_GRPC_ENDPOINT, insecure=True)
|
|
logger_provider.add_log_record_processor(BatchLogRecordProcessor(log_exporter))
|
|
|
|
otel_logging_handler = LoggingHandler(logger_provider=logger_provider)
|
|
log.add( # type: ignore
|
|
otel_logging_handler,
|
|
level=settings.LOG_STD_LEVEL,
|
|
format=settings.LOG_FORMAT,
|
|
filter=lambda record: request_id_filter(record),
|
|
)
|
|
|
|
LoggingInstrumentor().instrument(set_logging_format=True)
|
|
SQLAlchemyInstrumentor().instrument(engine=async_engine.sync_engine)
|
|
RedisInstrumentor().instrument_client(redis_client) # type: ignore
|
|
FastAPIInstrumentor.instrument_app(app, tracer_provider=tracer_provider)
|