当前位置： 技术文章>> Python 中的 asyncio 如何使用？

文章标题：Python 中的 asyncio 如何使用？
  • 文章分类: 后端
  • 5672 阅读

在Python中，asyncio 是一个用于编写单线程并发代码的库，它使用协程（coroutines）来实现非阻塞的IO操作。asyncio 使得编写异步代码变得更加直观和易于管理，尤其适用于IO密集型任务，如网络请求、文件操作等。下面，我们将深入探讨如何在Python中使用 asyncio，并通过实例展示其强大功能。

1. 理解异步编程与协程

在深入 asyncio 之前，理解异步编程和协程的概念至关重要。异步编程允许程序在等待某个操作（如网络请求）完成时，继续执行其他任务，从而提高程序的执行效率和响应性。协程是异步编程中的一个核心概念，它允许函数在执行过程中暂停和恢复，而不需要像线程那样占用额外的系统资源。

2. asyncio 的基础

asyncio 库提供了创建和管理协程、事件循环（event loop）以及任务（task）的API。事件循环是 asyncio 的核心，它负责调度和执行协程。任务则是协程的封装，可以被事件循环调度执行。

2.1 创建协程

在Python中，使用 async def 关键字定义协程函数。这样的函数在调用时不会立即执行，而是返回一个协程对象。要执行协程，你需要将其传递给事件循环或使用 await 关键字（在另一个协程内部）。

import asyncio

async def hello_world():
    print("Hello, world!")
    await asyncio.sleep(1)  # 模拟异步IO操作
    print("Hello again!")

# 协程对象
coro = hello_world()

# 获取当前事件循环
loop = asyncio.get_event_loop()

# 将协程添加到事件循环中执行
loop.run_until_complete(coro)

# 或者使用 asyncio.run()（Python 3.7+）
# asyncio.run(hello_world())

2.2 使用 await

await 关键字用于等待协程完成。它只能在 async def 定义的函数内部使用。await 可以调用另一个协程，并暂停当前协程的执行，直到等待的协程完成。

async def fetch_data():
    # 假设这是一个异步的HTTP请求
    await asyncio.sleep(2)  # 模拟网络延迟
    return "Data fetched"

async def process_data():
    data = await fetch_data()
    print(f"Processing {data}")

# 运行 process_data 协程
asyncio.run(process_data())

3. 并发执行多个协程

asyncio 允许你并发执行多个协程，而不需要为每个协程创建单独的线程。这通过 asyncio.gather() 或 asyncio.wait() 函数实现，它们可以等待多个协程完成。

3.1 使用 asyncio.gather()

asyncio.gather() 函数接受多个协程作为参数，并返回一个协程，该协程在所有传入的协程完成后完成。

async def task(name, delay):
    print(f"{name} started")
    await asyncio.sleep(delay)
    print(f"{name} finished")

async def main():
    await asyncio.gather(
        task("Task 1", 2),
        task("Task 2", 1),
        task("Task 3", 3),
    )

asyncio.run(main())

3.2 使用 asyncio.wait()

asyncio.wait() 函数提供了更灵活的等待方式，允许你指定等待哪些协程完成，以及如何处理未完成的协程。

async def main():
    done, pending = await asyncio.wait([
        task("Task 1", 2),
        task("Task 2", 1),
        task("Task 3", 3),
    ], return_when=asyncio.ALL_COMPLETED)

    for d in done:
        print(f"Completed: {d.result()}")

    # 处理 pending 协程（如果有的话）
    # 注意：在这个例子中，由于我们使用了 ALL_COMPLETED，所以不会有 pending 协程

asyncio.run(main())

4. 异步上下文管理器

Python 的 async with 语句允许你编写异步的上下文管理器，这在处理需要异步初始化和清理的资源时非常有用。

class AsyncContextManager:
    async def __aenter__(self):
        print("Entering context")
        return self

    async def __aexit__(self, exc_type, exc_val, exc_tb):
        print("Exiting context")

async def demo():
    async with AsyncContextManager():
        print("Inside context")
        await asyncio.sleep(1)

asyncio.run(demo())

5. 异步IO与库支持

许多现代Python库都提供了对 asyncio 的支持，允许你以异步方式执行IO操作，如网络请求、数据库操作等。例如，aiohttp 是一个用于异步HTTP客户端和服务器编程的库，aiopg 提供了异步的PostgreSQL支持。

5.1 使用 aiohttp 发送异步HTTP请求

import aiohttp

async def fetch(session, url):
    async with session.get(url) as response:
        return await response.text()

async def main():
    async with aiohttp.ClientSession() as session:
        html = await fetch(session, 'http://python.org')
        print(html[:100] + '...')

asyncio.run(main())

6. 实战应用：构建异步Web服务器

使用 aiohttp，你可以轻松地构建异步Web服务器，处理并发请求而无需为每个请求创建新的线程。

from aiohttp import web

async def handle(request):
    name = request.match_info.get('name', "Anonymous")
    text = f"Hello, {name}"
    return web.Response(text=text)

app = web.Application()
app.add_routes([web.get('/', handle),
                web.get('/{name}', handle)])

if __name__ == '__main__':
    web.run_app(app)

7. 深入探索与最佳实践

  • 避免阻塞调用：在协程中避免使用阻塞调用，如同步的IO操作或长时间运行的CPU密集型任务。
  • 合理使用并发：虽然 asyncio 允许你并发执行多个协程，但过多的并发可能会导致性能下降。根据系统资源和任务性质合理设置并发数。
  • 错误处理：使用 try...except 语句来捕获和处理协程中可能发生的异常。
  • 调试与日志：利用Python的日志模块记录协程的执行情况，有助于调试和性能分析。

结语

asyncio 是Python中一个强大的异步编程库，它使得编写高效、可扩展的异步代码变得简单。通过理解协程、事件循环和异步IO的概念，你可以利用 asyncio 来构建高性能的Web服务器、网络客户端和其他IO密集型应用。在码小课网站上，我们将继续深入探讨 asyncio 的高级特性和最佳实践，帮助你更好地掌握这一强大的工具。

推荐文章