paulmataruso/zulip
1
0
Fork 0
mirror of https://github.com/zulip/zulip.git synced 2025-11-05 14:35:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Handle SIGINT and SIGTERM in process_user_activity by closing communication gracefully

Browse Source 
(imported from commit dcf813fa8af1aaf5d6cc635268757e40c6c202ab)
This commit is contained in:
Leo Franchi
2013-02-15 11:03:28 -05:00
parent c2074813ca
commit d4a582fa1c
2 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
zephyr/management/commands/process_user_activity.py
View File

@@ -4,6 +4,8 @@ import simplejson
import pika
from zephyr.lib.actions import process_user_activity_event, process_user_presence_event
from zephyr.lib.queue import SimpleQueueClient
import sys
import signal
class Command(BaseCommand):
option_list = BaseCommand.option_list
@@ -20,8 +22,15 @@ class Command(BaseCommand):
print " [x] Received presence %r" % (event,)
process_user_presence_event(event)
def signal_handler(signal, frame):
print("[*] Closing and disconnecting from queues")
activity_queue.stop_consuming()
sys.exit(0)
signal.signal(signal.SIGTERM, signal_handler)
signal.signal(signal.SIGINT, signal_handler)
print ' [*] Waiting for messages. To exit press CTRL+C'
activity_queue.register_json_consumer('user_activity', callback_activity)
activity_queue.register_json_consumer('user_presence', callback_presence)
activity_queue.start_consuming()