File size: 2,501 Bytes
ef1ad9e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import logging
from datetime import datetime, timezone
from app.config.env import env
from pythonjsonlogger import jsonlogger
# --- azure-monitor library imports ---
from azure.monitor.opentelemetry import configure_azure_monitor
from opentelemetry import trace
from opentelemetry.trace import (
get_tracer_provider,
)
from opencensus.ext.azure.log_exporter import AzureLogHandler
# ---
# connection string for azure monitor
APPLICATIONINSIGHTS_CONNECTION_STRING = env.APPLICATIONINSIGHTS_CONNECTION_STRING
configure_azure_monitor(
connection_string=APPLICATIONINSIGHTS_CONNECTION_STRING)
class CustomJsonFormatter(jsonlogger.JsonFormatter):
"""Custom JSON formatter that adds timestamps and uppercases log levels."""
def add_fields(self, log_record, record, message_dict):
super(CustomJsonFormatter, self).add_fields(
log_record, record, message_dict)
# Add timestamp in UTC if not present
if not log_record.get("timestamp"):
now = datetime.now(timezone.utc).strftime("%Y-%m-%dT%H:%M:%S.%fZ")
log_record["timestamp"] = now
# Ensure log level is uppercase
if log_record.get("level"):
log_record["level"] = log_record["level"].upper()
else:
log_record["level"] = record.levelname.upper()
# Disable uvicorn access logger (prevents duplicate logging)
uvicorn_access = logging.getLogger("uvicorn.access")
uvicorn_access.disabled = True
uvicorn_access.propagate = False
# Create logger and tracer for user-service
logger = logging.getLogger("user_service")
tracer = trace.get_tracer(
"user_service", tracer_provider=get_tracer_provider())
def get_log_level(log_level):
"""Maps string log level to corresponding logging module level."""
valid_levels = {
"DEBUG": logging.DEBUG,
"INFO": logging.INFO,
"WARNING": logging.WARNING,
"ERROR": logging.ERROR,
}
# Default to ERROR for invalid levels
return valid_levels.get(log_level.upper(), logging.ERROR)
# Configure logging handler and formatter
logHandler = logging.StreamHandler()
formatter = CustomJsonFormatter("%(timestamp)s %(level)s %(name)s %(message)s")
logHandler.setFormatter(formatter)
logger.addHandler(logHandler)
logger.addHandler(AzureLogHandler(
connection_string=APPLICATIONINSIGHTS_CONNECTION_STRING)
)
logger.setLevel(logging.getLevelName(get_log_level(env.LOG_LEVEL)))
|