locust 2.41.2.dev6__tar.gz → 2.41.3__tar.gz

{locust-2.41.2.dev6 → locust-2.41.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: locust
-Version: 2.41.2.dev6
+Version: 2.41.3
 Summary: Developer-friendly load testing framework
 Project-URL: homepage, https://locust.io/
 Project-URL: repository, https://github.com/locustio/locust
@@ -30,7 +30,7 @@ Requires-Dist: flask-login>=0.6.3
 Requires-Dist: flask>=2.0.0
 Requires-Dist: gevent<25.8.1,>=24.10.1
 Requires-Dist: geventhttpclient>=2.3.1
-Requires-Dist: locust-cloud>=1.27.1
+Requires-Dist: locust-cloud>=1.27.2
 Requires-Dist: msgpack>=1.0.0
 Requires-Dist: psutil>=5.9.1
 Requires-Dist: pytest<9.0.0,>=8.3.3

{locust-2.41.2.dev6 → locust-2.41.3}/locust/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '2.41.2.dev6'
-__version_tuple__ = version_tuple = (2, 41, 2, 'dev6')
+__version__ = version = '2.41.3'
+__version_tuple__ = version_tuple = (2, 41, 3)
 
 __commit_id__ = commit_id = None

{locust-2.41.2.dev6 → locust-2.41.3}/locust/contrib/milvus.py

@@ -21,7 +21,7 @@ class BaseClient(ABC):
         pass
 
     @abstractmethod
-    def create_collection(self, schema, index_params) -> None:
+    def create_collection(self, schema, index_params, **kwargs) -> None:
         pass
 
     @abstractmethod
@@ -80,11 +80,12 @@ class MilvusV2Client(BaseClient):
     def close(self):
         self.client.close()
 
-    def create_collection(self, schema, index_params):
+    def create_collection(self, schema, index_params, **kwargs):
         self.client.create_collection(
             collection_name=self.collection_name,
             schema=schema,
             index_params=index_params,
+            **kwargs,
         )
 
     def insert(self, data):
@@ -299,7 +300,7 @@ class MilvusUser(User):
         timeout: int = 60,
         schema: CollectionSchema | None = None,
         index_params: IndexParams | None = None,
-        **client_kwargs,
+        **kwargs,  # enable_dynamic_field, num_shards, consistency_level etc. ref: https://milvus.io/api-reference/pymilvus/v2.6.x/MilvusClient/Collections/create_collection.md
     ):
         super().__init__(environment)
 
@@ -317,7 +318,7 @@ class MilvusUser(User):
             timeout=timeout,
         )
         if schema is not None:
-            self.client.create_collection(schema=schema, index_params=index_params)
+            self.client.create_collection(schema=schema, index_params=index_params, **kwargs)
 
     @staticmethod
     def _fire_event(request_type: str, name: str, result: dict[str, Any]):

{locust-2.41.2.dev6 → locust-2.41.3}/locust/main.py

@@ -584,11 +584,11 @@ See https://github.com/locustio/locust/wiki/Installation#increasing-maximum-numb
                 headless_master_greenlet = gevent.spawn(runner.start, options.num_users, options.spawn_rate)
                 headless_master_greenlet.link_exception(greenlet_exception_handler)
 
-        if options.run_time:
-            logger.info(f"Run time limit set to {options.run_time} seconds")
-            spawn_run_time_quit_greenlet()
-        elif not options.worker and not environment.shape_class:
-            logger.info("No run time limit set, use CTRL+C to interrupt")
+            if options.run_time:
+                logger.info(f"Run time limit set to {options.run_time} seconds")
+                spawn_run_time_quit_greenlet()
+            elif not environment.shape_class:
+                logger.info("No run time limit set, use CTRL+C to interrupt")
 
     if options.csv_prefix:
         gevent.spawn(stats_csv_writer.stats_writer).link_exception(greenlet_exception_handler)

{locust-2.41.2.dev6 → locust-2.41.3}/locust/runners.py

@@ -1434,9 +1434,9 @@ class WorkerRunner(DistributedRunner):
         while True:
             current_logs = get_logs()
 
-            if (len(current_logs) - len(self.logs)) > 10:
+            if (len(current_logs) - len(self.logs)) > 70:
                 logger.warning(
-                    "The worker attempted to send more than 10 log lines in one interval. Further log sending was disabled for this worker."
+                    "The worker attempted to send more than 70 log lines in one interval. Further log sending was disabled for this worker."
                 )
                 self._send_logs(get_logs())
                 break
@@ -1474,12 +1474,12 @@ class WorkerRunner(DistributedRunner):
             # dont complain about getting CTRL-C
             sys.exit(1)
         if not success:
-            if self.retry < 3:
+            if self.retry < 30 / CONNECT_TIMEOUT:  # lower log level during the first 30 seconds
                 logger.debug(
                     f"Failed to connect to master {self.master_host}:{self.master_port}{self.web_base_path}, retry {self.retry}/{CONNECT_RETRY_COUNT}."
                 )
             else:
-                logger.warning(
+                logger.info(
                     f"Failed to connect to master {self.master_host}:{self.master_port}{self.web_base_path}, retry {self.retry}/{CONNECT_RETRY_COUNT}."
                 )
             if self.retry > CONNECT_RETRY_COUNT:

{locust-2.41.2.dev6 → locust-2.41.3}/pyproject.toml

@@ -46,7 +46,7 @@ dependencies = [
     "Flask-Cors>=3.0.10",
     "pywin32; sys_platform == 'win32'",
     "setuptools>=70.0.0",
-    "locust-cloud>=1.27.1",
+    "locust-cloud>=1.27.2",
     "gevent>=24.10.1,<25.8.1",
     "python-socketio[client]>=5.13.0",
     "python-engineio>=4.12.2",