Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
26 changes: 15 additions & 11 deletions libs/executors/garf/executors/entrypoints/tracer.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,7 @@

import logging
import os
from typing import Optional

from opentelemetry import metrics, trace
from opentelemetry._logs import set_logger_provider
Expand All @@ -41,11 +42,18 @@
DEFAULT_SERVICE_NAME = 'garf'


def initialize_tracer():
resource = Resource.create(
{SERVICE_NAME: os.getenv('OTLP_SERVICE_NAME', DEFAULT_SERVICE_NAME)}
def _init_resource(otel_service_name: Optional[str] = None) -> Resource:
return Resource.create(
{
SERVICE_NAME: otel_service_name
or os.getenv('OTEL_SERVICE_NAME', DEFAULT_SERVICE_NAME)
}
)


def initialize_tracer(otel_service_name: Optional[str] = None):
resource = _init_resource(otel_service_name)

tracer_provider = TracerProvider(resource=resource)

if otel_endpoint := os.getenv('OTEL_EXPORTER_OTLP_ENDPOINT'):
Expand All @@ -71,10 +79,8 @@ def initialize_tracer():
trace.set_tracer_provider(tracer_provider)


def initialize_meter():
resource = Resource.create(
{SERVICE_NAME: os.getenv('OTLP_SERVICE_NAME', DEFAULT_SERVICE_NAME)}
)
def initialize_meter(otel_service_name: Optional[str] = None):
resource = _init_resource(otel_service_name)
meter_provider = MeterProvider(resource=resource)

if otel_endpoint := os.getenv('OTEL_EXPORTER_OTLP_ENDPOINT'):
Expand Down Expand Up @@ -106,10 +112,8 @@ def initialize_meter():
return meter_provider


def initialize_logger():
resource = Resource.create(
{SERVICE_NAME: os.getenv('OTLP_SERVICE_NAME', DEFAULT_SERVICE_NAME)}
)
def initialize_logger(otel_service_name: Optional[str] = None):
resource = _init_resource(otel_service_name)
logger_provider = LoggerProvider(resource=resource)
set_logger_provider(logger_provider)
if otel_endpoint := os.getenv('OTEL_EXPORTER_OTLP_ENDPOINT'):
Expand Down