refactor: simplify exponential_backoff (#758)

* refactor: simplify `exponential_backoff` * Add docstrings --------- Co-authored-by: Tom Christie <tom@tomchristie.com>
encode · Aug 3, 2023 · 3cca055 · 3cca055
1 parent 29f8e90
commit 3cca055
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 4 deletions.
diff --git a/httpcore/_async/connection.py b/httpcore/_async/connection.py
@@ -21,9 +21,16 @@
 
 
 def exponential_backoff(factor: float) -> Iterator[float]:
+    """
+    Generate a geometric sequence that has a ratio of 2 and starts with 0.
+
+    For example:
+    - `factor = 2`: `0, 2, 4, 8, 16, 32, 64, ...`
+    - `factor = 3`: `0, 3, 6, 12, 24, 48, 96, ...`
+    """
     yield 0
-    for n in itertools.count(2):
-        yield factor * (2 ** (n - 2))
+    for n in itertools.count():
+        yield factor * 2**n
 
 
 class AsyncHTTPConnection(AsyncConnectionInterface):

diff --git a/httpcore/_sync/connection.py b/httpcore/_sync/connection.py
@@ -21,9 +21,16 @@
 
 
 def exponential_backoff(factor: float) -> Iterator[float]:
+    """
+    Generate a geometric sequence that has a ratio of 2 and starts with 0.
+
+    For example:
+    - `factor = 2`: `0, 2, 4, 8, 16, 32, 64, ...`
+    - `factor = 3`: `0, 3, 6, 12, 24, 48, 96, ...`
+    """
     yield 0
-    for n in itertools.count(2):
-        yield factor * (2 ** (n - 2))
+    for n in itertools.count():
+        yield factor * 2**n
 
 
 class HTTPConnection(ConnectionInterface):