mirror of
https://github.com/mtan93/cachet-url-monitor.git
synced 2026-03-07 21:21:58 +00:00
a13a42d51c
* feat(multihreading): each url has it's own thread * Fixing broken unit tests * Improving readability when there are multiple URLs registerd and creating new action to upload metrics * Improving error message when there's no file found * Bumping the version Co-authored-by: Alex Berenshtein <aberenshtein@yotpo.com>
136 lines
3.9 KiB
Python
136 lines
3.9 KiB
Python
#!/usr/bin/env python
|
|
import logging
|
|
import sys
|
|
import threading
|
|
import time
|
|
|
|
import schedule
|
|
from yaml import load, SafeLoader
|
|
|
|
from cachet_url_monitor.configuration import Configuration
|
|
|
|
cachet_mandatory_fields = ['api_url', 'token']
|
|
|
|
|
|
class Agent(object):
|
|
"""Monitor agent that will be constantly verifying if the URL is healthy
|
|
and updating the component.
|
|
"""
|
|
|
|
def __init__(self, configuration, decorators=None):
|
|
self.configuration = configuration
|
|
if decorators is None:
|
|
decorators = []
|
|
self.decorators = decorators
|
|
|
|
def execute(self):
|
|
"""Will verify the API status and push the status and metrics to the
|
|
cachet server.
|
|
"""
|
|
self.configuration.evaluate()
|
|
self.configuration.push_metrics()
|
|
self.configuration.if_trigger_update()
|
|
|
|
for decorator in self.decorators:
|
|
decorator.execute(self.configuration)
|
|
|
|
def start(self):
|
|
"""Sets up the schedule based on the configuration file."""
|
|
schedule.every(self.configuration.endpoint['frequency']).seconds.do(self.execute)
|
|
|
|
|
|
class Decorator(object):
|
|
def execute(self, configuration):
|
|
pass
|
|
|
|
|
|
class UpdateStatusDecorator(Decorator):
|
|
def execute(self, configuration):
|
|
configuration.push_status()
|
|
|
|
|
|
class CreateIncidentDecorator(Decorator):
|
|
def execute(self, configuration):
|
|
configuration.push_incident()
|
|
|
|
|
|
class PushMetricsDecorator(Decorator):
|
|
def execute(self, configuration):
|
|
configuration.push_metrics()
|
|
|
|
|
|
class Scheduler(object):
|
|
def __init__(self, config_file, endpoint_index):
|
|
self.logger = logging.getLogger('cachet_url_monitor.scheduler.Scheduler')
|
|
self.configuration = Configuration(config_file, endpoint_index)
|
|
self.agent = self.get_agent()
|
|
|
|
self.stop = False
|
|
|
|
def get_agent(self):
|
|
action_names = {
|
|
'CREATE_INCIDENT': CreateIncidentDecorator,
|
|
'UPDATE_STATUS': UpdateStatusDecorator,
|
|
'PUSH_METRICS': PushMetricsDecorator,
|
|
}
|
|
actions = []
|
|
for action in self.configuration.get_action():
|
|
self.logger.info(f'Registering action {action}')
|
|
actions.append(action_names[action]())
|
|
return Agent(self.configuration, decorators=actions)
|
|
|
|
def start(self):
|
|
self.agent.start()
|
|
self.logger.info('Starting monitor agent...')
|
|
while not self.stop:
|
|
schedule.run_pending()
|
|
time.sleep(self.configuration.endpoint['frequency'])
|
|
|
|
|
|
class NewThread(threading.Thread):
|
|
def __init__(self, scheduler):
|
|
threading.Thread.__init__(self)
|
|
self.scheduler = scheduler
|
|
|
|
def run(self):
|
|
self.scheduler.start()
|
|
|
|
|
|
def validate_config():
|
|
if 'endpoints' not in config_file.keys():
|
|
fatal_error('Endpoints is a mandatory field')
|
|
|
|
if config_file['endpoints'] is None:
|
|
fatal_error('Endpoints array can not be empty')
|
|
|
|
for key in cachet_mandatory_fields:
|
|
if key not in config_file['cachet']:
|
|
fatal_error('Missing cachet mandatory fields')
|
|
|
|
|
|
def fatal_error(message):
|
|
logging.getLogger('cachet_url_monitor.scheduler').fatal("%s", message)
|
|
sys.exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
FORMAT = "%(levelname)9s [%(asctime)-15s] %(name)s - %(message)s"
|
|
logging.basicConfig(format=FORMAT, level=logging.INFO)
|
|
for handler in logging.root.handlers:
|
|
handler.addFilter(logging.Filter('cachet_url_monitor'))
|
|
|
|
if len(sys.argv) <= 1:
|
|
logging.getLogger('cachet_url_monitor.scheduler').fatal('Missing configuration file argument')
|
|
sys.exit(1)
|
|
|
|
try:
|
|
config_file = load(open(sys.argv[1], 'r'), SafeLoader)
|
|
except FileNotFoundError:
|
|
logging.getLogger('cachet_url_monitor.scheduler').fatal(f'File not found: {sys.argv[1]}')
|
|
sys.exit(1)
|
|
|
|
validate_config()
|
|
|
|
for endpoint_index in range(len(config_file['endpoints'])):
|
|
NewThread(Scheduler(config_file, endpoint_index)).start()
|