Skip to content

Commit f17c359

Browse files
authored
adding endpoint fix (#2270)
1 parent dba0002 commit f17c359

File tree

1 file changed

+16
-6
lines changed

1 file changed

+16
-6
lines changed

main/telemetry.py

Lines changed: 16 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,10 @@
55

66
from django.conf import settings
77
from opentelemetry import trace
8-
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
8+
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import (
9+
OTLPSpanExporter as OTLPSpanExporterGrpc,
10+
)
11+
from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
912
from opentelemetry.instrumentation.celery import CeleryInstrumentor
1013
from opentelemetry.instrumentation.django import DjangoInstrumentor
1114
from opentelemetry.instrumentation.psycopg import PsycopgInstrumentor
@@ -56,11 +59,18 @@ def configure_opentelemetry() -> Optional[TracerProvider]:
5659
headers = {}
5760

5861
try:
59-
otlp_exporter = OTLPSpanExporter(
60-
endpoint=otlp_endpoint,
61-
headers=headers,
62-
insecure=getattr(settings, "OPENTELEMETRY_INSECURE", True),
63-
)
62+
use_grpc = getattr(settings, "OPENTELEMETRY_USE_GRPC", False)
63+
if use_grpc:
64+
otlp_exporter = OTLPSpanExporterGrpc(
65+
endpoint=otlp_endpoint,
66+
headers=headers,
67+
insecure=getattr(settings, "OPENTELEMETRY_INSECURE", True),
68+
)
69+
else:
70+
otlp_exporter = OTLPSpanExporter(
71+
endpoint=otlp_endpoint,
72+
headers=headers,
73+
)
6474

6575
tracer_provider.add_span_processor(
6676
BatchSpanProcessor(

0 commit comments

Comments
 (0)