GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-05-12 02:21:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

bugfix: start locust slave on other machines

Browse Source
This commit is contained in:
debugtalk
2018-01-16 16:30:24 +08:00
parent 389e915a09
commit 25539414a2
1 changed files with 10 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
httprunner/locusts.py
View File

@@ -54,12 +54,16 @@ def start_master(sys_argv):
main()
def start_slave(sys_argv):
sys_argv.extend(["--slave"])
if "--slave" not in sys_argv:
sys_argv.extend(["--slave"])
sys.argv = sys_argv
main()
def run_locusts_on_cpu_cores(sys_argv, cpu_cores_num_value):
processes = []
manager = multiprocessing.Manager()
for _ in range(cpu_cores_num_value):
p_slave = multiprocessing.Process(target=start_slave, args=(sys_argv,))
p_slave.daemon = True
@@ -67,6 +71,9 @@ def run_locusts_on_cpu_cores(sys_argv, cpu_cores_num_value):
processes.append(p_slave)
try:
start_master(sys_argv)
if "--slave" in sys_argv:
[process.join() for process in processes]
else:
start_master(sys_argv)
except KeyboardInterrupt:
sys.exit(0)
manager.shutdown()