asdnotify: fix high CPU usage

Snawoot · Jan 26, 2020 · 88999db · 88999db
1 parent 6ef4606
commit 88999db
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/postfix_mta_sts_resolver/asdnotify.py b/postfix_mta_sts_resolver/asdnotify.py
@@ -17,6 +17,7 @@ def __init__(self):
         self._started = False
         self._loop = None
         self._queue = asyncio.Queue(MAX_QLEN)
+        self._monitor = False
 
     @property
     def started(self):
@@ -28,8 +29,11 @@ def _drain(self):
                 msg = self._queue.get_nowait()
                 self._queue.task_done()
                 self._send(msg)
+            if self._monitor:
+                self._loop.remove_writer(self._sock.fileno())
         except BlockingIOError:  # pragma: no cover
-            pass
+            self._monitor = True
+            self._loop.add_writer(self._sock.fileno(), self._drain)
         except OSError:
             pass
 
@@ -43,7 +47,6 @@ async def start(self):
         try:
             self._sock = socket.socket(socket.AF_UNIX, socket.SOCK_DGRAM)
             self._sock.setblocking(0)
-            self._loop.add_writer(self._sock.fileno(), self._drain)
             self._started = True
         except OSError:
             return False