check etcd for haproxy status instead of Postgres

Author: Winslett

haproxy_status.py

@@ -0,0 +1,37 @@
+#!/usr/bin/env python
+
+from BaseHTTPServer import BaseHTTPRequestHandler
+from helpers.etcd import Etcd
+from helpers.postgresql import Postgresql
+import sys, yaml, socket
+
+f = open(sys.argv[1], "r")
+config = yaml.load(f.read())
+f.close()
+
+etcd = Etcd(config["etcd"])
+postgresql = Postgresql(config["postgresql"])
+
+class StatusHandler(BaseHTTPRequestHandler):
+    def do_GET(self):
+        return self.do_ANY()
+    def do_OPTIONS(self):
+        return self.do_ANY()
+    def do_ANY(self):
+        if postgresql.name == etcd.current_leader()["hostname"]:
+          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()

haproxy_status.sh

@@ -3,6 +3,8 @@ etcd:
   scope: batman
   ttl: 30
   host: 127.0.0.1:4001
+haproxy_status:
+  listen: 127.0.0.1:15432
 postgresql:
   name: postgresql0
   listen: 127.0.0.1:5432

postgres0.yml

@@ -3,6 +3,8 @@ etcd:
   scope: batman
   ttl: 30
   host: 127.0.0.1:4001
+haproxy_status:
+  listen: 127.0.0.1:15433
 postgresql:
   name: postgresql1
   listen: 127.0.0.1:5433