add haproxy_status.py

haproxy_status.py

@@ -0,0 +1,40 @@
+#!/usr/bin/env python
+
+from BaseHTTPServer import BaseHTTPRequestHandler
+import sys, yaml, socket
+import requests
+
+f = open(sys.argv[1], "r")
+config = yaml.load(f.read())
+f.close()
+
+class StatusHandler(BaseHTTPRequestHandler):
+    def do_GET(self):
+        return self.do_ANY()
+    def do_OPTIONS(self):
+        return self.do_ANY()
+    def do_ANY(self):
+        base_url = "http://localhost:" + str(config["api_port"])
+        is_leader_resp = requests.get(base_url + "/ha/is_leader")
+        running_as_leader_resp = requests.get(
+            base_url + "/service/running_as_leader") 
+        is_healthy_resp = requests.get(base_url + "/service/is_healthy")
+        if is_healthy_resp.json()["data"]["is_healthy"] and \
+            is_leader_resp.json()["data"]["is_leader"] and \
+            running_as_leader_resp.json()["data"]["running_as_leader"]:
+          self.send_response(200)
+        else:
+          self.send_response(503)
+        self.end_headers()
+        self.wfile.write('\r\n')
+        return
+
+try:
+    from BaseHTTPServer import HTTPServer
+    host, port = config["haproxy_status"]["listen"].split(":")
+    server = HTTPServer((host, int(port)), StatusHandler)
+    print 'listening on %s:%s' % (host, port)
+    server.serve_forever()
+except KeyboardInterrupt:
+    print('^C received, shutting down server')
+    server.socket.close()

postgres0.yml

@@ -1,6 +1,8 @@
 loop_wait: 1000 #milliseconds
 data_dir: "data/postgres0" #canoe and pg data will go here
 api_port: 5000
+haproxy_status:
+        listen: 127.0.0.1:2345
 fsm:
   raft_port: 1234
   cluster_config_port: 1244

postgres1.yml

@@ -1,6 +1,8 @@
 loop_wait: 1000 #milliseconds
 data_dir: "data/postgres1" #canoe and pg data will go here
 api_port: 5001
+haproxy_status:
+        listen: 127.0.0.1:2346
 fsm:
   raft_port: 1235
   cluster_config_port: 1245

postgres2.yml

@@ -1,6 +1,8 @@
 loop_wait: 1000 #milliseconds
 data_dir: "data/postgres2" #canoe and pg data will go here
 api_port: 5002
+haproxy_status:
+        listen: 127.0.0.1:2347
 fsm:
   raft_port: 1236
   cluster_config_port: 1246