#!/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()