Consider using isolation_level="IMMEDIATE" for write connections about datasette HOT 5 OPEN

from locust import HttpUser, task, between
import random
import json
import os

class APIUser(HttpUser):
    host = "http://localhost:8001"
    # wait_time = between(0.05, 0.1) # in seconds

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.id_counter = 0
        self.api_key = os.environ.get('DS_API_KEY')
        if not self.api_key:
            raise ValueError("DS_API_KEY environment variable is not set")
        self.headers = {
            'Content-Type': 'application/json',
            'Authorization': f'Bearer {self.api_key}'
        }

    @task(3)
    def get_data(self):
        self.client.get("/data", headers=self.headers)

    @task(3)
    def get_mytable(self):
        self.client.get("/data/mytable?_sort_desc=id", headers=self.headers)

    @task(2)
    def get_mytable_with_id(self):
        self.client.get("/data/mytable?id=5", headers=self.headers)

    @task(4)
    def post_create_data(self):
        self.id_counter += 1
        payload = {
            "table": "mytable",
            "rows": [
                {
                    "id": self.id_counter,
                    "name": random.choice(["Tarantula", "Black Widow", "Huntsman", "Wolf Spider", "Jumping Spider"])
                }
            ],
            "pk": "id",
            "replace": True
        }
        self.client.post("/data/-/create", data=json.dumps(payload), headers=self.headers)