Only return from muxer on error or timeout

1 files changed, 9 insertions, 8 deletions

diff --git a/io_muxer.c b/io_muxer.c
index 31ebc53..3e6514e 100644
--- a/io_muxer.c
+++ b/io_muxer.c
@@ -84,15 +84,16 @@ int IoMuxer_OnWrite(struct IoMuxer* io_muxer, int fd,
   return 0;
 }
 
-int IoMuxer_Iterate(struct IoMuxer* io_muxer) {
-  CompressRecords(io_muxer);
-  int result = poll(io_muxer->pfds, io_muxer->size, -1);
-  if (result <= 0) return -1;
-  for (size_t i = 0; i < io_muxer->size; i++) {
-    if (io_muxer->pfds[i].revents)
-      io_muxer->closures[i].callback(io_muxer->closures[i].user);
+int IoMuxer_Iterate(struct IoMuxer* io_muxer, int timeout) {
+  for (;;) {
+    CompressRecords(io_muxer);
+    int result = poll(io_muxer->pfds, io_muxer->size, timeout);
+    if (result <= 0) return result;
+    for (size_t i = 0; i < io_muxer->size; i++) {
+      if (io_muxer->pfds[i].revents)
+        io_muxer->closures[i].callback(io_muxer->closures[i].user);
+    }
   }
-  return 0;
 }
 
 void IoMuxer_Destroy(struct IoMuxer* io_muxer) {