天天看點

Python程式設計:asyncio協程程式設計代碼示例

同步IO:

一旦遇到IO操作,如讀寫檔案、發送網絡資料時,就需要等待IO操作完成,才能繼續進行下一步操作

異步IO:

當代碼需要執行一個耗時的IO操作時,它隻發出IO指令,并不等待IO結果,然後就去執行其他代碼了。一段時間後,當IO傳回結果時,再通知CPU進行處理

消息循環:

主線程不斷地重複“讀取消息-處理消息”

當遇到IO操作時,代碼隻負責發出IO請求,不等待IO結果,然後直接結束本輪消息處理,進入下一輪消息處理過程。當IO操作完成後,将收到一條“IO完成”的消息,處理該消息時就可以直接擷取IO操作結果。

代碼示例

1、簡單示例

# -*- coding: utf-8 -*-

import asyncio


async def main():
    await asyncio.sleep(3)


asyncio.get_event_loop().run_until_complete(main())
      

2、多任務切換示例

建立3 個任務,每個任務耗時2s ,如果是同步IO則需要

6s

執行時間。

使用協程多任務切換最終耗時

2s

多一點點

# -*- coding: utf-8 -*-

import time
import asyncio


async def job(job_id):   # async 形式的功能
    print("start job: %s" % job_id)
    await asyncio.sleep(2)  # 等待 2 秒, 期間切換其他任務
    print("end job: %s" % job_id)


async def main(_loop):
    # 建立 3 個任務, 但是不執行
    tasks = [
        _loop.create_task(job(job_id)) for job_id in range(0, 3)
    ]

    await asyncio.wait(tasks)  # 執行并等待所有任務完成


if __name__ == '__main__':
    start = time.time()

    loop = asyncio.get_event_loop()      # 建立 loop
    loop.run_until_complete(main(loop))  # 執行 loop
    loop.close()                         # 關閉 loop

    print("time: %s" % (time.time() - start))

"""
start job: 0
start job: 1
start job: 2
end job: 0
end job: 1
end job: 2
time: 2.0042531490325928
"""
      

參考

  1. 加速爬蟲: 異步加載 Asyncio
  2. 廖雪峰-異步IO