【代码片段】利用python多进程进行接口并发测试

最新推荐文章于 2024-08-11 10:00:00 发布

lovenliu

最新推荐文章于 2024-08-11 10:00:00 发布

阅读量3k

点赞数 1

分类专栏： python 文章标签： python 进程

本文链接：https://blog.csdn.net/lypeng_/article/details/98059263

版权

本文介绍如何使用Python的多进程进行接口并发测试。测试目的是检查服务在并发请求下是否存在阻塞，通过创建多个进程模拟并发请求，并观察总执行时间。示例代码展示了创建10个进程来执行10个并发请求的实现方式。

摘要由CSDN通过智能技术生成

【代码片段】利用python多进程进行接口并发测试

- - - - 前言
      - 代码

前言

最近公司服务类的项目有个测试并发的需求，测试的目的并非是服务器能抗住多大的并发，而是测试服务在并发请求下是否是阻塞的。比如单个并发的执行时间大搞是两秒的话，那么10个并发的总执行时间是所有请求执行时间的总和还是最大请求时间。

代码

import time
import requests
from multiprocessing import Process
from multiprocessing import Pool

data = {
    "times": 10, # 并发量
    "method": "POST",
    "url": "http://xxx.com/xxx",
    "header": {
        "Content-Type": "application/json",
        "user-agent": "python-mock/0.0.1"
    },
    "body": {
        # 参数
    }
}

def run_task(idx):
    response = requests.post(data["url"], json=data["body"], headers=data["header"])
    if response.status_code == 200:
        result = response.content.decode('utf-8')
    else:
        result = "访问失败"
    print("第 %s 次执行：%s \n" % (idx, result)