meshagent-otel 0.1.0__py3-none-any.whl

meshagent/otel/__init__.py

@@ -0,0 +1,130 @@
+from opentelemetry.sdk.resources import SERVICE_NAME, Resource
+from opentelemetry import trace
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import BatchSpanProcessor, ConsoleSpanExporter
+
+from opentelemetry import metrics
+from opentelemetry.exporter.otlp.proto.http.metric_exporter import OTLPMetricExporter
+from opentelemetry.sdk._logs.export import ConsoleLogExporter
+from opentelemetry.sdk.metrics import MeterProvider
+from opentelemetry.sdk.metrics.export import (
+    PeriodicExportingMetricReader,
+    ConsoleMetricExporter,
+)
+from opentelemetry import _logs
+from opentelemetry.sdk._logs import LoggerProvider, LoggingHandler
+from opentelemetry.sdk._logs.export import BatchLogRecordProcessor
+from opentelemetry.exporter.otlp.proto.http._log_exporter import OTLPLogExporter
+from opentelemetry.sdk.metrics.export import AggregationTemporality
+from opentelemetry.sdk.metrics import Counter, Histogram
+import logging
+
+import os
+
+from .version import __version__
+
+
+def _call_once(fn):
+    called = False
+    result = None
+
+    def wrapper(*args, **kwargs):
+        nonlocal called, result
+        if not called:
+            result = fn(*args, **kwargs)
+            called = True
+        return result
+
+    return wrapper
+
+
+@_call_once
+def otel_config(*, level=logging.INFO, service_name: str = "meshagent-service"):
+    attributes = {
+        SERVICE_NAME: service_name,
+    }
+
+    if os.getenv("MESHAGENT_PROJECT_ID") is not None:
+        attributes["project"] = os.getenv("MESHAGENT_PROJECT_ID")
+
+    if os.getenv("MESHAGENT_SESSION_ID") is not None:
+        attributes["session"] = os.getenv("MESHAGENT_SESSION_ID")
+
+    if os.getenv("MESHAGENT_ROOM") is not None:
+        attributes["room"] = os.getenv("MESHAGENT_ROOM")
+
+    resource = Resource.create(attributes=attributes)
+
+    logger_provider = None
+    tracer_provider = None
+    meter_provider = None
+
+    add_console_exporters = False
+
+    otel_endpoint = os.getenv("OTEL_ENDPOINT")
+
+    if otel_endpoint is not None:
+        otel_logs_endpoint = otel_endpoint + "/v1/logs"
+        otel_traces_endpoint = otel_endpoint + "/v1/traces"
+        otel_metrics_endpoint = otel_endpoint + "/v1/metrics"
+
+        if otel_logs_endpoint is not None:
+            logs_exporter = OTLPLogExporter(
+                endpoint=otel_logs_endpoint,
+            )
+            logger_provider = LoggerProvider(resource=resource)
+            _logs.set_logger_provider(logger_provider)
+
+            logger_provider.add_log_record_processor(
+                BatchLogRecordProcessor(logs_exporter)
+            )
+
+            if add_console_exporters:
+                logger_provider.add_log_record_processor(
+                    BatchLogRecordProcessor(ConsoleLogExporter())
+                )
+
+        if otel_traces_endpoint is not None:
+            tracer_provider = TracerProvider(resource=resource)
+            processor = BatchSpanProcessor(
+                OTLPSpanExporter(endpoint=otel_traces_endpoint)
+            )
+            tracer_provider.add_span_processor(processor)
+            if add_console_exporters:
+                tracer_provider.add_span_processor(
+                    BatchSpanProcessor(ConsoleSpanExporter())
+                )
+            trace.set_tracer_provider(tracer_provider)
+
+        if otel_metrics_endpoint is not None:
+            reader = PeriodicExportingMetricReader(
+                exporter=OTLPMetricExporter(
+                    endpoint=otel_metrics_endpoint,
+                    preferred_temporality={
+                        Counter: AggregationTemporality.DELTA,
+                        Histogram: AggregationTemporality.DELTA,
+                    },
+                ),
+                export_interval_millis=1000,
+            )
+
+            readers = [
+                reader,
+            ]
+            if add_console_exporters:
+                readers.append(PeriodicExportingMetricReader(ConsoleMetricExporter()))
+
+            meter_provider = MeterProvider(resource=resource, metric_readers=readers)
+            metrics.set_meter_provider(meter_provider)
+
+    if logger_provider is not None:
+        logging_handler = LoggingHandler(level=level, logger_provider=logger_provider)
+        root = logging.getLogger()
+        root.setLevel(level)
+        root.addHandler(logging_handler)
+    else:
+        logging.basicConfig(level=level)
+
+
+__all__ = [otel_config, __version__]

meshagent/otel/version.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

meshagent_otel-0.1.0.dist-info/METADATA

@@ -0,0 +1,27 @@
+Metadata-Version: 2.4
+Name: meshagent-otel
+Version: 0.1.0
+Summary: CLI for Meshagent
+License-Expression: Apache-2.0
+Project-URL: Documentation, https://docs.meshagent.com
+Project-URL: Website, https://www.meshagent.com
+Project-URL: Source, https://www.meshagent.com
+Requires-Python: >=3.12
+Description-Content-Type: text/markdown
+Requires-Dist: opentelemetry-distro~=0.54b1
+Requires-Dist: opentelemetry-exporter-otlp-proto-http~=1.33
+
+## MeshAgent Otel
+
+The ``meshagent.otel`` package is a streamlined way to enable otel in your meshagent projects.
+
+Check out the [CLI Quickstart](https://docs.meshagent.com/cli/getting_started) for more details.
+
+---
+### Learn more about MeshAgent on our website or check out the docs for additional examples!
+
+**Website**: [www.meshagent.com](https://www.meshagent.com/)
+
+**Documentation**: [docs.meshagent.com](https://docs.meshagent.com/)
+
+---

meshagent_otel-0.1.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+meshagent/otel/__init__.py,sha256=KfrR3uku7nB0AY8dscq4CGoxDQYZ2OtSFzcnt90x1gs,4489
+meshagent/otel/version.py,sha256=kUR5RAFc7HCeiqdlX36dZOHkUI5wI6V_43RpEcD8b-0,22
+meshagent_otel-0.1.0.dist-info/METADATA,sha256=Qfu-U3gmH-EZzBVtYe0ew85VPxHSK0PT6_X0-OZCjOA,882
+meshagent_otel-0.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+meshagent_otel-0.1.0.dist-info/top_level.txt,sha256=GlcXnHtRP6m7zlG3Df04M35OsHtNXy_DY09oFwWrH74,10
+meshagent_otel-0.1.0.dist-info/RECORD,,

meshagent_otel-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

meshagent_otel-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+meshagent