baseimage-tutorial-framework/tfw/components/process_management/process_log_handler.py

44 lines
1.2 KiB
Python
Raw Normal View History

2019-06-10 13:32:45 +00:00
import logging
from .log_inotify_observer import LogInotifyObserver
2018-05-30 13:21:36 +00:00
LOG = logging.getLogger(__name__)
class ProcessLogHandler:
2019-08-08 13:05:37 +00:00
keys = ['process.log']
2018-05-31 12:03:11 +00:00
2019-07-23 13:32:50 +00:00
def __init__(self, *, process_name, supervisor_uri, log_tail=0):
self.connector, self._monitor = None, None
2018-05-30 13:21:36 +00:00
self.process_name = process_name
2019-07-23 13:32:50 +00:00
self._supervisor_uri = supervisor_uri
self._initial_log_tail = log_tail
2018-05-30 13:21:36 +00:00
self.command_handlers = {
2019-08-08 12:45:59 +00:00
'process.log.set': self.handle_set
2018-05-30 13:21:36 +00:00
}
def start(self):
self._monitor = LogInotifyObserver(
connector=self.connector,
process_name=self.process_name,
supervisor_uri=self._supervisor_uri,
log_tail=self._initial_log_tail
)
self._monitor.start()
def handle_event(self, message, _):
2018-05-30 13:21:36 +00:00
try:
self.command_handlers[message['key']](message)
2018-05-30 13:21:36 +00:00
except KeyError:
LOG.error('IGNORING MESSAGE: Invalid message received: %s', message)
def handle_set(self, data):
if data.get('name'):
self._monitor.process_name = data['name']
if data.get('tail'):
self._monitor.log_tail = data['tail']
def cleanup(self):
2019-06-27 13:22:48 +00:00
self._monitor.stop()