fix pooler so finisher is optional

2020-09-01 08:22:05 +02:00 · 2020-09-01 08:22:05 +02:00 · 9a3598b599
parent 3b31919f75
commit 9a3598b599
1 changed files with 2 additions and 1 deletions
--- a/respodns/util.py
+++ b/respodns/util.py
@ -64,7 +64,8 @@ def make_pooler(pool_size, finisher=None):
        desired_size = 0 if finish else pool_size - 1
        while len(pending) > desired_size:
            done, pending = await wait(pending, return_when=FIRST_COMPLETED)
-            finisher(done, pending)
+            if finisher is not None:
+                finisher(done, pending)

    return pooler