asyncio執行多個request請求

我們先寫一個flask小程序
demo.py

from flask import Flask
import time

app = Flask(__name__)


@app.route('/')
def index():
    time.sleep(3)
    return 'Hello!'

@app.route('/tests')
def tests():
    time.sleep(3)
    return 'Hello tests!'

if __name__ == '__main__':
    app.run(threaded=True)  # 這表明 Flask 啓動了多線程模式,不然默認是隻有一個線程的。

然後我們寫個請求的文件

import asyncio
import aiohttp
import time
from concurrent.futures import ProcessPoolExecutor
start = time.time()

async def gets(url):
    session = aiohttp.ClientSession()
    response = await session.get(url)
    result = await response.text()
    await session.close()
    return result

async def requestindexs():
    url = 'http://127.0.0.1:5000'
    print('go go go ',url)
    result = await gets(url)
    print('Get response from',url,'Result:',result)

async def requesttests():
    url = 'http://127.0.0.1:5000/tests'
    print('go go go ',url)
    result = await gets(url)
    print('Get response from',url,'Result:',result)

def main():
    tasks = []   # 我們要把所有的任務都註冊到tasks中,建議最多500
    # for _ in range(200):
    #     index = asyncio.ensure_future(requestindexs())
    #     tests = asyncio.ensure_future(requesttests())
    #     tasks.append(index)
    #     tasks.append(tests)

    index = [tasks.append(asyncio.ensure_future(requestindexs())) for _ in range(200)]
    tests = [tasks.append(asyncio.ensure_future(requesttests())) for _ in range(200)]
    loop = asyncio.get_event_loop()
    loop.run_until_complete(asyncio.wait(tasks))
    loop.close()


if __name__ == '__main__':
    main()
    end = time.time()
    print('Cost time:', end - start)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章