psycopg/psycopg2
1
1
Fork 0
mirror of https://github.com/psycopg/psycopg2.git synced 2025-07-29 09:29:46 +03:00
Code Issues Packages Projects Releases Wiki Activity

Handle forking somewhat more safely

Browse Source
This commit is contained in:
Jonathan Mortensen 2020-12-04 18:37:24 -08:00 committed by GitHub
parent e85ef2298b
commit 5a2c8c6b04
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
lib/pool.py
View File

@ -185,3 +185,24 @@ class ThreadedConnectionPool(AbstractConnectionPool):
self._closeall() self._closeall()
finally: finally:
self._lock.release() self._lock.release()
class MultiProcessThreadedConnectionPool(ThreadedConnectionPool):
def __init__(self, *args, **kwargs):
self.pid = os.getpid()
self.args = args
self.kwargs = kwargs
super().__init__(*args, **kwargs)
def getconn(self, *args, **kwargs):
current_pid = os.getpid()
if current_pid != self.pid:
super().__init__(*self.args, **self.kwargs)
self.pid = current_pid
return super().getconn(*args, **kwargs)
def putconn(self, *args, **kwargs):
current_pid = os.getpid()
if current_pid != self.pid:
raise Exception("Returned a connection to the pool that was acquired pre-fork. Don't use pre-fork connections post-fork.")
return super().putconn(*args, **kwargs)