locust 2.31.3.dev32__tar.gz → 2.31.4__tar.gz

{locust-2.31.3.dev32 → locust-2.31.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: locust
-Version: 2.31.3.dev32
+Version: 2.31.4
 Summary: Developer-friendly load testing framework
 Home-page: https://locust.io/
 License: MIT

{locust-2.31.3.dev32 → locust-2.31.4}/locust/_version.py

@@ -14,7 +14,7 @@ __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
 
-__version__ = "2.31.3.dev32"
+__version__ = "2.31.4"
 version = __version__
-__version_tuple__ = (2, 31, 3, "dev32")
+__version_tuple__ = (2, 31, 4)
 version_tuple = __version_tuple__

locust-2.31.4/locust/contrib/postgres.py

@@ -0,0 +1,45 @@
+from locust import User, events
+
+import time
+
+import psycopg
+
+
+class PostgresClient:
+    def __init__(self, conn_string):
+        self.connection = psycopg.connect(conn_string)
+
+    def execute_query(self, query):
+        start_time = time.time()
+        try:
+            self.connection.execute(query)
+            response_time = int((time.time() - start_time) * 1000)
+            events.request.fire(
+                request_type="postgres_success",
+                name="execute_query",
+                response_time=response_time,
+                response_length=0,
+            )
+        except Exception as e:
+            response_time = int((time.time() - start_time) * 1000)
+            events.request.fire(
+                request_type="postgres_failure",
+                name="execute_query",
+                response_time=response_time,
+                response_length=0,
+                exception=e,
+            )
+
+    def close(self):
+        self.connection.close()
+
+
+class PostgresUser(User):
+    abstract = True
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.client = PostgresClient(conn_string=self.conn_string)
+
+    def on_stop(self):
+        self.client.close()

{locust-2.31.3.dev32 → locust-2.31.4}/locust/stats.py

@@ -919,6 +919,7 @@ def update_stats_history(runner: Runner) -> None:
         "current_fail_per_sec": [timestamp, stats.total.current_fail_per_sec or 0],
         "total_avg_response_time": [timestamp, proper_round(stats.total.avg_response_time, digits=2)],
         "user_count": [timestamp, runner.user_count or 0],
+        "time": timestamp,
     }
     stats.history.append(r)
 

{locust-2.31.3.dev32 → locust-2.31.4}/locust/web.py

@@ -618,8 +618,6 @@ class WebUI:
             else None
         )
 
-        start_time = format_utc_timestamp(stats.start_time)
-
         self.template_args = {
             "locustfile": self.environment.locustfile,
             "state": self.environment.runner.state,
@@ -637,7 +635,6 @@ class WebUI:
                 and not (self.userclass_picker_is_active or self.environment.shape_class.use_common_options)
             ),
             "stats_history_enabled": options and options.stats_history_enabled,
-            "start_time": start_time,
             "tasks": dumps({}),
             "extra_options": extra_options,
             "run_time": options and options.run_time,