1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
|
from __future__ import absolute_import
import select
import socket
try:
from eventlet.patcher import is_monkey_patched as is_eventlet
except ImportError:
is_eventlet = lambda module: False # noqa
POLL_READ = 0x001
POLL_ERR = 0x008 | 0x010 | 0x2000
class _epoll(object):
def __init__(self):
self._epoll = select.epoll()
def register(self, fd, events):
self._epoll.register(fd, events)
def unregister(self, fd):
try:
self._epoll.unregister(fd)
except socket.error:
pass
def poll(self, timeout):
return self._epoll.poll(timeout and timeout / 1000.0 or -1)
class _kqueue(object):
def __init__(self):
self._kqueue = select.kqueue()
self._active = {}
def register(self, fd, events):
self._control(fd, events, select.KQ_EV_ADD)
self._active[fd] = events
def unregister(self, fd):
events = self._active.pop(fd)
try:
self._control(fd, events, select.KQ_EV_DELETE)
except socket.error:
pass
def _control(self, fd, events, flags):
self._kqueue.control([select.kevent(fd, filter=select.KQ_FILTER_READ,
flags=flags)], 0)
def poll(self, timeout):
kevents = self._kqueue.control(None, 1000,
timeout and timeout / 1000.0 or timeout)
events = {}
for kevent in kevents:
fd = kevent.ident
if kevent.filter == select.KQ_FILTER_READ:
events[fd] = events.get(fd, 0) | POLL_READ
if kevent.filter == select.KQ_EV_ERROR:
events[fd] = events.get(fd, 0) | POLL_ERR
return events.items()
class _select(object):
def __init__(self):
self._all = self._rfd, self._efd = set(), set()
def register(self, fd, events):
if events & POLL_ERR:
self._efd.add(fd)
self._rfd.add(fd)
elif events & POLL_READ:
self._rfd.add(fd)
def unregister(self, fd):
self._rfd.discard(fd)
self._efd.discard(fd)
def poll(self, timeout):
read, _write, error = select.select(self._rfd, [], self._efd, timeout)
events = {}
for fd in read:
fd = fd.fileno()
events[fd] = events.get(fd, 0) | POLL_READ
for fd in error:
fd = fd.fileno()
events[fd] = events.get(fd, 0) | POLL_ERR
return events.items()
if is_eventlet(select):
# use Eventlet's non-blocking version of select.select
poll = _select
elif hasattr(select, "epoll"):
# Py2.6+ Linux
poll = _epoll
elif hasattr(select, "kqueue"):
# Py2.6+ on BSD / Darwin
poll = _kqueue
else:
poll = _select
|
