Git-Mirrors/brozzler
1
0
Fork 0
mirror of https://github.com/internetarchive/brozzler.git synced 2025-08-21 12:28:09 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

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
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 kombu.simple
import json import json
import umbra.hq import umbra.hq
import signal
arg_parser = argparse.ArgumentParser(prog=os.path.basename(__file__), arg_parser = argparse.ArgumentParser(prog=os.path.basename(__file__),
description="brozzler-hq - headquarters of distributed brozzler crawl", description="brozzler-hq - headquarters of distributed brozzler crawl",
@ -201,7 +202,22 @@ class BrozzlerHQ:
logging.info("brozzler-hq starting") 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) db = BrozzlerHQDb(db_file=args.db_file)
hq = BrozzlerHQ(amqp_url=args.amqp_url, db=db) hq = BrozzlerHQ(amqp_url=args.amqp_url, db=db)
hq.run() try:
hq.run()
except ShutdownRequested as e:
logging.info("{}".format(e))