Python asyncio 高级应用指南_python的asyncio用处-优快云博客

Python asyncio 高级应用指南

1. 基本概念

asyncio 是 Python 3.4+ 引入的异步 I/O 框架，它允许我们编写并发代码，而不需要使用多线程或多进程。

import asyncio

async def main():
    print("Hello")
    await asyncio.sleep(1)
    print("World")

asyncio.run(main())

2. 协程和任务

2.1 协程

协程是一种特殊的函数，它可以在执行过程中暂停，并在稍后恢复执行。

async def greet(name):
    print(f"Hello, {name}!")
    await asyncio.sleep(1)
    print(f"Goodbye, {name}!")

async def main():
    await greet("Alice")
    await greet("Bob")

asyncio.run(main())

2.2 任务

任务是协程的一个包装器，它允许我们并发执行多个协程。

async def greet(name):
    print(f"Hello, {name}!")
    await asyncio.sleep(1)
    print(f"Goodbye, {name}!")

async def main():
    task1 = asyncio.create_task(greet("Alice"))
    task2 = asyncio.create_task(greet("Bob"))
    
    await task1
    await task2

asyncio.run(main())

3. 高级特性

3.1 并发执行

使用 asyncio.gather 可以并发执行多个协程。

async def factorial(n):
    if n == 0:
        return 1
    await asyncio.sleep(0.1)
    return n * await factorial(n-1)

async def main():
    results = await asyncio.gather(
        factorial(5),
        factorial(10),
        factorial(15)
    )
    print(results)

asyncio.run(main())

3.2 超时处理

使用 asyncio.wait_for 可以为协程设置超时。

async def slow_operation():
    await asyncio.sleep(2)
    return "Done"

async def main():
    try:
        result = await asyncio.wait_for(slow_operation(), timeout=1)
        print(result)
    except asyncio.TimeoutError:
        print("Operation timed out")

asyncio.run(main())

3.3 事件循环

事件循环是 asyncio 的核心，它负责调度和执行协程。

async def main():
    print("Hello")
    await asyncio.sleep(1)
    print("World")

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

# 运行协程
loop.run_until_complete(main())

# 关闭事件循环
loop.close()

4. 实际应用场景

4.1 网络请求

使用 aiohttp 库可以进行异步 HTTP 请求。

import aiohttp
import asyncio

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

async def main():
    urls = [
        "https://api.github.com/users/octocat",
        "https://api.github.com/users/github",
        "https://api.github.com/users/microsoft"
    ]
    
    async with aiohttp.ClientSession() as session:
        tasks = [fetch_url(session, url) for url in urls]
        results = await asyncio.gather(*tasks)
        
        for url, result in zip(urls, results):
            print(f"{url}: {len(result)} bytes")

asyncio.run(main())

4.2 文件 I/O

使用 aiofiles 库可以进行异步文件 I/O 操作。

import aiofiles
import asyncio

async def read_file(filename):
    async with aiofiles.open(filename, 'r') as f:
        content = await f.read()
    return content

async def write_file(filename, content):
    async with aiofiles.open(filename, 'w') as f:
        await f.write(content)

async def main():
    # 读取文件
    content = await read_file("input.txt")
    print(f"Read {len(content)} bytes")
    
    # 写入文件
    await write_file("output.txt", content)
    print("File written successfully")

asyncio.run(main())

4.3 数据库操作

使用 asyncpg 库可以进行异步 PostgreSQL 数据库操作。

import asyncpg
import asyncio

async def main():
    # 连接数据库
    conn = await asyncpg.connect(
        user='postgres',
        password='password',
        database='test',
        host='localhost'
    )
    
    # 执行查询
    rows = await conn.fetch('SELECT * FROM users')
    for row in rows:
        print(row)
    
    # 关闭连接
    await conn.close()

asyncio.run(main())