Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
Krish Moodbidri
self-reg-form
Commits
a7358eec
Commit
a7358eec
authored
Mar 11, 2021
by
Krish Moodbidri
Browse files
added timeout function
parent
b84cd7e2
Pipeline
#4254
failed with stage
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
tasks.py
View file @
a7358eec
...
...
@@ -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
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment