60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
import sys
|
|
import errno
|
|
from eventlet import patcher
|
|
select = patcher.original('select')
|
|
time = patcher.original('time')
|
|
|
|
from eventlet.hubs.hub import BaseHub, READ, WRITE
|
|
|
|
try:
|
|
BAD_SOCK = set((errno.EBADF, errno.WSAENOTSOCK))
|
|
except AttributeError:
|
|
BAD_SOCK = set((errno.EBADF,))
|
|
|
|
class Hub(BaseHub):
|
|
def _remove_bad_fds(self):
|
|
""" Iterate through fds, removing the ones that are bad per the
|
|
operating system.
|
|
"""
|
|
for fd in self.listeners[READ].keys() + self.listeners[WRITE].keys():
|
|
try:
|
|
select.select([fd], [], [], 0)
|
|
except select.error, e:
|
|
if e.args[0] == errno.EBADF:
|
|
self.remove_descriptor(fd)
|
|
|
|
def wait(self, seconds=None):
|
|
readers = self.listeners[READ]
|
|
writers = self.listeners[WRITE]
|
|
if not readers and not writers:
|
|
if seconds:
|
|
time.sleep(seconds)
|
|
return
|
|
try:
|
|
r, w, er = select.select(readers.keys(), writers.keys(), readers.keys() + writers.keys(), seconds)
|
|
except select.error, e:
|
|
if e.args[0] == errno.EINTR:
|
|
return
|
|
elif e.args[0] in BAD_SOCK:
|
|
self._remove_bad_fds()
|
|
return
|
|
else:
|
|
raise
|
|
|
|
for fileno in er:
|
|
for reader in readers.get(fileno, ()):
|
|
reader(fileno)
|
|
for writer in writers.get(fileno, ()):
|
|
writer(fileno)
|
|
|
|
for listeners, events in ((readers, r), (writers, w)):
|
|
for fileno in events:
|
|
try:
|
|
l_list = listeners[fileno]
|
|
if l_list:
|
|
l_list[0](fileno)
|
|
except self.SYSTEM_EXCEPTIONS:
|
|
raise
|
|
except:
|
|
self.squelch_exception(fileno, sys.exc_info())
|