locust给用户设置id （Kafka生产者）

确定过眼神！

已于 2023-05-26 19:45:05 修改

阅读量326

点赞数

文章标签： kafka 分布式

于 2023-05-26 19:36:44 首次发布

本文链接：https://blog.csdn.net/qq_33684240/article/details/130892947

版权

最近做的一个项目测试，模拟1000 个用户循环向kafka发送数据，代码如下：

from locust_plugins.users import KafkaUser
from locust import task, run_single_user, TaskSet
import json
import time

class MyTaskSet(TaskSet):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.user_id =None # 存储当前用户ID
        self.num = 0


    @task
    def my_task(self):
        if self.user_id is None:
            self.user_id = MyUser.current_user_id
            MyUser.current_user_id += 1
        while True:
            msg = {
                "uuid": "603F447B37D341fd88D8CBB8945EFE79",
                "callid": "NTA5YTA0NTMyZjExY2Y5Y2MxMmIwY2NiYjRmOTIxOGE.",
                "caller": str(self.user_id),
                "callee": "99999",
                "direction": 1,
                "seq": 8,
                "sendTime": 1667294609.829818,
                "spendTime": 20,
                "voiceTime": 4500,
                "new": False,
                "eof": False,
                "recPath": "",
                "timestamp": time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),
                "text": f"{str(self.user_id)}第{self.num}次说现在增值税税率是多少"
            }
            msg_str = json.dumps(msg, ensure_ascii=False)
            self.client.send("uasr-assist-events", msg_str.encode('utf-8'))
            print(f"{str(self.user_id)}第{self.num}次说现在增值税税率是多少")
            self.num += 1
            time.sleep(0.2)


class MyUser(KafkaUser):
    bootstrap_servers = '192.168.10.108:9092'
    producer_settings = {'acks': 1}
    current_user_id = 7000
    tasks = [MyTaskSet]

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)


if __name__ == "__main__":
    run_single_user(MyUser)