mirror of
https://github.com/hatchet-dev/hatchet.git
synced 2025-12-30 05:09:44 -06:00
8e80faf2d6
* api changes * doc changes * move docs * generated * generate * pkg * backmerge main * revert to main * revert main * race? * remove go tests
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
import os
|
|
from typing import cast
|
|
|
|
from opentelemetry.exporter.otlp.proto.http.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
|
|
from opentelemetry.trace import NoOpTracerProvider
|
|
|
|
trace_provider: TracerProvider | NoOpTracerProvider
|
|
|
|
if os.getenv("CI", "false") == "true":
|
|
trace_provider = NoOpTracerProvider()
|
|
else:
|
|
resource = Resource(
|
|
attributes={
|
|
SERVICE_NAME: os.getenv("HATCHET_CLIENT_OTEL_SERVICE_NAME", "test-service")
|
|
}
|
|
)
|
|
|
|
headers = dict(
|
|
[
|
|
cast(
|
|
tuple[str, str],
|
|
tuple(
|
|
os.getenv(
|
|
"HATCHET_CLIENT_OTEL_EXPORTER_OTLP_HEADERS", "foo=bar"
|
|
).split("=")
|
|
),
|
|
)
|
|
]
|
|
)
|
|
|
|
processor = BatchSpanProcessor(
|
|
OTLPSpanExporter(
|
|
endpoint=os.getenv(
|
|
"HATCHET_CLIENT_OTEL_EXPORTER_OTLP_ENDPOINT", "http://localhost:4317"
|
|
),
|
|
headers=headers,
|
|
),
|
|
)
|
|
|
|
trace_provider = TracerProvider(resource=resource)
|
|
|
|
trace_provider.add_span_processor(processor)
|