Merge pull request #5 from nullkarma/master

Added Health Service and Check Action

Author: LindsayHill

CHANGES.md

@@ -1,5 +1,10 @@
 # Change Log
 
+## 0.4.0
+
+ - Added consul.health_service action to query a services status and group the result by hostname
+ - Added consul.health_check action to query a services status and group the result by check
+
 ## 0.3.0
 
  - Added consul.list action to get a list of Keys from consul under a given <root> key

actions/health_check.py

@@ -0,0 +1,15 @@
+from lib import action
+from collections import defaultdict
+
+
+class ConsulCheckHealth(action.ConsulBaseAction):
+    def run(self, service):
+        index, nodes = self.consul.health.checks(service=service)
+        s = defaultdict(dict)
+        for node in nodes:
+            try:
+                s[node.get('CheckID')][node.get('Status')].append(node.get('Node'))
+            except KeyError:
+                s[node.get('CheckID')][node.get('Status')] = []
+                s[node.get('CheckID')][node.get('Status')].append(node.get('Node'))
+        return s

actions/health_check.yaml

@@ -0,0 +1,11 @@
+name: health_check
+runner_type: run-python
+description: "Query health status of a check in consul"
+enabled: true
+entry_point: "health_check.py"
+parameters:
+    service:
+        type: string
+        description: "Service to query in Consul"
+        required: true
+

actions/health_service.py

@@ -0,0 +1,16 @@
+from lib import action
+from collections import defaultdict
+
+
+class ConsulServiceHealth(action.ConsulBaseAction):
+    def run(self, service, tag=None):
+        index, nodes = self.consul.health.service(service, tag=tag)
+        s = defaultdict(dict)
+        for node in nodes:
+            for check in node.get('Checks'):
+                try:
+                    s[check.get('Node')][check.get('Status')].append(check.get('CheckID'))
+                except KeyError:
+                    s[check.get('Node')][check.get('Status')] = []
+                    s[check.get('Node')][check.get('Status')].append(check.get('CheckID'))
+        return s

actions/health_service.yaml

@@ -0,0 +1,14 @@
+name: health_service
+runner_type: run-python
+description: "Query health status of a service in consul"
+enabled: true
+entry_point: "health_service.py"
+parameters:
+    service:
+        type: string
+        description: "Service to query in Consul"
+        required: true
+    tag:
+        type: string
+        description: "Optional tags to query"
+        required: false

pack.yaml

@@ -3,6 +3,6 @@
 ref: consul
 name: consul
 description: consul
-version: 0.3.0
+version: 0.4.0
 author: jfryman
 email: [email protected]