Fix issue where flusher task stops running

Fixes issue #624 A cancelled future in the flush_queue can cause the flusher task to fail with InvalidStateError, due to attempting to set the result on a "done" future. Signed-off-by: Debby Mendez <[email protected]>
nats-io · Nov 19, 2024 · 97bb402 · 97bb402
1 parent 7e7883e
commit 97bb402
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/nats/aio/client.py b/nats/aio/client.py
@@ -2084,7 +2084,9 @@ async def _flusher(self) -> None:
                 # RuntimeError in case the event loop is closed
                 break
             finally:
-                future.set_result(None)
+                # future might have been cancelled.  See issue #624
+                if not future.done():
+                    future.set_result(None)
 
     async def _ping_interval(self) -> None:
         while True: