1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
# Copyright 2013 Linaro Limited
# Author: Neil Williams <neil.williams@linaro.org>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
# MA 02110-1301, USA.
#
import os
import signal
import sys
import logging
import daemon
try:
import daemon.pidlockfile as pidlockfile
except ImportError:
from lockfile import pidlockfile
from logging.handlers import WatchedFileHandler
from subprocess import Popen
# Daemonise support for lava-server daemons, copied from
# lava-coordinator.
# pylint: disable=superfluous-parens,invalid-name,logging-not-lazy
child = None
def signal_handler(signum, frame): # pylint: disable=unused-argument
global child # pylint: disable=global-statement
try:
if signum == signal.SIGHUP:
logging.info("Forwarding a SIGHUP to the child")
child.send_signal(signum)
else:
logging.info("Closing daemon and child %d" % child.pid)
child.send_signal(signum)
child = None
sys.exit(os.EX_OK)
except Exception as e:
raise Exception('Error in signal handler: ' + str(e))
def getDaemonLogger(filePath, log_format=None, loglevel=logging.INFO):
logger = logging.getLogger()
logger.setLevel(loglevel)
try:
watchedHandler = WatchedFileHandler(filePath)
except Exception as e: # pylint: disable=broad-except
return e, None
watchedHandler.setFormatter(logging.Formatter(log_format or '%(asctime)s %(msg)s'))
logger.addHandler(watchedHandler)
return logger, watchedHandler
def daemonise(pidfile, logfile):
global child # pylint: disable=global-statement
client_logger, watched_file_handler = getDaemonLogger(logfile, loglevel=logging.DEBUG)
if isinstance(client_logger, Exception):
print("Fatal error creating client_logger: " + str(client_logger))
sys.exit(os.EX_OSERR)
# noinspection PyArgumentList
lockfile = pidlockfile.PIDLockFile(pidfile)
if lockfile.is_locked():
logging.error("PIDFile {} already locked".format(pidfile))
sys.exit(os.EX_OSERR)
context = daemon.DaemonContext(
detach_process=True,
working_directory=os.getcwd(),
pidfile=lockfile,
files_preserve=[watched_file_handler.stream],
stderr=watched_file_handler.stream,
stdout=watched_file_handler.stream)
context.signal_map = {
signal.SIGTERM: signal_handler,
signal.SIGHUP: signal_handler
}
# pass the args down to the process to be run in the daemon context
args = sys.argv
args.pop(0)
args.insert(0, 'lava-server') # each daemon shares a call to the django wrapper
arg_str = " ".join(args)
with context:
logging.info("Running LAVA Daemon")
child = Popen(args)
# lazy logging does not populate the strings at this point.
logging.debug("LAVA Daemon: %s pid: %d" % (arg_str, child.pid)) # pylint: disable=logging-not-lazy
child.communicate()
logging.info("Closing LAVA Daemon.")
return 0
|