Fix bug with get_records retry mentioned in issue NerdWalletOSS#31

Crowesesse · Jul 30, 2024 · 5878e16 · 5878e16
1 parent 1a238c8
commit 5878e16
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/kinesis/consumer.py b/src/kinesis/consumer.py
@@ -50,9 +50,10 @@ def loop(self):
                 # until we complete a successful get_records call
                 loop_status = min((
                     30,
-                    (self.retries or 1) * 2
+                    (self.retries * 2) or 1
                 ))
                 log.debug("Retrying get_records (#%d %ds): %s", self.retries+1, loop_status, exc)
+                self.retries += 1
             else:
                 log.error("Client error occurred while reading: %s", exc)
                 loop_status = False