实现Python压力测试工具｜Python 主题月

Python压力测试工具

写这个东西的初衷是因为，马上就要压力测试，想要写一个工具并可以根据结果生成图表,在该项目基础上可以定制化开发，喜欢或对你有帮助给个start谢谢啦

完整代码github地址

用到的技术和依赖包

1. 多线程
2. 锁
3. requests

4. pyecharts

   配置了两种发起请求方式(这里如果数量过多还可以再优化多线程循环)

5. 快速请求，将请求池中的请求全部发送

6. 慢速请求，在指定的时间内全部发送完毕 ```python def fastrun(begin, end): log.logger.warning("开始index{} 结束index{}".format(begin, end)) runstarttime = time.time() log.logger.info('开始发起快速异步请求....') for request in syncPool[int(begin):int(end)]: request.start() log.logger.warning("请求消耗时间 {:.2f} 豪秒".format((time.time() - runstart_time) * 1000))

   def slowrun(begin, end, slowTime): log.logger.warning("开始index{} 结束index{}".format(begin, end)) runstarttime = time.time() log.logger.info('开始发起慢速异步请求....') if slowTime == 0: fastrun(begin, end) else: waitTime = round(((slowTime * 1000) / len(syncPool)) / 1000, 3) log.logger.info("延迟时间为{}毫秒".format(waitTime * 1000)) for request in syncPool[int(begin):int(end)]: request.start() time.sleep(waitTime) log.logger.warning("请求消耗时间 {} 豪秒".format((time.time() - runstarttime) * 1000)) 数据统计 python def out(length): log.logger.warning(f"总请求次数 {int(length)} 次") log.logger.warning(f"成功请求 {len(sync.success)} 次") log.logger.warning(f"失败请求 {len(sync.fail)} 次") log.logger.warning(f"限流请求 {sync.limit} 次") log.logger.warning("成功率 {:.2f} %".format((len(sync.success) / int(length)) * 100)) log.logger.warning("失败率 {:.2f} %".format((len(sync.fail) / int(length)) * 100)) log.logger.warning("限流百分比 {:.2f} %".format(sync.limit / int(length) * 100)) log.logger.warning("最长请求时间 {:.2f} 毫秒".format(max(sync.responsetime))) log.logger.warning("最短请求时间 {:.2f} 毫秒".format(min(sync.responsetime))) log.logger.warning("平均请求时间 {:.2f} 毫秒".format(sum(sync.responsetime) / len(sync.responsetime))) # 打印成功请求的request中信息 for success in sync.success: log.logger.debug(success.content) # 打印失败请求的request中信息 for fail in sync.fail: if fail is not None: log.logger.debug(fail.content) 多线程请求对象 python class SyncRequestTask(threading.Thread):

   def init(self, threadId, url, method, params, header, timeout=10): threading.Thread.init(self) self.setName(f"sync-{threadId}") self.url = url self.method = method self.params = params self.timeout = timeout self.header = header

   # 发送请求 def request(self): req = None try: if self.method == 'GET': req = self.doGet() self.add(req) else: req = self.doPost() self.add(req) except Exception as e: print(e) fail.append(req)

   def doGet(self): startTime = time.time()

   s = sessions.session()
      req = s.get(self.url, headers=self.header, timeout=self.timeout)
      data_build(self.getName(), time.time() - startTime)
      req.close()
      return req

   def doPost(self): # requestbody = json.dumps(self.params) s = sessions.session() startTime = time.time() req = s.post(self.url, data=self.params, headers=self.header, timeout=self.timeout) databuild(self.getName(), time.time() - startTime) req.close() return req

   @staticmethod def add(request): global lock global limit lock.acquire() if request.statuscode == 200: success.append(request) elif request.statuscode == 429: limit += 1 else: fail.append(request) lock.release()

   def run(self): # 开始发送请求 self.request() ```