python的一個接口壓測腳本

import requests
import queue  #Queue模塊中提供了同步的、線程安全的隊列類，包括        
                     #FIFO（先入先出)隊列Queue，LIFO（后入先出）隊列 
                     #LifoQueue，和優先級隊列PriorityQueue。這些隊列都 
                     #實現了鎖原語，可在多線程通信中直接使用。
import threading
import time

status_code_list = []
exec_time = 0
class MyThreadPool:
    def __init__(self, maxsize): #定義隊列時有一個默認的參數 
        #maxsize, 如果不指定隊列的長度，即manxsize=0,那么隊列的長 
      #度為無限長，如果定義了大於0的值，那么隊列的長度就是maxsize。
        self.maxsize = maxsize
        self._pool = queue.Queue(maxsize) 
                  #maxsize設置隊列的大小為pool的大小
        for _ in range(maxsize):    #為什么用一個下划線，因為實際上這 
                    #里沒用到這個變量，所以用一個符號就可以了。
            self._pool.put(threading.Thread)    #往pool里放線程數

    def get_thread(self):
        return self._pool.get()

    def add_thread(self):
        self._pool.put(threading.Thread)

def request_time(func):
    def inner(*args, **kwargs):
        global exec_time
        start_time = time.time()
        func(*args, **kwargs)
        end_time = time.time()
        exec_time = end_time-start_time

    return inner


def get_url(url):
    global x,status_code_list
    headers = {'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/65.0.3325.146 Safari/537.36',
               }
    response = requests.get(url,headers=headers)
    code = response.status_code
    status_code_list.append(code)
    print(code)
    return code


def get_count(_url='http://news.baidu.com/sports',_count=100):    # :param count: 每個線程請求的數量
    global status_code_list,url,count
    for i in range(count):
        get_url(url)

def request_status():
    count_num = len(status_code_list)
    set_code_list = set(status_code_list)
    status_dict = {}
    for i in set_code_list:
        status_dict[i] = str(status_code_list).count(str(i))
    echo_str(count_num, set_code_list, status_dict)

def echo_str(count_num,set_code_list,status_dict):
    print('=======================================')
    print('請求總次數:%s'%count_num)
    print('請求時長:%s秒'%int(exec_time))
    second_request = count_num/int(exec_time)
    print('每秒請求約:%s次'%int(second_request))
    print('狀態碼 | 次數')

    for k,v in status_dict.items():
        print(str(k)+'    | '+str(v))
    print('=======================================')


@request_time
def run(url,thread_num=10,thread_pool=10):
    '''
    :param thread_num: 總共執行的線程數(總的請求數=總共執行的線程數*每個線程循環請求的數量)
    :param thread_pool: 線程池數量
    :param url: 請求的域名地址
    '''
    global x,status_code_list
    pool = MyThreadPool(thread_pool)
    for i in range(thread_num):
        t = pool.get_thread()
        obj = t(target=get_count)
        obj.start()
        obj.join()


if __name__ == '__main__':
    count = 10  #單個線程的請求數
    url = 'http://baidu.com'
    run(url,100,100)
    request_status()