90 lines
3.8 KiB
Python
Executable File
90 lines
3.8 KiB
Python
Executable File
#!.venv/bin/python
|
|
|
|
# Import necessary libraries
|
|
import logging
|
|
import asyncio
|
|
|
|
# Import custom modules for configuration and database connection
|
|
from utils.config import loader_matlab_elab as setting
|
|
from utils.database import WorkflowFlags
|
|
from utils.database.matlab_query import get_matlab_command
|
|
from utils.csv.loaders import get_next_csv_atomic
|
|
from utils.orchestrator_utils import run_orchestrator, worker_context
|
|
from utils.database.loader_action import update_status, unlock
|
|
|
|
# Initialize the logger for this module
|
|
logger = logging.getLogger()
|
|
|
|
# Delay tra un processamento CSV e il successivo (in secondi)
|
|
ELAB_PROCESSING_DELAY = 0.2
|
|
# Tempo di attesa se non ci sono record da elaborare
|
|
NO_RECORD_SLEEP = 60
|
|
|
|
async def worker(worker_id: int, cfg: object, pool: object) -> None:
|
|
"""Esegue il ciclo di lavoro per l'elaborazione dei dati caricati.
|
|
|
|
Il worker preleva un record dal database che indica dati pronti per
|
|
l'elaborazione, esegue un comando Matlab associato e attende
|
|
prima di iniziare un nuovo ciclo.
|
|
|
|
Args:
|
|
worker_id (int): L'ID univoco del worker.
|
|
cfg (object): L'oggetto di configurazione.
|
|
pool (object): Il pool di connessioni al database.
|
|
"""
|
|
# Imposta il context per questo worker
|
|
worker_context.set(f"W{worker_id:02d}")
|
|
|
|
debug_mode = logging.getLogger().getEffectiveLevel() == logging.DEBUG
|
|
logger.info("Avviato")
|
|
|
|
while True:
|
|
try:
|
|
logger.info("Inizio elaborazione")
|
|
|
|
record = await get_next_csv_atomic(pool, cfg.dbrectable, WorkflowFlags.DATA_LOADED, WorkflowFlags.DATA_ELABORATED)
|
|
|
|
if record:
|
|
id, unit_type, tool_type, unit_name, tool_name = [x.lower().replace(" ", "_") if isinstance(x, str) else x for x in record]
|
|
tool_elab_info = await get_matlab_command(cfg, tool_name, unit_name, pool)
|
|
if tool_elab_info:
|
|
if tool_elab_info['statustools'].lower() in cfg.elab_status:
|
|
logger.info(f"Elaborazione id {id} per {unit_name} {tool_name} ")
|
|
|
|
matlab_cmd = f"timeout {cfg.matlab_timeout} ./run_{tool_elab_info['matcall']}.sh {cfg.matlab_runtime} {unit_name} {tool_name}"
|
|
proc = await asyncio.create_subprocess_shell(
|
|
matlab_cmd,
|
|
cwd=cfg.matlab_func_path,
|
|
stdout=asyncio.subprocess.PIPE,
|
|
stderr=asyncio.subprocess.PIPE
|
|
)
|
|
|
|
stdout, stderr = await proc.communicate()
|
|
|
|
if proc.returncode != 0:
|
|
logger.error("Errore durante l'elaborazione")
|
|
logger.error(stderr.decode().strip())
|
|
with open(f"{cfg.matlab_error_path}{unit_name}{tool_name}_output_error.txt", "w") as f:
|
|
f.write(stderr.decode().strip())
|
|
else:
|
|
logger.info(stdout.decode().strip())
|
|
await update_status(cfg, id, WorkflowFlags.DATA_ELABORATED, pool)
|
|
await unlock(cfg, id, pool)
|
|
await asyncio.sleep(ELAB_PROCESSING_DELAY)
|
|
else:
|
|
logger.info(f"id {id} - {unit_name} - {tool_name} {tool_elab_info['statustools']}: MatLab calc by-passed.")
|
|
else:
|
|
logger.info("Nessun record disponibile")
|
|
await asyncio.sleep(NO_RECORD_SLEEP)
|
|
|
|
except Exception as e:
|
|
logger.error(f"Errore durante l'esecuzione: {e}", exc_info=debug_mode)
|
|
await asyncio.sleep(1)
|
|
|
|
|
|
async def main():
|
|
"""Funzione principale che avvia l'elab_orchestrator."""
|
|
await run_orchestrator(setting.Config, worker)
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main()) |