locust 2.31.6.dev25__tar.gz → 2.31.7.dev6__tar.gz

{locust-2.31.6.dev25 → locust-2.31.7.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust
-Version: 2.31.6.dev25
+Version: 2.31.7.dev6
 Summary: Developer-friendly load testing framework
 Home-page: https://locust.io/
 License: MIT

{locust-2.31.6.dev25 → locust-2.31.7.dev6}/locust/_version.py

@@ -14,7 +14,7 @@ __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
 
-__version__ = "2.31.6.dev25"
+__version__ = "2.31.7.dev6"
 version = __version__
-__version_tuple__ = (2, 31, 6, "dev25")
+__version_tuple__ = (2, 31, 7, "dev6")
 version_tuple = __version_tuple__

locust-2.31.7.dev6/locust/contrib/mongodb.py

@@ -0,0 +1,41 @@
+from locust import User, events
+
+import time
+
+from pymongo import MongoClient
+from pymongo.errors import PyMongoError
+
+
+class MongoDBClient(MongoClient):
+    def __init__(self, conn_string, db_name):
+        super().__init__(conn_string)
+        self.db = self.client[db_name]
+
+    def execute_query(self, collection_name, query):
+        start_time = time.time()
+        try:
+            collection = self.db[collection_name]
+            collection.find(query)
+
+            response_time = int((time.time() - start_time) * 1000)
+            events.request.fire(request_type="MONGODB", name="QUERY", response_time=response_time, response_length=0)
+        except PyMongoError as e:
+            response_time = int((time.time() - start_time) * 1000)
+            events.request.fire(
+                request_type="MONGODB",
+                name="QUERY",
+                response_time=response_time,
+                response_length=0,
+                exception=e,
+            )
+
+
+class MongoDBUser(User):
+    abstract = True
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.client = MongoDBClient(conn_string=self.conn_string, db_name=self.db_name)
+
+    def on_stop(self):
+        self.client.close()

{locust-2.31.6.dev25 → locust-2.31.7.dev6}/locust/main.py

@@ -199,6 +199,8 @@ def main():
             sys.exit(1)
 
     children = []
+    logger = logging.getLogger(__name__)
+    logger.info(f"Starting Locust {version}")
 
     if options.processes:
         if os.name == "nt":
@@ -298,7 +300,6 @@ def main():
 
                 atexit.register(kill_workers, children)
 
-    logger = logging.getLogger(__name__)
     greenlet_exception_handler = greenlet_exception_logger(logger)
 
     if options.stop_timeout:
@@ -686,7 +687,6 @@ See https://github.com/locustio/locust/wiki/Installation#increasing-maximum-numb
     gevent.signal_handler(signal.SIGTERM, sig_term_handler)
 
     try:
-        logger.info(f"Starting Locust {version}")
         if options.class_picker:
             logger.info("Locust is running with the UserClass Picker Enabled")
         if options.autostart and not options.headless:

{locust-2.31.6.dev25 → locust-2.31.7.dev6}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry_dynamic_versioning.backend"
 [tool.poetry]
 name = "locust"
 description = "Developer-friendly load testing framework"
-version = "2.31.6.dev25"
+version = "2.31.7.dev6"
 license = "MIT"
 readme = "README.md"
 authors = ["Jonatan Heyman", "Lars Holmberg"]