jan/openreplay
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(chalice): catch unkeyed connection

Browse source
This commit is contained in:
Taha Yassine Kraiem 2022-09-14 14:04:41 +01:00
parent 9de9d43a8a
commit d989bd3e83
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
api/chalicelib/utils/pg_client.py
View file

@ -38,8 +38,15 @@ class ORThreadedConnectionPool(psycopg2.pool.ThreadedConnectionPool):
raise e
def putconn(self, *args, **kwargs):
super().putconn(*args, **kwargs)
self._semaphore.release()
try:
super().putconn(*args, **kwargs)
self._semaphore.release()
except psycopg2.pool.PoolError as e:
if str(e) == "trying to put unkeyed connection":
print("!!! trying to put unkeyed connection")
print(f"env-PG_POOL:{config('PG_POOL', cast=bool, default=None)}")
return
raise e
postgreSQL_pool: ORThreadedConnectionPool = None