Skip to content

Commit

Permalink
fix: missing files for small refactor: env_vars to env
Browse files Browse the repository at this point in the history
  • Loading branch information
nampereira committed Jan 9, 2024
1 parent 47a98a9 commit aa891ad
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 5 deletions.
2 changes: 1 addition & 1 deletion arena/utils/__init__.py
@@ -1,4 +1,4 @@
from .utils import Utils
from .cmd_interpreter import ArenaCmdInterpreter
from .arena_telemetry import ArenaTelemetry
from .program_stats import ProgramStats
from .program_info import ProgramRunInfo
9 changes: 5 additions & 4 deletions arena/utils/arena_telemetry.py
Expand Up @@ -13,12 +13,12 @@
from opentelemetry.trace import NoOpTracerProvider, Status, StatusCode
from opentelemetry.sdk.trace.export import BatchSpanProcessor, ConsoleSpanExporter, SpanExporter, SpanExportResult

from ..env_vars import (
from ..env import (
ARENA_TELEMETRY,
OTLP_ENDPOINT
OTLP_ENDPOINT,
ENV_DEFAULTS
)

OTLP_ENDPOINT_DFT = "http://localhost:4317"
TRACE_TOPIC_DFT = "realm/ns/scene/t/traces"

class MQTTSpanExporter(SpanExporter):
Expand Down Expand Up @@ -56,6 +56,7 @@ def force_flush(self, timeout_millis: int = 30000) -> bool:
def shutdown(self) -> None:
print("shutdown exporter!")
pass

class ArenaTelemetry():

parent_span: Span = None
Expand All @@ -70,7 +71,7 @@ def __init__(self, name=sys.argv[0], id=None):
})

env_telemetry = os.environ.get(ARENA_TELEMETRY, 'None')
otlp_endpoint = os.environ.get(OTLP_ENDPOINT, OTLP_ENDPOINT_DFT)
otlp_endpoint = os.environ.get(OTLP_ENDPOINT, ENV_DEFAULTS.get(OTLP_ENDPOINT))
tel_exporters = {
'otlp': lambda: OTLPSpanExporter(otlp_endpoint, insecure=True),
'mqtt': lambda: MQTTSpanExporter(),
Expand Down

0 comments on commit aa891ad

Please sign in to comment.