diff options
Diffstat (limited to 'plugins/config_generator.py')
| -rw-r--r-- | plugins/config_generator.py | 186 |
1 files changed, 186 insertions, 0 deletions
diff --git a/plugins/config_generator.py b/plugins/config_generator.py new file mode 100644 index 0000000..041ee8b --- /dev/null +++ b/plugins/config_generator.py @@ -0,0 +1,186 @@ +#!/usr/bin/python +# +# config_generator.py - Nagios configuration generator for gluster +# entities. Copyright (C) 2014 Red Hat Inc +# +# 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 +# + +from jinja2 import Environment, FileSystemLoader +import os +import shutil + + +class GlusterNagiosConfManager: + + def __init__(self, configDir, configTemplateDir, hostTemplateName): + self.configDir = configDir + self.configTemplateDir = configTemplateDir + self.hostTemplateName = hostTemplateName + self.__loadJinja() + + def __loadJinja(self): + self.jinjaEnv = Environment( + loader=FileSystemLoader(self.configTemplateDir)) + self.hostTemplate = self.jinjaEnv.get_template(self.hostTemplateName) + + def __createHostConfig(self, host): + hostConfigStr = self.hostTemplate.render(host=host) + hostConfig = {'name': host['host_name'], 'config': hostConfigStr} + return hostConfig + + def createHost(self, hostName, alias, template, + address, hostGroups, checkCommand, services): + host = {} + host['host_name'] = hostName + host['alias'] = alias + host['use'] = template + host['address'] = address + if checkCommand is not None: + host['check_command'] = checkCommand + if hostGroups is not None: + host['hostgroups'] = hostGroups + + if services is not None: + host['host_services'] = services + return host + + def __createVolumeUtilizationService(self, volume, clusterName): + volumeService = {} + volumeService['host_name'] = clusterName + volumeService['use'] = 'gluster-service-with-graph' + serviceDesc = 'Volume Utilization - %s' % (volume['name']) + volumeService['service_description'] = serviceDesc + volumeService['_VOL_NAME'] = volume['name'] + checkCommand = 'check_vol_utilization!%s!%s!70!90' % \ + (clusterName, volume['name']) + volumeService['check_command'] = checkCommand + volumeService['notes'] = "Volume type : %s" % (volume['typeStr']) + return volumeService + + def __createVolumeStatusService(self, volume, clusterName): + volumeService = {} + volumeService['host_name'] = clusterName + volumeService['use'] = 'gluster-service-with-graph' + serviceDesc = 'Volume Status - %s' % (volume['name']) + volumeService['service_description'] = serviceDesc + volumeService['_VOL_NAME'] = volume['name'] + checkCommand = 'check_vol_utilization!%s!%s!70!90' % \ + (clusterName, volume['name']) + volumeService['check_command'] = checkCommand + volumeService['notes'] = "Volume type : %s" % (volume['typeStr']) + return volumeService + + def createClusterUtilizationService(self, clusterName): + service = {} + service['host_name'] = clusterName + service['use'] = 'gluster-service-with-graph' + service['service_description'] = 'Cluster Utilization' + service['check_command'] = 'check_cluster_vol_usage!80!90' + return service + + def createClusterAutoConfigService(self, clusterName, hostIp): + service = {} + service['host_name'] = clusterName + service['use'] = 'generic-service' + service['service_description'] = 'Cluster Auto Config' + service['check_command'] = "gluster_auto_discovery!%s!%s" % ( + hostIp, clusterName) + service['check_interval'] = '1440' + return service + + def createrVolumeServices(self, volumes, clusterName): + volumeServices = [] + for volume in volumes: + volumeService = self.__createVolumeUtilizationService(volume, + clusterName) + volumeServices.append(volumeService) + return volumeServices + + def __createBrickUtilizationService(self, brick, hostName): + brickService = {} + brickService['use'] = 'brick-service' + brickService['host_name'] = hostName + serviceDesc = "Brick-%s:%s" % (hostName, brick['brickpath']) + brickService['service_description'] = serviceDesc + brickService['display_name'] = serviceDesc + brickService['_BRICK_DIR'] = brick['brickpath'] + return brickService + + def __createBrickStatusService(self, brick, hostName): + brickService = {} + brickService['use'] = 'brick-service' + brickService['host_name'] = hostName + serviceDesc = "Brick-%s:%s-Status" % (hostName, brick['brickpath']) + brickService['service_description'] = serviceDesc + brickService['display_name'] = serviceDesc + brickService['_BRICK_DIR'] = brick['brickpath'] + return brickService + + def createBrickServices(self, host): + brickServices = [] + for brick in host['bricks']: + brickService = self.__createBrickUtilizationService( + brick, host['hostip']) + brickServices.append(brickService) + return brickServices + + def generateNagiosConfigFromGlusterCluster(self, cluster): + hostsConfigs = [] + clusterServices = self.createrVolumeServices( + cluster.get('volumes'), cluster['name']) + clusterServices.append(self.createClusterUtilizationService( + cluster['name'])) + clusterServices.append(self.createClusterAutoConfigService( + cluster['name'], cluster['hosts'][0]['hostip'])) + clusterHostConfig = self.createHost( + cluster['name'], cluster['name'], "gluster-cluster", + cluster['name'], "", "check_dummy", clusterServices) + hostsConfigs.append(clusterHostConfig) + for host in cluster['hosts']: + brickServices = self.createBrickServices(host) + hostGroups = "gluster_hosts,%s" % (cluster['name']) + hostConfig = self.createHost( + host['hostip'], host['hostip'], "gluster-host", + host['hostip'], hostGroups, "", brickServices) + hostsConfigs.append(hostConfig) + self.generateConfigFiles(hostsConfigs) + return hostsConfigs + + def generateConfigFiles(self, hosts): + clusterConfig = {'name': None, 'hostConfigs': []} + clusterConfigDir = None + for host in hosts: + if host['use'] == 'gluster-cluster': + clusterConfigDir = self.configDir + "/" + host['host_name'] + self.__prepareConfDir(clusterConfigDir) + clusterConfig['name'] = host['host_name'] + hostConfig = self.__createHostConfig(host) + clusterConfig['hostConfigs'].append(hostConfig) + for hostConfig in clusterConfig['hostConfigs']: + self.__writeHostConfig(clusterConfigDir, hostConfig) + + def __prepareConfDir(self, confDir): + if os.path.exists(confDir): + # Deleting the config dir to write new configs + shutil.rmtree(confDir) + os.mkdir(confDir) + + def __writeHostConfig(self, clusterConfigDir, hostConfig): + if clusterConfigDir is None: + raise Exception("Cluster configuration directory can't None") + configFilePath = clusterConfigDir + "/" + hostConfig['name'] + ".cfg" + with open(configFilePath, 'w') as configFile: + configFile.write(hostConfig['config']) |