架构师训练营 -- 第七周作业

import sys
import time
import requests
import threading

# 记录当前已经完成多少任务
g_sum = 0
g_result = []
R = threading.Lock()


def send_request(url):
    # 发送请求
    start_time = time.perf_counter()
    res = requests.get(url)
    end_time = time.perf_counter()
    elapsed = (end_time - start_time)
    return elapsed


def test_thread_func(url):
    while True:
        # 判断是否还有任务需要处理
        task_id = 0
        R.acquire()
        global g_sum
        if g_sum <= 0:
            R.release()
            break
        task_id = g_sum
        g_sum -= 1
        R.release()

        elapsed = send_request(url)
        R.acquire()
        global g_result
        g_result.append(elapsed)
        print(task_id, elapsed)
        R.release()


def report(result):
    avg = sum(result) / len(result)
    print("avg:", avg)
    result.sort()
    print("95:", result[95])


def main(argv):
    if len(argv) != 3:
        print(u"请求格式: URL 请求总次数 并发数")
        return False
    url = argv[0]
    total = argv[1]
    global g_sum
    g_sum = int(total)
    concurrent = argv[2]

    # 依据并发数，初始化线程
    thread_list = []
    for i in range(int(concurrent)):
        test_thread = threading.Thread(target=test_thread_func, args=(url,))
        thread_list.append(test_thread)

    for t in thread_list:
        t.setDaemon(True)
        t.start()
    for t in thread_list:
        t.join()

    global g_result
    report(g_result)


if __name__ == "__main__":
    main(sys.argv[1:])