Merge pull request #58 from sileht/sileht/pacemakerwrapper

Add peacemaker wrapper for agent unloadbalancable

scripts/oschecks/pacemaker_host_check.py

@@ -0,0 +1,75 @@
+#!/usr/bin/env python
+# -*- encoding: utf-8 -*-
+# Openstack Monitoring script for Sensu / Nagios
+#
+# Copyright © 2013-2014 eNovance <licensing@enovance.com>
+#
+# Author:Mehdi Abaakouk <mehdi.abaakouk@enovance.com>
+#
+# Licensed 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 argparse
+import subprocess
+import os
+
+try:
+    import utils
+except ImportError:
+    from oschecks import utils
+
+
+def pacemaker_host_check():
+    parser = argparse.ArgumentParser(
+        description='Check amqp connection of an OpenStack service.')
+    parser.add_argument('-r', dest='pacemaker_resource',
+                        help='pacemaker resource', required=True)
+    parser.add_argument('-s', dest='script', required=True,
+                        help='Script')
+    options = parser.parse_args()
+
+    local_hostname = subprocess.check_output(['hostname', '-s']).strip()
+
+    try:
+        output = subprocess.check_output(['pcs', 'status'])
+    except subprocess.CalledProcessError as e:
+        utils.critical('pcs status with status %s: %s' %
+                       e.returncode, e.output)
+    except OSError:
+        utils.critical('pcs not found')
+    for line in output.splitlines():
+        line = " ".join(line.strip().split())  # Sanitize separator
+        if not line:
+            continue
+
+        resource, remaining = line.split(None, 1)
+        if resource == options.pacemaker_resource:
+            agent, __, remaining = remaining.partition(' ')
+            if ' ' in remaining:
+                status, __, current_hostname = remaining.partition(' ')
+            else:
+                status, current_hostname = remaining, ''
+            if status != "Started":
+                utils.critical("pacemaker resource %s is not started (%s)" %
+                               (resource, status))
+            if current_hostname != local_hostname:
+                utils.ok("pacemaker resource %s doesn't on this node "
+                         "(but on %s)" % (resource, current_hostname))
+            os.execvp(options.script, [options.script])
+
+    else:
+        utils.critical('pacemaker resource %s not found' %
+                       options.pacemaker_resource)
+
+
+if __name__ == '__main__':
+    utils.safe_run(pacemaker_host_check)

scripts/oschecks/utils.py

@@ -23,6 +23,7 @@ import copy
 import itertools
 import time
 import traceback
+import os
 import sys
 
 import psutil
@@ -51,10 +52,15 @@ def ok(msg):
 
 
 def check_process_name(name, p):
+    if p.name == name:
+        return True
     # name can be truncated and a script so check also if it can be an
     # argument to an interpreter
-    return p.name == name or \
-        len(p.cmdline) > 1 and p.cmdline[1].find(name) != -1
+    if len(p.cmdline) > 0 and os.path.basename(p.cmdline[0]) == name:
+        return True
+    if len(p.cmdline) > 1 and os.path.basename(p.cmdline[1]) == name:
+        return True
+    return False
 
 
 def check_process_exists_and_amqp_connected(name):