python学习笔记（threading接口性能压力测试）

又是新的一周

延续上周的进度 关于多进程的学习

今天实践下

初步设计的接口性能压力测试代码如下：

 #!/usr/bin/env python
 # -*- coding: utf_8 -*-

 import threading
 import requests
 import time
 import re
 from time import sleep

 # -------接口性能测试配置-------
 method = "post"
 # 接口类型
 url = "http://localhost:8081/swcw/back/sysLogin.action"
 # 接口地址
 data = {"username": "admin", "password": ""}
 # 接口参数
 thread_num = 20
 # 线程数
 one_work_num = 2
 # 每个线程循环次数
 loop_sleep = 1
 # 每次请求时间间隔
 response_time = []
 # 平均响应时间列表
 error = []
 # 错误信息列表

 class CreateThread:
     def __init__(self):
         pass

     @classmethod
     def thread_api(cls):
         global results
         try:
             if method == "post":
                 results = requests.post(url, data)
             if method == "get":
                 results = requests.get(url, data)
             return results
         except requests.ConnectionError:
             return results
     # 接口函数

     @classmethod
     def thread_response(cls):
         responsetime = float(CreateThread.thread_api().elapsed.microseconds) / 1000
         return responsetime
     # 获取响应时间 单位ms

     @classmethod
     def thread_response_avg(cls):
         avg = 0.000
         l = len(response_time)
         for num in response_time:
             avg += 1.000 * num / l
         return avg
     # 获取平均相应时间 单位ms

     @classmethod
     def thread_time(cls):
         return time.asctime(time.localtime(time.time()))
     # 获取当前时间格式

     @classmethod
     def thread_error(cls):
         try:
             pa = u"个人信息"
             pattern = re.compile(pa)
             match = pattern.search(CreateThread.thread_api().text)
             if CreateThread.thread_api().status_code == 200:
                 pass
                 if match.group() == pa:
                     pass
             else:
                 error.append(CreateThread.thread_api().status_code)
         except AttributeError:
             error.append("登录失败")
     # 获取错误的返回状态码

     @classmethod
     def thread_work(cls):
         threadname = threading.currentThread().getName()
         print "[", threadname, "] Sub Thread Begin"
         for i in range(one_work_num):
             CreateThread.thread_api()
             print "接口请求时间： ", CreateThread.thread_time()
             response_time.append(CreateThread.thread_response())
             CreateThread.thread_error()
             sleep(loop_sleep)
         print "[", threadname, "] Sub Thread End"
     # 工作线程循环

     @classmethod
     def thread_main(cls):
         start = time.time()
         threads = []
         for i in range(thread_num):
             t = threading.Thread(target=CreateThread.thread_work())
             t.setDaemon(True)
             threads.append(t)
         for t in threads:
             t.start()
         # 启动所有线程
         for t in threads:
             t.join()
         # 主线程中等待所有子线程退出
         end = time.time()

         print "========================================================================"
         print "接口性能测试开始时间：", time.asctime(time.localtime(start))
         print "接口性能测试结束时间：", time.asctime(time.localtime(end))
         print "接口地址：", url
         print "接口类型：", method
         print "线程数：", thread_num
         print "每个线程循环次数：", one_work_num
         print "每次请求时间间隔：", loop_sleep
         print "总请求数：", thread_num * one_work_num
         print "错误请求数：", len(error)
         print "总耗时（秒）：", end - start
         print "每次请求耗时（秒）：", (end - start) / (thread_num * one_work_num)
         print "每秒承载请求数（TPS)：", (thread_num * one_work_num) / (end - start)
         print "平均响应时间（毫秒）：", CreateThread.thread_response_avg()

 if __name__ == '__main__':
     CreateThread.thread_main()