Locust 压测socket接口样例

import socket
from locust import User, task, between, constant

class SocketUser(User):
# 初始化，建立Socket连接
def on_start(self):
self.client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
self.client.connect(('localhost', 8080)) # 替换为目标Socket服务器的地址和端口

# 结束测试，关闭Socket连接
def on_stop(self):
    self.client.close()

# 编写自定义的发送和接收数据的方法
def send_and_receive(self, data):
    self.client.sendall(data.encode())
    response = self.client.recv(1024).decode()
    return response

# 定义一个任务，模拟发送数据并接收响应
@task
def test_socket_api(self):
    request_data = "Your Socket Request Data"  # 替换为实际的请求数据
    response = self.send_and_receive(request_data)
    # 在此可以根据响应进行断言、处理等操作，根据需求自行扩展

设置性能参数

class SocketUser(SocketUser):
wait_time = constant(0) # 每个任务之间不等待（可以根据需要调整）

运行命令：locust -f your_script_name.py --host=http://your_socket_server

然后访问 http://localhost:8089 进行测试