openstack/deb-python-eventlet
Code Issues Proposed changes

Merge

Browse Source
This commit is contained in:
Ryan Williams
2010-05-27 10:50:12 -07:00
parent f1cc6364dd 15629d15d8
commit 2dcb10406e
1 changed files with 4 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
eventlet/support/psycopg2_patcher.py
View File

@@ -38,20 +38,15 @@ def make_psycopg_green():
extensions.set_wait_callback(eventlet_wait_callback)
def eventlet_wait_callback(conn, timeout=-1,
# access these objects with LOAD_FAST instead of LOAD_GLOBAL lookup
POLL_OK=extensions.POLL_OK,
POLL_READ=extensions.POLL_READ,
POLL_WRITE=extensions.POLL_WRITE,
trampoline=trampoline):
def eventlet_wait_callback(conn, timeout=-1):
"""A wait callback useful to allow eventlet to work with Psycopg."""
while 1:
state = conn.poll()
if state == POLL_OK:
if state == extensions.POLL_OK:
break
elif state == POLL_READ:
elif state == extensions.POLL_READ:
trampoline(conn.fileno(), read=True)
elif state == POLL_WRITE:
elif state == extensions.POLL_WRITE:
trampoline(conn.fileno(), write=True)
else:
raise psycopg2.OperationalError(