fix: worker disconnections when urllib3>=1.26.16

locustio · Jul 22, 2024 · 593fb54 · 593fb54
1 parent c4b6571
commit 593fb54
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/locust/__init__.py b/locust/__init__.py
@@ -10,11 +10,17 @@
         # dont show a massive callstack if trio is not installed
         os._exit(1)
 
-from gevent import monkey
-
 if not os.getenv("LOCUST_SKIP_MONKEY_PATCH", None):
+    from gevent import monkey, queue
+
     monkey.patch_all()
 
+    if not os.getenv("LOCUST_SKIP_URLLIB3_PATCH", None):
+        import urllib3
+
+        urllib3.connectionpool.ConnectionPool.QueueCls = queue.LifoQueue
+        # https://github.com/locustio/locust/issues/2812
+
 from ._version import version as __version__
 from .contrib.fasthttp import FastHttpUser
 from .debug import run_single_user