add logging and retry to connect to the rabbitmq server
This commit is contained in:
+14
-8
@@ -3,11 +3,13 @@ import os
|
||||
import sys
|
||||
|
||||
from src.daemons.registry import DAEMONS
|
||||
from src.logging.logger import LogWriter
|
||||
from src.messaging.rabbitmq_client import rabbitmq_client
|
||||
from src.messaging.topology_setup import apply_topology
|
||||
from src.models.configs_read.daemons_json import daemons_config
|
||||
from src.models.rabbitmq_models.email import email_topology
|
||||
|
||||
writer = LogWriter()
|
||||
|
||||
async def run_one(daemon_name: str) -> None:
|
||||
daemon_cls = DAEMONS.get(daemon_name)
|
||||
@@ -39,15 +41,19 @@ async def run_enabled_from_config() -> None:
|
||||
async def main() -> None:
|
||||
channel = await rabbitmq_client.get_channel()
|
||||
await apply_topology(channel, email_topology)
|
||||
if len(sys.argv) < 2:
|
||||
print("Usage: python run_daemon.py <daemon_name> | --all")
|
||||
sys.exit(1)
|
||||
writer_task = asyncio.create_task(writer.log_writer())
|
||||
try:
|
||||
if len(sys.argv) < 2:
|
||||
print("Usage: python run_daemon.py <daemon_name> | --all")
|
||||
sys.exit(1)
|
||||
|
||||
arg = sys.argv[1]
|
||||
if arg == "--all":
|
||||
await run_enabled_from_config()
|
||||
else:
|
||||
await run_one(arg)
|
||||
arg = sys.argv[1]
|
||||
if arg == "--all":
|
||||
await run_enabled_from_config()
|
||||
else:
|
||||
await run_one(arg)
|
||||
finally:
|
||||
writer_task.cancel()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
Reference in New Issue
Block a user