1. taskset 包含def 任务函数
from locust import HttpLocust, TaskSet
def job1(l):
l.client.get('/')
print('this is huifu')
def job2(l):
l.client.get('/')
print('i am 20')
class UserBehavior(TaskSet):
tasks = {job1:1, job2:2}
class WebsiteUser(HttpLocust):
task_set = UserBehavior
min_wait = 3000
max_wait = 6000
2. taskset 包含taskset 任务
from locust import Locust, TaskSet, task
class my_task(TaskSet):
@task
def print_name(self,name='huifu'):
print('This is %s'%name)
@task
def print_add(self,addr='wuxi'):
print('i am in %s'%addr)
class qin_task(TaskSet):
@task
def print_age(self,age=20):
print('i am %i'%age)
class MyTaskSet(TaskSet):
tasks = {my_task:1,qin_task:1}
class MyLocust(Locust):
task_set = MyTaskSet
max_wait = 1000
min_wait = 1000