Git-Mirrors/brozzler
1
0
Fork 0
mirror of https://github.com/internetarchive/brozzler.git synced 2025-04-20 23:56:34 -04:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

clean shutdown for brozzler-hq

Browse Source
This commit is contained in:
Noah Levitt 2015-07-11 18:18:54 -07:00
parent 384120928c
commit b0f3b8a5e3
1 changed files with 17 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
bin/brozzler-hq
View File

@ -13,6 +13,7 @@ import kombu
import kombu.simple
import json
import umbra.hq
import signal
arg_parser = argparse.ArgumentParser(prog=os.path.basename(__file__),
description="brozzler-hq - headquarters of distributed brozzler crawl",
@ -201,7 +202,22 @@ class BrozzlerHQ:
logging.info("brozzler-hq starting")
class ShutdownRequested(Exception):
pass
def sigterm(signum, frame):
raise ShutdownRequested("shutdown requested (caught SIGTERM)")
def sigint(signum, frame):
raise ShutdownRequested("shutdown requested (caught SIGINT)")
signal.signal(signal.SIGTERM, sigterm)
signal.signal(signal.SIGINT, sigint)
db = BrozzlerHQDb(db_file=args.db_file)
hq = BrozzlerHQ(amqp_url=args.amqp_url, db=db)
hq.run()
try:
hq.run()
except ShutdownRequested as e:
logging.info("{}".format(e))