diff --git a/tasks.py b/tasks.py index d96da2783878a45012013f076b96c3bc0506ea3e..6001440ae2c6c9cb25440d7767509c4fc1be3a87 100644 --- a/tasks.py +++ b/tasks.py @@ -2,6 +2,7 @@ import vars import sys import json import time +import signal from celery import Celery from flask_socketio import SocketIO @@ -13,6 +14,7 @@ broker_url = vars.broker_url celery = Celery('flask_user_reg', broker=broker_url) socketio = SocketIO(message_queue=vars.message_queue) +timeout = 60 def callback(channel, method, properties, body): msg = json.loads(body) @@ -33,6 +35,9 @@ def send_msg(event, room): print("Post '{}' to room '{}'".format(event,room)) socketio.emit(event, room=room) +def timeout_handler(signum, frame): + print("Process timeout, there's might some issue with agents") + rc_util.rc_rmq.stop_consume() @celery.task def celery_create_account(json, session): @@ -47,6 +52,12 @@ def celery_create_account(json, session): print(username) rc_util.add_account(username, email, fullname, reason) print('sent account info') + + # Set initial timeout timer + signal.signal(signal.SIGALRM, timeout_handler) + signal.setitimer(signal.ITIMER_REAL, timeout) + + print('Waiting for completion...') rc_util.consume(username, routing_key=f'complete.{username}', callback=callback) send_msg('account ready', room)