Changeset View
Changeset View
Standalone View
Standalone View
swh/scheduler/backend.py
Show First 20 Lines • Show All 67 Lines • ▼ Show 20 Lines | def __init__(self, db, min_pool_conns=1, max_pool_conns=10): | ||||
self._pool = psycopg2.pool.ThreadedConnectionPool( | self._pool = psycopg2.pool.ThreadedConnectionPool( | ||||
min_pool_conns, | min_pool_conns, | ||||
max_pool_conns, | max_pool_conns, | ||||
db, | db, | ||||
cursor_factory=psycopg2.extras.RealDictCursor, | cursor_factory=psycopg2.extras.RealDictCursor, | ||||
) | ) | ||||
self._db = None | self._db = None | ||||
def get_current_version(self): | |||||
return self.current_version | |||||
def get_db(self): | def get_db(self): | ||||
if self._db: | if self._db: | ||||
return self._db | return self._db | ||||
return BaseDb.from_pool(self._pool) | return BaseDb.from_pool(self._pool) | ||||
def put_db(self, db): | def put_db(self, db): | ||||
if db is not self._db: | if db is not self._db: | ||||
db.put_conn() | db.put_conn() | ||||
▲ Show 20 Lines • Show All 1,060 Lines • Show Last 20 Lines |