psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2024-11-23 01:16:34 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update methods of ThreadedConnectionPool

Browse Source 
Updated methods of the ThreadedConnectionPool class to
use the elegant context manager syntax.
This commit is contained in:
Shubham Sharma 2021-09-21 22:46:30 +05:30
parent e9a62bf418
commit 766f06f290
1 changed files with 4 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
lib/pool.py
View File

@ -181,24 +181,15 @@ class ThreadedConnectionPool(AbstractConnectionPool):
def getconn(self, key=None): def getconn(self, key=None):
"""Get a free connection and assign it to 'key' if not None.""" """Get a free connection and assign it to 'key' if not None."""
self._lock.acquire() with self._lock:
try:
return self._getconn(key) return self._getconn(key)
finally:
self._lock.release()
def putconn(self, conn=None, key=None, close=False): def putconn(self, conn=None, key=None, close=False):
"""Put away an unused connection.""" """Put away an unused connection."""
self._lock.acquire() with self._lock:
try:
self._putconn(conn, key, close) self._putconn(conn, key, close)
finally:
self._lock.release()
def closeall(self): def closeall(self):
"""Close all connections (even the one currently in use.)""" """Close all connections (even the one currently in use.)"""
self._lock.acquire() with self._lock:
try:
self._closeall() self._closeall()
finally:
self._lock.release()