-
Notifications
You must be signed in to change notification settings - Fork 30
/
common.py
34 lines (27 loc) · 1.11 KB
/
common.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import os
import string
from opentelemetry import trace
from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter
from opentelemetry.sdk.resources import SERVICE_NAME, Resource
from opentelemetry.sdk.trace import TracerProvider
from opentelemetry.sdk.trace.export import BatchSpanProcessor
def get_otlp_exporter():
ls_access_token = os.environ.get("LS_ACCESS_TOKEN")
return OTLPSpanExporter(
endpoint="ingest.lightstep.com:443",
headers=(("lightstep-access-token", ls_access_token),),
)
def get_tracer():
span_exporter = get_otlp_exporter()
provider = TracerProvider()
if not os.environ.get("OTEL_RESOURCE_ATTRIBUTES"):
# Service name is required for most backends
resource = Resource(attributes={
SERVICE_NAME: "test-py-manual-otlp"
})
provider = TracerProvider(resource=resource)
print("Using default service name")
processor = BatchSpanProcessor(span_exporter)
provider.add_span_processor(processor)
trace.set_tracer_provider(provider)
return trace.get_tracer(__name__)