This commit is contained in:
0Xiao0
2025-02-06 10:39:47 +08:00
parent 6ed4d1b2e3
commit 7faa37e22c

37
main.py
View File

@ -14,26 +14,25 @@ class Main():
def init_log_handlers(self): def init_log_handlers(self):
with open(".env.yaml", "r") as file: with open(".env.yaml", "r") as file:
config = yaml.safe_load(file) config = yaml.safe_load(file)
if config["log"] is None: try:
log_handler_confg = config["log"]["handlers"]
log_handlers = [logging.StreamHandler()]
for handler_config in log_handler_confg:
if handler_config["type"] == "loki":
log_handlers.append(LokiLoggerHandler(
url=handler_config["url"],
additional_headers={"X-Odin-Auth": handler_config["x-odin-auth"]},
labels=handler_config["labels"],
label_keys={},
timeout=10,
enable_structured_loki_metadata=True,
loki_metadata={"service": "user-service", "version": "1.0.0"},
insecure_ssl_verify=False
))
logging.basicConfig(level=logging.DEBUG, handlers=log_handlers)
logging.info("logging init finish")
except KeyError:
return return
if config["log"]["handlers"] is None:
return
log_handler_confg = config["log"]["handlers"]
log_handlers = [logging.StreamHandler()]
for handler_config in log_handler_confg:
if handler_config["type"] == "loki":
log_handlers.append(LokiLoggerHandler(
url=handler_config["url"],
additional_headers={"X-Odin-Auth": handler_config["x-odin-auth"]},
labels=handler_config["labels"],
label_keys={},
timeout=10,
enable_structured_loki_metadata=True,
loki_metadata={"service": "user-service", "version": "1.0.0"},
insecure_ssl_verify=False
))
logging.basicConfig(level=logging.DEBUG, handlers=log_handlers)
logging.info("logging init finish")
def __init__(self) -> None: def __init__(self) -> None:
self.init_log_handlers() self.init_log_handlers()