add service check

drip-deployer/docker_check.py

+#! /usr/bin/env python
+
+# Copyright 2017 --Yang Hu--
+#
+# 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.
+
+
+__author__ = 'Yang Hu'
+
+import paramiko, os
+from vm_info import VmInfo
+
+
+def docker_check(vm, compose_name):
+    try:
+        print "%s: ====== Start Check Docker Services  ======" % (vm.ip)
+        paramiko.util.log_to_file("deployment.log")
+        ssh = paramiko.SSHClient()
+        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+        ssh.connect(vm.ip, username=vm.user, key_filename=vm.key)
+        retstr = "\n===========  Cluster Node Information =============== \n"
+        stdin, stdout, stderr = ssh.exec_command("sudo docker node ls")
+        ret = stdout.readlines()
+        for i in ret: retstr += i.encode()
+
+        retstr += "\n============  Services Information ================ \n"
+        stdin, stdout, stderr = ssh.exec_command("sudo docker stack ps %s" % (compose_name))
+        ret = stdout.readlines()
+        for i in ret: retstr += i.encode()
+        print "%s: =========== Check Finished ==============" % (vm.ip)
+    except Exception as e:
+        print '%s: %s' % (vm.ip, e)
+        return "ERROR:" + vm.ip + " " + str(e)
+    ssh.close()
+    return retstr
+
+
+
+
+def run(vm_list, compose_name):
+    for i in vm_list:
+        if i.role == "master":
+            ret = docker_check(i, compose_name)
+            if "ERROR:" in ret:
+                return ret
+    return ret
\ No newline at end of file

drip-deployer/rpc_server.py

@@ -10,6 +10,7 @@ import docker_engine
 import docker_swarm
 import docker_compose
 import docker_service
+import docker_check
 import control_agent
 import ansible_playbook
 import sys, argparse
@@ -112,6 +113,9 @@ def handleDelivery(message):
     elif manager_type == "scale":
         ret = docker_service.run(vm_list, name_of_deployment, name_of_service, number_of_containers)
         return ret
+    elif manager_type == "service check":
+        ret = docker_check.run(vm_list, compose_name)
+        return ret
     else:
         return "ERROR: invalid cluster"