diff options
Diffstat (limited to 'benchmarks/py-locust/files/extra-EXAMPLES-semaphore_wait.py')
-rw-r--r-- | benchmarks/py-locust/files/extra-EXAMPLES-semaphore_wait.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/benchmarks/py-locust/files/extra-EXAMPLES-semaphore_wait.py b/benchmarks/py-locust/files/extra-EXAMPLES-semaphore_wait.py new file mode 100644 index 000000000000..563b89c95d63 --- /dev/null +++ b/benchmarks/py-locust/files/extra-EXAMPLES-semaphore_wait.py @@ -0,0 +1,25 @@ +from locust import HttpLocust, TaskSet, task, events + +from gevent.coros import Semaphore +all_locusts_spawned = Semaphore() +all_locusts_spawned.acquire() + +def on_hatch_complete(**kw): + all_locusts_spawned.release() + +events.hatch_complete += on_hatch_complete + +class UserTasks(TaskSet): + def on_start(self): + all_locusts_spawned.wait() + self.wait() + + @task + def index(self): + self.client.get("/") + +class WebsiteUser(HttpLocust): + host = "http://127.0.0.1:8089" + min_wait = 2000 + max_wait = 5000 + task_set = UserTasks |