Locust taskset 组织支持多种形式

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

猜你喜欢

转载自blog.csdn.net/warm77/article/details/80264555