mirror of
https://github.com/outbackdingo/patroni.git
synced 2026-08-26 15:40:21 +00:00
Alexander Kukushkin
d202f72a42
Server always returns json which contains some status of postgres: * instance type: master/slave * xlog location for master * xlog received_location, replayed_location This server could be used by haproxy GET / returns 200 if there is postgres master behind governor. GET /slave returns 200 is there is postgres slave behind governor In all other cases it returns 503 Currently server always listening on 0.0.0.0:8080, but it should be configurable. In the next versions this rest api could also report some status of etcd and could be used by is_healthiest_node method
70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
import json
|
|
import logging
|
|
import psycopg2
|
|
import sys
|
|
|
|
from threading import Thread
|
|
|
|
if sys.hexversion >= 0x03000000:
|
|
from http.server import BaseHTTPRequestHandler, HTTPServer
|
|
else:
|
|
from BaseHTTPServer import BaseHTTPRequestHandler, HTTPServer
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class RestApiHandler(BaseHTTPRequestHandler):
|
|
|
|
def do_GET(self):
|
|
try:
|
|
response = self.get_postgresql_status()
|
|
except (psycopg2.OperationalError, psycopg2.InterfaceError):
|
|
response = {'running': False}
|
|
|
|
path = '/master' if self.path == '/' else self.path
|
|
status_code = 200 if response['running'] and response['role'] in path else 503
|
|
|
|
self.send_response(status_code)
|
|
self.send_header('Content-Type', 'application/json')
|
|
self.end_headers()
|
|
self.wfile.write(json.dumps(response).encode('utf-8'))
|
|
|
|
def get_postgresql_status(self):
|
|
if not self.server.governor.postgresql.is_running():
|
|
return {'running': False}
|
|
cursor = self.server.cursor()
|
|
cursor.execute("""SELECT to_char(pg_postmaster_start_time(), 'YYYY-MM-DD HH24:MI:SS.MS TZ'),
|
|
pg_is_in_recovery(),
|
|
pg_current_xlog_location(),
|
|
pg_last_xlog_receive_location(),
|
|
pg_last_xlog_replay_location(),
|
|
pg_is_in_recovery() AND pg_is_xlog_replay_paused()""")
|
|
row = cursor.fetchone()
|
|
return {
|
|
'running': True,
|
|
'postmaster_start_time': row[0],
|
|
'role': 'slave' if row[1] else 'master',
|
|
'xlog': ({
|
|
'received_location': row[3],
|
|
'replayed_location': row[4],
|
|
'paused': row[5]} if row[1] else {
|
|
'location': row[2]
|
|
})
|
|
}
|
|
|
|
|
|
class RestApiServer(HTTPServer, Thread):
|
|
|
|
def __init__(self, governor, listen_address='0.0.0.0', listen_port=8080):
|
|
HTTPServer.__init__(self, (listen_address, listen_port), RestApiHandler)
|
|
Thread.__init__(self, target=self.serve_forever)
|
|
self.governor = governor
|
|
self._cursor_holder = None
|
|
self.daemon = True
|
|
|
|
def cursor(self):
|
|
if not self._cursor_holder or self._cursor_holder.closed != 0:
|
|
self._cursor_holder = self.governor.postgresql.connection().cursor()
|
|
return self._cursor_holder
|