Skip to content
Empty file added ddtrace/aiobs/__init__.py
Empty file.
Empty file.
Empty file.
25 changes: 25 additions & 0 deletions ddtrace/aiobs/integrations/ray/tracer.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
import os
os.environ["DD_TRACE_OTEL_ENABLED"] = "true"

from opentelemetry.trace import set_tracer_provider
from ddtrace.opentelemetry import TracerProvider
from ddtrace.trace import tracer
from ddtrace.trace import TraceFilter


class RayTraceFilter(TraceFilter):
def process_trace(self, trace):
for span in trace:
span.span_type = "ray." + span.name
span.name = "ray.job"
span.service = os.environ.get("DD_SERVICE", "unspecified-ray-job")
span.set_metric("_dd.djm.enabled", 1)
span.set_metric("_dd.filter.kept", 1)
span.set_metric("_dd.measured", 1)
span.set_metric("_sampling_priority_v1", 2)
return trace


def setup_tracing() -> None:
tracer.configure(trace_processors=[RayTraceFilter()])
set_tracer_provider(TracerProvider())
Loading