locust 2.43.4.dev19__tar.gz → 2.43.4.dev25__tar.gz

{locust-2.43.4.dev19 → locust-2.43.4.dev25}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: locust
-Version: 2.43.4.dev19
+Version: 2.43.4.dev25
 Summary: Developer-friendly load testing framework
 Project-URL: homepage, https://locust.io/
 Project-URL: repository, https://github.com/locustio/locust

{locust-2.43.4.dev19 → locust-2.43.4.dev25}/locust/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '2.43.4.dev19'
-__version_tuple__ = version_tuple = (2, 43, 4, 'dev19')
+__version__ = version = '2.43.4.dev25'
+__version_tuple__ = version_tuple = (2, 43, 4, 'dev25')
 
 __commit_id__ = commit_id = None

{locust-2.43.4.dev19 → locust-2.43.4.dev25}/locust/stats.py

@@ -119,6 +119,27 @@ PERCENTILES_TO_STATISTICS = [0.95, 0.99]
 PERCENTILES_TO_CHART = [0.5, 0.95]
 
 
+def bucket_response_time(response_time: int | float) -> int:
+    """Round response time to reduce unique histogram keys.
+
+    Rounds to ~2 significant digits, so that 147 becomes 150, 3432 becomes 3400
+    and 58760 becomes 59000. This limits the dict to ~310 unique keys, which is
+    important for bandwidth in distributed mode.
+
+    This is a module-level function so it can be replaced at runtime for custom
+    bucketing, e.g. rounding to 3 significant figures or using fixed-width bins.
+    See :ref:`customizing-response-time-bucketing` in the docs.
+    """
+    if response_time < 100:
+        return round(response_time)
+    elif response_time < 1000:
+        return int(round(response_time, -1))
+    elif response_time < 10000:
+        return int(round(response_time, -2))
+    else:
+        return int(round(response_time, -3))
+
+
 class RequestStatsAdditionError(Exception):
     pass
 
@@ -381,14 +402,7 @@ class StatsEntry:
         # to avoid to much data that has to be transferred to the master node when
         # running in distributed mode, we save the response time rounded in a dict
         # so that 147 becomes 150, 3432 becomes 3400 and 58760 becomes 59000
-        if response_time < 100:
-            rounded_response_time = round(response_time)
-        elif response_time < 1000:
-            rounded_response_time = round(response_time, -1)
-        elif response_time < 10000:
-            rounded_response_time = round(response_time, -2)
-        else:
-            rounded_response_time = round(response_time, -3)
+        rounded_response_time = bucket_response_time(response_time)
 
         # increase request count for the rounded key in response time dict
         self.response_times[rounded_response_time] += 1