Marmelator/pyinfra
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 36 Wiki Activity

refactor: update tracing info for Azure Monitor

Browse Source
This commit is contained in:
Jonathan Kössler 2024-06-11 14:31:06 +02:00
parent 7dbcdf1650
commit 9661d75d8a
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
pyinfra/examples.py
View File

@ -2,7 +2,6 @@ from dynaconf import Dynaconf
from fastapi import FastAPI
from kn_utils.logging import logger
from azure.monitor.opentelemetry import configure_azure_monitor
from opentelemetry import trace
from pyinfra.config.loader import get_pyinfra_validators, validate_settings
from pyinfra.queue.callback import Callback
@ -32,23 +31,23 @@ def start_standard_queue_consumer(
"""
validate_settings(settings, get_pyinfra_validators())
logger.info(f"Starting webserver and queue consumer...")
logger.info("Starting webserver and queue consumer...")
app = app or FastAPI()
queue_manager = QueueManager(settings)
if settings.metrics.prometheus.enabled:
logger.info(f"Prometheus metrics enabled.")
logger.info("Prometheus metrics enabled.")
app = add_prometheus_endpoint(app)
callback = make_prometheus_processing_time_decorator_from_settings(settings)(callback)
if settings.tracing.opentelemetry.enabled:
logger.info(f"OpenTelemetry tracing enabled.")
logger.info("OpenTelemetry tracing enabled.")
if settings.tracing.opentelemetry.azure_monitoring:
# Configure OpenTelemetry to use Azure Monitor with the
# APPLICATIONINSIGHTS_CONNECTION_STRING environment variable.
logger.info("Started Azure Monitoring.")
logger.info("Azure Monitor tracing enabled.")
configure_azure_monitor()
else:
setup_trace(settings)