多線程配合協程

協程配合線程

asyncio.run_coroutine_threadsafe

該方法的語法如下:

asyncio.run_coroutine_threadsafe(coro, loop)

其實在協程中也可以使用多線程,有時候我們需要在主線程中啟動一個子線程去做別的任務，這個時候我們就要用到下面的方法了，先上一個流暢的Python中的代碼。

import time
import asyncio
from  threading import Thread

now = lambda: time.time()


def start_loop(loop):
    asyncio.set_event_loop(loop)
    loop.run_forever()


async def do_some_work(x):
    print(f'Waiting {x}')
    await asyncio.sleep(x)
    print(f'Done after {x}s')


def more_work(x):
    print(f'More work {x}')
    time.sleep(x)
    print('Finished more work {x}')


start = now()
# 主線程中創建一個 new_loop
new_loop = asyncio.get_event_loop()
# 創建子線程 在其中開啟無限事件循環
t = Thread(target=start_loop, args=(new_loop,))
t.start()
print(f'TIME: {time.time() - start}')

# 在主線程中新注冊協程對象
# 這樣即可在子線程中進行事件循環的並發操作 同時主線程又不會被 block 
# 一共執行的時間大概在 6 s 左右 
asyncio.run_coroutine_threadsafe(do_some_work(6), new_loop)
asyncio.run_coroutine_threadsafe(do_some_work(4), new_loop)

上述的例子，主線程中創建一個new_loop，然后在另外的子線程中開啟一個無限事件循環。主線程通過run_coroutine_threadsafe新注冊協程對象。這樣就能在子線程中進行事件循環的並發操作，同時主線程又不會被block。一共執行的時間大概在6s左右