Updated version of asyncore.py from Sam Rushing:

Adds support for using select.poll() if it's available

    Move a 'map is None' test out of an else branch and into the right place
This commit is contained in:
Andrew M. Kuchling 2001-01-24 15:50:19 +00:00
parent 506f0b1fc6
commit af6963c2f0
1 changed files with 40 additions and 4 deletions

View File

@ -145,16 +145,52 @@ def poll2 (timeout=0.0, map=None):
except KeyError: except KeyError:
pass pass
def poll3 (timeout=0.0, map=None):
# Use the poll() support added to the select module in Python 2.0
if map is None:
map=socket_map
# timeout is in milliseconds
timeout = int(timeout*1000)
pollster = select.poll()
if map:
l = []
for fd, obj in map.items():
flags = 0
if obj.readable():
flags = select.POLLIN
if obj.writable():
flags = flags | select.POLLOUT
if flags:
pollster.register(fd, flags)
r = pollster.poll (timeout)
for fd, flags in r:
try:
obj = map[fd]
try:
if (flags & select.POLLIN):
obj.handle_read_event()
if (flags & select.POLLOUT):
obj.handle_write_event()
except ExitNow:
raise ExitNow
except:
obj.handle_error()
except KeyError:
pass
def loop (timeout=30.0, use_poll=0, map=None): def loop (timeout=30.0, use_poll=0, map=None):
if map is None:
map=socket_map
if use_poll: if use_poll:
poll_fun = poll2 if hasattr (select, 'poll'):
poll_fun = poll3
else:
poll_fun = poll2
else: else:
poll_fun = poll poll_fun = poll
if map is None:
map=socket_map
while map: while map:
poll_fun (timeout, map) poll_fun (timeout, map)