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