Daemonize the worker.
This commit is contained in:
parent
2fcd8df42b
commit
607bf3ecc4
4 changed files with 47 additions and 11 deletions
|
@ -2,6 +2,7 @@ import logging
|
|||
import json
|
||||
import daemon
|
||||
import time
|
||||
import argparse
|
||||
|
||||
from apscheduler.scheduler import Scheduler
|
||||
|
||||
|
@ -29,13 +30,33 @@ def process_work_items():
|
|||
else:
|
||||
logger.debug('No work today.')
|
||||
|
||||
FORMAT = '%(asctime)-15s - %(levelname)s - %(pathname)s - %(funcName)s - %(message)s'
|
||||
logging.basicConfig(format=FORMAT, level=logging.DEBUG)
|
||||
|
||||
sched = Scheduler()
|
||||
sched.start()
|
||||
parser = argparse.ArgumentParser(description='Worker daemon to compute diffs')
|
||||
parser.add_argument('-D', action='store_true', default=False,
|
||||
help='Run the worker in daemon mode.')
|
||||
parser.add_argument('--log', default='diffsworker.log',
|
||||
help='Specify the log file for the worker as a daemon.')
|
||||
args = parser.parse_args()
|
||||
|
||||
sched.add_interval_job(process_work_items, seconds=10)
|
||||
def start_worker(args):
|
||||
FORMAT = '%(asctime)-15s - %(levelname)s - %(pathname)s - %(funcName)s - %(message)s'
|
||||
|
||||
while True:
|
||||
time.sleep(60 * 60 * 24) # sleep one day, basically forever
|
||||
if args.D:
|
||||
logging.basicConfig(filename=args.log, format=FORMAT, level=logging.DEBUG)
|
||||
else:
|
||||
logging.basicConfig(format=FORMAT, level=logging.DEBUG)
|
||||
|
||||
sched = Scheduler()
|
||||
sched.start()
|
||||
|
||||
sched.add_interval_job(process_work_items, seconds=10)
|
||||
|
||||
while True:
|
||||
time.sleep(60 * 60 * 24) # sleep one day, basically forever
|
||||
|
||||
if args.D:
|
||||
with daemon.DaemonContext():
|
||||
start_worker(args)
|
||||
|
||||
else:
|
||||
start_worker(args)
|
||||
|
|
Reference in a new issue