lightgbm_score/pressure_test.py

"""
@author: luojunhui
"""
import time
import requests
import argparse
from concurrent.futures import ThreadPoolExecutor


def request_data(_url):
    index = _url.split("#")[0]
    url = _url.split("#")[1]
    body = {
        "version": "v2",
        "features": {
            "channel": "小年糕",
            "out_user_id": "66506470",
            "mode": "author",
            "out_play_cnt": 698,
            "out_like_cnt": 1,
            "out_share_cnt": 41,
            "title": "收藏好这九种中成药，让你少往医院跑。?",
            "lop": 701 / 24698,  # like_cnt + 700 / play_cnt + 24000
            "duration": 180
        }
    }
    t = time.time()
    res = requests.post(url, json=body)
    e = time.time()
    print(index)
    print(e - t)
    print(res.json())


if __name__ == "__main__":
    parser = argparse.ArgumentParser()  # 新建参数解释器对象
    parser.add_argument("--thread")
    args = parser.parse_args()
    thread = int(args.thread)
    dt = ["{}#http://127.0.0.1:8000/lightgbm_score".format(i) for i in range(1, 1001)]
    total_s = time.time()
    with ThreadPoolExecutor(max_workers=thread) as pool:
        pool.map(request_data, dt)
    total_e = time.time()
    print(total_e - total_s)