mirror of
https://github.com/apache/cloudstack.git
synced 2025-10-26 08:42:29 +01:00
84 lines
3.6 KiB
Python
Executable File
84 lines
3.6 KiB
Python
Executable File
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
import logging
|
|
from cs.CsDatabag import CsDataBag
|
|
from CsFile import CsFile
|
|
import json
|
|
|
|
MON_CONFIG = "/etc/monitor.conf"
|
|
HC_CONFIG = "/root/health_checks_data.json"
|
|
|
|
|
|
class CsMonitor(CsDataBag):
|
|
""" Manage Monitor script schedule and health checks for router """
|
|
|
|
def get_basic_check_interval(self):
|
|
return self.dbag["health_checks_basic_run_interval"] if "health_checks_basic_run_interval" in self.dbag else 3
|
|
|
|
def get_advanced_check_interval(self):
|
|
return self.dbag["health_checks_advanced_run_interval"] if "health_checks_advanced_run_interval" in self.dbag else 0
|
|
|
|
def setupMonitorConfigFile(self):
|
|
if "config" in self.dbag:
|
|
procs = [x.strip() for x in self.dbag['config'].split(',')]
|
|
file = CsFile(MON_CONFIG)
|
|
for proc in procs:
|
|
bits = [x for x in proc.split(':')]
|
|
if len(bits) < 5:
|
|
continue
|
|
for i in range(0, 4):
|
|
file.add(bits[i], -1)
|
|
file.commit()
|
|
|
|
def setupHealthCheckCronJobs(self):
|
|
cron_rep_basic = self.get_basic_check_interval()
|
|
cron_rep_advanced = self.get_advanced_check_interval()
|
|
cron = CsFile("/etc/cron.d/process")
|
|
cron.deleteLine("root /usr/bin/python /root/monitorServices.py")
|
|
cron.add("SHELL=/bin/bash", 0)
|
|
cron.add("PATH=/usr/local/sbin:/usr/local/bin:/sbin:/bin:/usr/sbin:/usr/bin", 1)
|
|
if cron_rep_basic > 0:
|
|
cron.add("*/" + str(cron_rep_basic) + " * * * * root /usr/bin/python /root/monitorServices.py basic", -1)
|
|
if cron_rep_advanced > 0:
|
|
cron.add("*/" + str(cron_rep_advanced) + " * * * * root /usr/bin/python /root/monitorServices.py advanced", -1)
|
|
cron.commit()
|
|
|
|
def setupHealthChecksConfigFile(self):
|
|
hc_data = {}
|
|
hc_data["health_checks_basic_run_interval"] = self.get_basic_check_interval()
|
|
hc_data["health_checks_advanced_run_interval"] = self.get_advanced_check_interval()
|
|
hc_data["health_checks_enabled"] = self.dbag["health_checks_enabled"] if "health_checks_enabled" in self.dbag else False
|
|
|
|
if "excluded_health_checks" in self.dbag:
|
|
excluded_checks = self.dbag["excluded_health_checks"]
|
|
hc_data["excluded_health_checks"] = [ch.strip() for ch in excluded_checks.split(",")] if len(excluded_checks) > 0 else []
|
|
else:
|
|
hc_data["excluded_health_checks"] = []
|
|
|
|
if "health_checks_config" in self.dbag:
|
|
hc_data["health_checks_config"] = self.dbag["health_checks_config"]
|
|
else:
|
|
hc_data["health_checks_config"] = {}
|
|
|
|
with open(HC_CONFIG, 'w') as f:
|
|
json.dump(hc_data, f, ensure_ascii=False, indent=4)
|
|
|
|
def process(self):
|
|
self.setupMonitorConfigFile()
|
|
self.setupHealthChecksConfigFile()
|
|
self.setupHealthCheckCronJobs()
|