欢迎来到徐庆高(Tea)的个人博客网站
磨难很爱我,一度将我连根拔起。从惊慌失措到心力交瘁,我孤身一人,但并不孤独无依。依赖那些依赖我的人,信任那些信任我的人,帮助那些给予我帮助的人。如果我愿意,可以分裂成无数面镜子,让他们看见我,就像看见自己。察言观色和模仿学习是我的领域。像每个深受创伤的人那样,最终,我学会了随遇而安。
当前位置: 日志文章 > 详细内容

Python异步编程之await与asyncio基本用法详解

2025年08月18日 Python
在 python 中,await 和 asyncio 是异步编程的核心工具,用于高效处理 i/o 密集型任务(如网络请求、文件读写、数据库操作等)。它们的核心思想是通过 协程(coroutine) 和

在 python 中,awaitasyncio 是异步编程的核心工具,用于高效处理 i/o 密集型任务(如网络请求、文件读写、数据库操作等)。它们的核心思想是通过 协程(coroutine)事件循环(event loop) 实现非阻塞并发,避免线程切换的开销。

一、核心概念

协程(coroutine)
async def 定义的函数,返回一个协程对象,可以通过 await 挂起执行,让出控制权。

async def my_coroutine():
    await asyncio.sleep(1)
    print("done")

事件循环(event loop)
异步程序的核心调度器,负责执行协程并在 i/o 操作时切换任务。

可等待对象(awaitable)
包括协程、asyncio.taskasyncio.future。只有可等待对象才能被 await

二、使用场景

  1. 高并发网络请求
    如爬虫、api 调用等需要同时处理大量连接的场景。
  2. web 服务器
    如 fastapi、sanic 等异步框架处理 http 请求。
  3. 数据库操作
    异步驱动(如 asyncpgaiomysql)避免阻塞主线程。
  4. 实时通信
    websocket、聊天服务器等需要长连接的场景。

三、基本用法

1. 定义协程

async def fetch_data(url):
    # 模拟网络请求
    await asyncio.sleep(1)
    return f"data from {url}"

2. 运行协程

async def main():
    result = await fetch_data("https://example.com")
    print(result)
# python 3.7+ 推荐方式
asyncio.run(main())

3. 并发执行多个任务

使用 asyncio.gather()asyncio.create_task()

async def main():
    # 同时执行多个协程
    task1 = asyncio.create_task(fetch_data("url1"))
    task2 = asyncio.create_task(fetch_data("url2"))
    # 等待所有任务完成
    results = await asyncio.gather(task1, task2)
    print(results)

四、关键 api 详解

  • asyncio.run(coro)
    • 启动事件循环并运行协程(python 3.7+)。
  • asyncio.create_task(coro)
    • 将协程包装为 task,加入事件循环并发执行。
  • asyncio.gather(*coros)
    • 并发执行多个协程,返回结果列表。
  • asyncio.sleep(delay)
    • 非阻塞等待(模拟 i/o 操作)。

五、高级用法

1. 控制并发量

使用信号量(semaphore)限制同时运行的任务数:

async def limited_fetch(url, semaphore):
    async with semaphore:
        return await fetch_data(url)
async def main():
    semaphore = asyncio.semaphore(5)  # 最大并发5
    tasks = [limited_fetch(url, semaphore) for url in urls]
    await asyncio.gather(*tasks)

2. 超时控制

async def fetch_with_timeout():
    try:
        async with asyncio.timeout(3):  # python 3.11+
            await fetch_data("slow_url")
    except timeouterror:
        print("timeout!")

3. 回调与 future

async def main():
    loop = asyncio.get_running_loop()
    future = loop.create_future()
    def callback():
        future.set_result("done")
    loop.call_soon(callback)
    result = await future
    print(result)

六、常见错误

忘记 await
协程不会被自动执行:

async def main():
    fetch_data("url")  # 错误!没有 await

阻塞主线程
在协程中调用同步代码(如 time.sleep())会阻塞事件循环:

async def bad_example():
    time.sleep(1)  # 错误!应使用 await asyncio.sleep(1)

滥用并发
异步不适合 cpu 密集型任务,此时应使用多进程。

七、完整示例

import asyncio
async def download(url, delay):
    print(f"start downloading {url}")
    await asyncio.sleep(delay)
    print(f"finished {url}")
    return url
async def main():
    urls = [
        ("https://site1.com", 1),
        ("https://site2.com", 2),
        ("https://site3.com", 3),
    ]
    tasks = [asyncio.create_task(download(url, delay)) for url, delay in urls]
    results = await asyncio.gather(*tasks)
    print("all done:", results)
if __name__ == "__main__":
    asyncio.run(main())

输出:

start downloading https://site1.com
start downloading https://site2.com
start downloading https://site3.com
finished https://site1.com
finished https://site2.com
finished https://site3.com
all done: ['https://site1.com', 'https://site2.com', 'https://site3.com']

八、总结

  • 适用场景:i/o 密集型任务,如网络、文件、数据库操作。
  • 关键点
    • 使用 async def 定义协程,用 await 挂起阻塞操作。
    • 通过 asyncio.create_task()asyncio.gather() 实现并发。
    • 避免在协程中调用阻塞同步代码。

通过合理使用 asyncio,可以在单线程内高效处理成千上万的并发连接。

到此这篇关于python异步编程之await与asyncio基本用法详解的文章就介绍到这了,更多相关python await与asyncio内容请搜索代码网以前的文章或继续浏览下面的相关文章希望大家以后多多支持代码网!