asyncio+aiohttp实现视频流压测

# -*- coding: utf-8 -*-
# @Author: liyuexuan
# time: 2023/03/02 18:15

import asyncio

import aiohttp
from aiohttp import ClientSession
import time

import requests

a = time.time()


async def get_sp(url, machine_id, token, out_time):
    """
    获取视频流
    :param url: 请求地址
    :param machine_id: 机台
    :param token: token
    :param out_time: 超时时间
    :return:
    """
    timeout = aiohttp.ClientTimeout(total=out_time)
    async with ClientSession(timeout=timeout) as session:
        url = url + "/api/equipment/stream/" + str(machine_id)
        cookie = {"remember_token": token}
        async with session.get(url, cookies=cookie) as resp:
            # 边访问视频流边处理
            async for chunk in resp.content.iter_chunked(1024):
                print("------------------------")
            # 边播边下载
            # with open('sp.txt', 'wb') as fd:
            #     async for chunk in resp.content.iter_chunked(1024):
            #         fd.write(chunk)


def login(url):
    # 登录
    headers = {"Content-Type": "application/json"}
    api = "/api/user/login"
    param = {"username": "admin", "password": "xxxxxx", "remember": True}
    res = requests.post(url + api, json=param, headers=headers)
    token = res.cookies.get("remember_token")
    return token


async def main():
    url = "http://192.168.xx.xx:8000"
    task_list = []
    token = login(url)
    # 并发次数
    count = 1
    # 机台id
    eid_list = ["1", "2", "3", "4"]
    # 执行时长
    out_time = 60 * 60 * 1
    for eid in eid_list:
        for i in range(count):
            task = asyncio.create_task(get_sp(url, eid, token, out_time))
            task_list.append(task)
    done, pending = await asyncio.wait(task_list, timeout=None)
    # 得到执行结果
    for done_task in done:
        print(f"{time.time()} 得到执行结果 {done_task.result()}")


start_time = time.time()
loop = asyncio.get_event_loop()
loop.run_until_complete(main())
print("总耗时: ", time.time() - start_time)