Make kqueue code more similar to epoll code

python-trio · Jun 2, 2020 · f1db80b · f1db80b
1 parent c6697db
commit f1db80b
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/trio/_core/_io_kqueue.py b/trio/_core/_io_kqueue.py
@@ -29,6 +29,7 @@ def __attrs_post_init__(self):
  self._force_wakeup.wakeup_sock, select.KQ_FILTER_READ, select.KQ_EV_ADD
  )
  self._kqueue.control([force_wakeup_event], 0)
+ self._force_wakeup_fd = self._force_wakeup.wakeup_sock.fileno()
 
  def statistics(self):
  tasks_waiting = 0
@@ -66,14 +67,10 @@ def get_events(self, timeout):
  def process_events(self, events):
  for event in events:
  key = (event.ident, event.filter)
- try:
- receiver = self._registered[key]
- except KeyError:
- if event.ident == self._force_wakeup.wakeup_sock.fileno():
- self._force_wakeup.drain()
- continue
- else: # pragma: no cover
- raise
+ if event.ident == self._force_wakeup_fd:
+ self._force_wakeup.drain()
+ continue
+ receiver = self._registered[key]
  if event.flags & select.KQ_EV_ONESHOT:
  del self._registered[key]
  if type(receiver) is _core.Task: