manyi/script/test_nginx.py

import requests
import os


from concurrent.futures import ThreadPoolExecutor, as_completed

from base import init_env

init_env(True)


TOTAL = 10000
WORKER = 100
LOGICAL_CODE = "123456"
domain = os.environ.get("MY_DOMAIN")

URL = "http://{}/userLogin?l={}".format(domain, LOGICAL_CODE)
HEADER = {
    "user-agent": ""
}


# print URL
# raise Exception("123")


def request_url():
    try:
        res = requests.get(url=URL, headers=HEADER, timeout = 15)
    except Exception as e:
        return "error"
    else:
        print res.status_code

    return res.status_code


with ThreadPoolExecutor(max_workers=WORKER) as excutor:
    tasks = list()

    for i in range(TOTAL):
        tasks.append(excutor.submit(request_url))


for task in as_completed(tasks):
    print task.result()