Add PostgresUser to examples

examples/database/README.md

@@ -3,29 +3,19 @@
 Read the instruction below for your specific database
 
 ## PostgreSQL
-### How to run the test
-- Prerequisites:
-    If you do not have them installed already
-    - `pip3 install psycopg3` - https://www.psycopg.org/psycopg3/docs/basic/install.html
-
-    - `pip3 install locust` - https://docs.locust.io/en/stable/installation.html
 
-- Set your environment variables for: 
-    - PGHOST
-    - PGPORT
-    - PGDATABASE
-    - PGUSER
-    - PGPASSWORD
-
-    so they can be picked up from `postgres_test.py`
+### How to run the test
 
-- Also, update the methods of the UserTasks class to run queries that will hit your database.
+- Prerequisites:
 
-#### Run test from browser:
-- Open the terminal and run `locust -f postgres_test.py` this will give you a URL to follow. Likely `http://0.0.0.0:8089` from here, you can put in custom parameters to run your test
+  - `psycopg3` - https://www.psycopg.org/psycopg3/docs/basic/install.html
 
-#### Run test from terminal:
+- Set your environment variables for:
 
-- Insert this into your terminal `locust -f name_of_test_file.py --users=500 --spawn-rate=10 --run-time='15s' --autostart --autoquit 3`
-- For a full list of all the configuration options that can be used run `locust --help` and visit https://docs.locust.io/en/stable/configuration.html#configuration
+  - PGHOST
+  - PGPORT
+  - PGDATABASE
+  - PGUSER
+  - PGPASSWORD
 
+- Run locust as usual, see https://docs.locust.io/en/stable/quickstart.html

examples/database/postgresql/postgres_test.py

@@ -1,10 +1,9 @@
 from locust import TaskSet, between, task
+from locust.contrib.postgres import PostgresUser
 
 import os
 import random
 
-from base_postgres_user import PostgresUser
-
 
 class UserTasks(TaskSet):
     @task

examples/database/postgresql/base_postgres_user.py → locust/contrib/postgres.py

@@ -43,7 +43,7 @@ def request_handler(*args, **kwargs):
 
 class PostgresUser(User):
     abstract = True
-    client_class = PostgresClient
 
     def __init__(self, *args, **kwargs):
-        self.client = self.client_class()
+        super().__init__(*args, **kwargs)
+        self.client = PostgresClient()