mirror of
https://github.com/outbackdingo/patroni.git
synced 2026-09-03 01:59:45 +00:00
Alexander KukushkinandGitHub
a4bd6a9b4b
* Convert postgresql.py into a package * Factor out cancellable process into a separate class * Factor out connection handler into a separate class * Move postmaster into postgresql package * Factor out pg_rewind into a separate class * Factor out bootstrap into a separate class * Factor out slots handler into a separate class * Factor out postgresql config handler into a separate class * Move callback_executor into postgresql package This is just a careful refactoring, without code changes.
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
import logging
|
|
import psycopg2
|
|
|
|
from contextlib import contextmanager
|
|
from threading import Lock
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Connection(object):
|
|
|
|
def __init__(self):
|
|
self._lock = Lock()
|
|
self._connection = None
|
|
self._cursor_holder = None
|
|
|
|
def set_conn_kwargs(self, conn_kwargs):
|
|
self._conn_kwargs = conn_kwargs
|
|
|
|
def get(self):
|
|
with self._lock:
|
|
if not self._connection or self._connection.closed != 0:
|
|
self._connection = psycopg2.connect(**self._conn_kwargs)
|
|
self._connection.autocommit = True
|
|
self.server_version = self._connection.server_version
|
|
return self._connection
|
|
|
|
def cursor(self):
|
|
if not self._cursor_holder or self._cursor_holder.closed or self._cursor_holder.connection.closed != 0:
|
|
logger.info("establishing a new patroni connection to the postgres cluster")
|
|
self._cursor_holder = self.get().cursor()
|
|
return self._cursor_holder
|
|
|
|
def close(self):
|
|
if self._connection and self._connection.closed == 0:
|
|
self._connection.close()
|
|
logger.info("closed patroni connection to the postgresql cluster")
|
|
self._cursor_holder = self._connection = None
|
|
|
|
|
|
@contextmanager
|
|
def get_connection_cursor(**kwargs):
|
|
with psycopg2.connect(**kwargs) as conn:
|
|
conn.autocommit = True
|
|
with conn.cursor() as cur:
|
|
yield cur
|