def run_as_daemon(loader, interval):
"""
Given a loader object, start it as a daemon process.
"""
log.info("The loader will run as a daemon.")
# We need to preserve the file descriptor for any log files.
rootlogger = logging.getLogger()
log_files = [x.stream for x in rootlogger.handlers]
dc = DaemonContext(files_preserve=log_files)
try:
# Note: because we need to be compatible with python 2.4, we can't use
# with dc:
# here - we need to call the open() and close() methods
# manually.
dc.open()
loader.startup()
# every <interval> seconds, process the records in the incoming
# directory
while True:
loader.load_all_msgs()
time.sleep(interval)
except SystemExit, e:
log.info("Received the shutdown signal: " + str(e))
loader.shutdown()
dc.close()
def run_as_daemon(loader, interval):
"""
Given a loader object, start it as a daemon process.
"""
log.info("The loader will run as a daemon.")
# We need to preserve the file descriptor for any log files.
rootlogger = logging.getLogger()
log_files = [x.stream for x in rootlogger.handlers]
dc = DaemonContext(files_preserve=log_files)
try:
# Note: because we need to be compatible with python 2.4, we can't use
# with dc:
# here - we need to call the open() and close() methods
# manually.
dc.open()
loader.startup()
# every <interval> seconds, process the records in the incoming
# directory
while True:
loader.load_all_msgs()
time.sleep(interval)
except SystemExit, e:
log.info("Received the shutdown signal: " + str(e))
loader.shutdown()
dc.close()
class Daemon(object):
"""
Controller for a callable running in a separate background process.
"""
start_message = u"{1} started with pid {0}"
def __init__(self, app, pidfile=None, stdout=sys.stdout, stderr=sys.stderr, **options):
"""
Set up the parameters of a new runner.
The `app` argument must have the following attributes:
* `run`: Callable that will be invoked when the daemon is
started.
"""
self.app = app
if pidfile:
self.pidfile = make_pidlockfile(pidfile, 5)
else:
self.pidfile = None
self.daemon_context = DaemonContext(
stdout=stdout,
stderr=stderr,
pidfile=self.pidfile,
**options
)
def start(self):
"""
Open the daemon context and run the application.
"""
if self.pidfile and is_pidfile_stale(self.pidfile):
self.pidfile.break_lock()
try:
self.daemon_context.open()
except lockfile.AlreadyLocked:
raise DaemonRunnerStartFailureError(
u"PID file %(pidfile_path)r already locked".format(
self.pidfile.path))
pid = os.getpid()
message = self.start_message.format(pid, self.app.name)
emit_message(message)
signal.signal(signal.SIGHUP, self.restart)
self.app.run()
def stop(self):
"""
Exit the daemon process specified in the current PID file.
"""
if not self.pidfile:
self.daemon_context.close()
if not self.pidfile.is_locked():
raise DaemonRunnerStopFailureError(
u"PID file {0} not locked".format(self.pidfile.path))
if is_pidfile_stale(self.pidfile):
self.pidfile.break_lock()
else:
self._terminate_daemon_process()
def restart(self, *args):
"""
Stop, then start.
"""
self.stop()
self.start()
def _terminate_daemon_process(self):
"""
Terminate the daemon process specified in the current PID file.
"""
pid = self.pidfile.read_pid()
try:
os.kill(pid, signal.SIGTERM)
except OSError, exc:
raise DaemonRunnerStopFailureError(
u"Failed to terminate {0}: {1}".format(pid, exc))
class Daemon(object):
"""
Controller for a callable running in a separate background process.
"""
start_message = u"{1} started with pid {0}"
def __init__(self, app, pidfile=None, stdout=sys.stdout, stderr=sys.stderr, **options):
"""
Set up the parameters of a new runner.
The `app` argument must have the following attributes:
* `run`: Callable that will be invoked when the daemon is
started.
"""
self.app = app
if pidfile:
self.pidfile = make_pidlockfile(pidfile, 5)
else:
self.pidfile = None
self.daemon_context = DaemonContext(
stdout=stdout,
stderr=stderr,
pidfile=self.pidfile,
**options
)
def start(self):
"""
Open the daemon context and run the application.
"""
if self.pidfile and is_pidfile_stale(self.pidfile):
self.pidfile.break_lock()
try:
self.daemon_context.open()
except lockfile.AlreadyLocked:
raise DaemonRunnerStartFailureError(
u"PID file %(pidfile_path)r already locked".format(
self.pidfile.path))
pid = os.getpid()
message = self.start_message.format(pid, self.app.name)
emit_message(message)
signal.signal(signal.SIGHUP, self.restart)
self.app.run()
def stop(self):
"""
Exit the daemon process specified in the current PID file.
"""
if not self.pidfile:
self.daemon_context.close()
if not self.pidfile.is_locked():
raise DaemonRunnerStopFailureError(
u"PID file {0} not locked".format(self.pidfile.path))
if is_pidfile_stale(self.pidfile):
self.pidfile.break_lock()
else:
self._terminate_daemon_process()
def restart(self, *args):
"""
Stop, then start.
"""
self.stop()
self.start()
def _terminate_daemon_process(self):
"""
Terminate the daemon process specified in the current PID file.
"""
pid = self.pidfile.read_pid()
try:
os.kill(pid, signal.SIGTERM)
except OSError, exc:
raise DaemonRunnerStopFailureError(
u"Failed to terminate {0}: {1}".format(pid, exc))
