python实现websocket的客户端压力测试
作者:刚刚登录 时间:2023-09-30 11:19:57
使用python进行websocket的客户端压力测试,这个代码是从github上 找到。然后简单修改了下。大神运用了进程池,以及线程池的内容。所以保存下来,学习学习
然后需要说明的是:本次用的python2.7,也尝试用python3.6,但是老实出现websocket-client包和python3不能兼容的情况,提示没有相关的方法。所以不得已最后又采用了python2
# -*- coding:utf-8 -*-
# __author__ == 'chenmingle'
import websocket
import time
import threading
import json
import multiprocessing
import uuid
from threadpool import ThreadPool, makeRequests
# 修改成自己的websocket地址
WS_URL = "xxxx"
# 定义进程数
processes = 4
# 定义线程数(每个文件可能限制1024个,可以修改fs.file等参数)
thread_num = 700
index = 1
def on_message(ws, message):
# print(message)
pass
def on_error(ws, error):
print(error)
pass
def on_close(ws):
# print("### closed ###")
pass
def on_open(ws):
global index
index = index + 1
def send_thread():
# 设置你websocket的内容
# 每隔10秒发送一下数据使链接不中断
while True:
ws.send(u'hello服务器')
time.sleep(10)
t = threading.Thread(target=send_thread)
t.start()
def on_start(num):
time.sleep(5)
# websocket.enableTrace(True)
ws = websocket.WebSocketApp(WS_URL + str(num),
on_message=on_message,
on_error=on_error,
on_close=on_close)
ws.on_open = on_open
ws.run_forever()
def thread_web_socket():
# 线程池
pool_list = ThreadPool(thread_num)
num = list()
# 设置开启线程的数量
for ir in range(thread_num):
num.append(ir)
requests = makeRequests(on_start, num)
[pool_list.putRequest(req) for req in requests]
pool_list.wait()
if __name__ == "__main__":
# 进程池
pool = multiprocessing.Pool(processes=processes)
# 设置开启进程的数量
for i in xrange(processes):
pool.apply_async(thread_web_socket)
pool.close()
pool.join()
来源:https://blog.csdn.net/m0_37581001/article/details/83624240
标签:python,websocket,压力测试
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
深入了解Vue3中props的原理与使用
2024-05-09 15:09:17
Golang 语言控制并发 Goroutine的方法
2024-02-14 21:18:19
windows下MySQL数据库移动到其它盘
2024-01-21 01:08:32
![](https://img.aspxhome.com/file/2023/9/101549_0s.png)
Python 分发包中添加额外文件的方法
2023-02-15 08:43:11
简单了解mysql方言dialect
2024-01-21 03:15:15
![](https://img.aspxhome.com/file/2023/8/80008_0s.png)
Python混合使用同步和异步函数的方法
2021-07-19 05:01:05
![](https://img.aspxhome.com/file/2023/6/114766_0s.gif)
恢复被删除的数据 Log Explorer for SQL Server 4.2 (一)
2010-07-01 19:24:00
![](https://img.aspxhome.com/file/UploadPic/20107/1/le001-56s.jpg)
pycharm激活码免费分享适用最新pycharm2020.2.3永久激活
2021-12-11 10:17:09
![](https://img.aspxhome.com/file/2023/5/89225_0s.png)
前端图片懒加载的原理与3种实现方式举例
2024-04-17 10:20:02
![](https://img.aspxhome.com/file/2023/1/136311_0s.png)
NodeJs读取JSON文件格式化时的注意事项
2024-05-02 17:37:03
C#从数据库读取图片并保存的两种方法
2024-01-12 14:39:22
PHP中类静态调用和范围解析操作符的区别解析
2023-11-15 07:38:39
Python Celery异步任务队列使用方法解析
2023-05-18 02:34:53
通过案例解析python鸭子类型相关原理
2022-05-03 12:38:57
python PyGame五子棋小游戏
2023-09-29 13:32:45
![](https://img.aspxhome.com/file/2023/2/95862_0s.png)
python 布尔注入原理及渗透过程示例
2022-11-21 01:04:21
![](https://img.aspxhome.com/file/2023/8/83968_0s.jpg)
完美的js验证网址url(正则表达式)
2008-06-07 09:36:00
ASP函数验证带小数点数字格式
2010-01-02 20:41:00
flask框架自定义过滤器示例【markdown文件读取和展示功能】
2023-03-07 19:22:27
python常见读取语音的3种方法速度对比
2021-03-08 11:12:33