34 lines
1.5 KiB
Python
34 lines
1.5 KiB
Python
from opentelemetry import metrics, trace
|
|
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import \
|
|
OTLPSpanExporter
|
|
from opentelemetry.exporter.otlp.proto.http.metric_exporter import \
|
|
OTLPMetricExporter
|
|
from opentelemetry.instrumentation.django import DjangoInstrumentor
|
|
from opentelemetry.instrumentation.celery import CeleryInstrumentor
|
|
from opentelemetry.sdk.metrics import MeterProvider
|
|
from opentelemetry.sdk.metrics.export import PeriodicExportingMetricReader
|
|
from opentelemetry.sdk.resources import SERVICE_NAME, Resource
|
|
from opentelemetry.sdk.trace import TracerProvider
|
|
from opentelemetry.sdk.trace.export import BatchSpanProcessor
|
|
|
|
|
|
def setup_tracing(tracer_endpoint, service_name):
|
|
# Instrument Django
|
|
DjangoInstrumentor().instrument()
|
|
CeleryInstrumentor().instrument()
|
|
# Set up a tracer provider
|
|
span_exporter = OTLPSpanExporter(endpoint=f"{tracer_endpoint}/v1/traces")
|
|
processor = BatchSpanProcessor(span_exporter=span_exporter)
|
|
|
|
attributes = {SERVICE_NAME: service_name}
|
|
resource = Resource(attributes=attributes)
|
|
trace_provider = TracerProvider(resource=resource)
|
|
trace_provider.add_span_processor(span_processor=processor)
|
|
trace.set_tracer_provider(tracer_provider=trace_provider)
|
|
|
|
reader = PeriodicExportingMetricReader(
|
|
OTLPMetricExporter(endpoint=f"{tracer_endpoint}/v1/metrics"))
|
|
meter_provider = MeterProvider(resource=resource, metric_readers=[reader])
|
|
metrics.set_meter_provider(meter_provider=meter_provider)
|
|
|
|
|