mirror of
https://github.com/psycopg/psycopg2.git
synced 2024-11-11 19:46:36 +03:00
fa032f09fb
If the network is down, trying to read blocking will hang the process hard (ctrl-c not working). Send a cancel signal instead (as suggested in http://archives.postgresql.org/pgsql-hackers/2012-07/msg00903.php) and go back into a green polling: this should allow a further error (e.g. another ctrl-c) to break the loop. In this case we cannot assume anything about the state of the connection, so we close it.
49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
#!/usr/bin/env python
|
|
"""Test for issue #113: test with error during green processing
|
|
"""
|
|
|
|
DSN = 'dbname=test'
|
|
|
|
# import eventlet.patcher
|
|
# eventlet.patcher.monkey_patch()
|
|
|
|
import os
|
|
import signal
|
|
import psycopg2
|
|
from psycopg2 import extensions
|
|
from eventlet.hubs import trampoline
|
|
|
|
panic = []
|
|
|
|
def wait_cb(conn):
|
|
"""A wait callback useful to allow eventlet to work with Psycopg."""
|
|
while 1:
|
|
if panic:
|
|
raise Exception('whatever')
|
|
|
|
state = conn.poll()
|
|
if state == extensions.POLL_OK:
|
|
break
|
|
elif state == extensions.POLL_READ:
|
|
trampoline(conn.fileno(), read=True)
|
|
elif state == extensions.POLL_WRITE:
|
|
trampoline(conn.fileno(), write=True)
|
|
else:
|
|
raise psycopg2.OperationalError(
|
|
"Bad result from poll: %r" % state)
|
|
|
|
extensions.set_wait_callback(wait_cb)
|
|
|
|
def handler(signum, frame):
|
|
panic.append(True)
|
|
|
|
signal.signal(signal.SIGHUP, handler)
|
|
|
|
conn = psycopg2.connect(DSN)
|
|
curs = conn.cursor()
|
|
print "PID", os.getpid()
|
|
curs.execute("select pg_sleep(1000)")
|
|
|
|
# You can unplug the network cable etc. here.
|
|
# Kill -HUP will raise an exception in the callback.
|