Add "stopping" state. Fixes issue of load rebalancing and restarting …

…when trying to stop a distributed load and slaves rejoin.
locustio · Mar 19, 2019 · 8b17c28 · 8b17c28 · SteggyLeggy · Jun 14, 2019
1 parent 68ac9d3
commit 8b17c28
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/locust/__init__.py b/locust/__init__.py
@@ -1,4 +1,4 @@
 from .core import HttpLocust, Locust, TaskSet, TaskSequence, task, seq_task
 from .exception import InterruptTaskSet, ResponseError, RescheduleTaskImmediately
 
-__version__ = "0.11.0"
+__version__ = "0.11.1"
diff --git a/locust/runners.py b/locust/runners.py
@@ -23,7 +23,7 @@
 # global locust runner singleton
 locust_runner = None
 
-STATE_INIT, STATE_HATCHING, STATE_RUNNING, STATE_CLEANUP, STATE_STOPPED, STATE_MISSING = ["ready", "hatching", "running", "cleanup", "stopped", "missing"]
+STATE_INIT, STATE_HATCHING, STATE_RUNNING, STATE_CLEANUP, STATE_STOPPING, STATE_STOPPED, STATE_MISSING = ["ready", "hatching", "running", "cleanup", "stopping", "stopped", "missing"]
 SLAVE_REPORT_INTERVAL = 3.0
 
 
@@ -313,6 +313,7 @@ def start_hatching(self, locust_count, hatch_rate):
         self.state = STATE_HATCHING
 
     def stop(self):
+        self.state = STATE_STOPPING
         for client in self.clients.all:
             self.server.send_to_client(Message("stop", None, client.id))
         events.master_stop_hatching.fire()