25개 이상의 토픽을 선택하실 수 없습니다. Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

65 lines
2.0KB

  1. import errno
  2. import sys
  3. from eventlet import patcher, support
  4. from eventlet.hubs import hub
  5. select = patcher.original('select')
  6. time = patcher.original('time')
  7. try:
  8. BAD_SOCK = set((errno.EBADF, errno.WSAENOTSOCK))
  9. except AttributeError:
  10. BAD_SOCK = set((errno.EBADF,))
  11. def is_available():
  12. return hasattr(select, 'select')
  13. class Hub(hub.BaseHub):
  14. def _remove_bad_fds(self):
  15. """ Iterate through fds, removing the ones that are bad per the
  16. operating system.
  17. """
  18. all_fds = list(self.listeners[self.READ]) + list(self.listeners[self.WRITE])
  19. for fd in all_fds:
  20. try:
  21. select.select([fd], [], [], 0)
  22. except select.error as e:
  23. if support.get_errno(e) in BAD_SOCK:
  24. self.remove_descriptor(fd)
  25. def wait(self, seconds=None):
  26. readers = self.listeners[self.READ]
  27. writers = self.listeners[self.WRITE]
  28. if not readers and not writers:
  29. if seconds:
  30. time.sleep(seconds)
  31. return
  32. reader_fds = list(readers)
  33. writer_fds = list(writers)
  34. all_fds = reader_fds + writer_fds
  35. try:
  36. r, w, er = select.select(reader_fds, writer_fds, all_fds, seconds)
  37. except select.error as e:
  38. if support.get_errno(e) == errno.EINTR:
  39. return
  40. elif support.get_errno(e) in BAD_SOCK:
  41. self._remove_bad_fds()
  42. return
  43. else:
  44. raise
  45. for fileno in er:
  46. readers.get(fileno, hub.noop).cb(fileno)
  47. writers.get(fileno, hub.noop).cb(fileno)
  48. for listeners, events in ((readers, r), (writers, w)):
  49. for fileno in events:
  50. try:
  51. listeners.get(fileno, hub.noop).cb(fileno)
  52. except self.SYSTEM_EXCEPTIONS:
  53. raise
  54. except:
  55. self.squelch_exception(fileno, sys.exc_info())
  56. support.clear_sys_exc_info()