Python异步编程：asyncio模块的探索

大家好，我是涛哥，本文内容来自 涛哥聊Python ，转载请标原创。

更多Python学习内容：http://ipengtao.com

在Python编程中，异步编程是一种重要的技术，可以有效地提升程序的性能和响应速度，特别是在处理I/O密集型任务时。asyncio模块是Python标准库中用于异步编程的核心模块，它提供了事件循环、协程和任务等基本构件。本文将详细介绍asyncio模块的使用方法和高级技巧，帮助全面掌握Python异步编程。

异步编程概述

异步编程是一种并发编程方式，通过事件循环调度多个任务的执行，而不需要等待每个任务完成。这种方式可以有效地利用CPU资源，提高程序的执行效率。

同步与异步的区别

同步编程中，任务按顺序执行，一个任务完成后才能开始下一个任务。而异步编程中，多个任务可以同时启动，在等待I/O操作时，CPU可以执行其他任务。

异步编程的优点

• 高效处理I/O密集型任务
• 提高程序响应速度
• 更好地利用系统资源

asyncio模块基础

事件循环

事件循环是异步编程的核心，用于调度和执行协程。asyncio模块提供了创建和管理事件循环的方法。

示例：创建事件循环

import asyncio

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

# 获取当前事件循环
loop = asyncio.get_event_loop()
# 运行事件循环
loop.run_until_complete(main())
# 关闭事件循环
loop.close()

协程

协程是异步编程中的基本单元，通过async def定义，使用await关键字等待耗时操作。

示例：定义和运行协程

import asyncio

async def say_hello():
    print("Hello")
    await asyncio.sleep(1)
    print("Hello again")

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

任务

任务用于并发执行协程，asyncio模块提供了创建和管理任务的方法。

示例：创建和运行任务

import asyncio

async def task_1():
    print("Task 1 started")
    await asyncio.sleep(1)
    print("Task 1 completed")

async def task_2():
    print("Task 2 started")
    await asyncio.sleep(2)
    print("Task 2 completed")

async def main():
    task1 = asyncio.create_task(task_1())
    task2 = asyncio.create_task(task_2())

    await task1
    await task2

asyncio.run(main())

高级使用

并发运行多个任务

使用asyncio.gather并发运行多个任务。

示例：并发运行任务

import asyncio

async def task_1():
    print("Task 1 started")
    await asyncio.sleep(1)
    print("Task 1 completed")

async def task_2():
    print("Task 2 started")
    await asyncio.sleep(2)
    print("Task 2 completed")

async def main():
    await asyncio.gather(task_1(), task_2())

asyncio.run(main())

超时处理

使用asyncio.wait_for设置超时时间。

示例：任务超时

import asyncio

async def long_task():
    await asyncio.sleep(5)
    return "Task completed"

async def main():
    try:
        result = await asyncio.wait_for(long_task(), timeout=2)
        print(result)
    except asyncio.TimeoutError:
        print("Task timed out")

asyncio.run(main())

处理异常

在异步编程中，可以使用try-except语句处理异常。

示例：处理协程中的异常

import asyncio

async def faulty_task():
    raise ValueError("An error occurred")

async def main():
    try:
        await faulty_task()
    except ValueError as e:
        print(f"Caught an exception: {e}")

asyncio.run(main())

信号量控制并发数

使用asyncio.Semaphore控制并发任务的数量。

示例：信号量

import asyncio

semaphore = asyncio.Semaphore(2)

async def limited_task(name):
    async with semaphore:
        print(f"Task {name} started")
        await asyncio.sleep(1)
        print(f"Task {name} completed")

async def main():
    await asyncio.gather(
        limited_task("A"),
        limited_task("B"),
        limited_task("C"),
        limited_task("D")
    )

asyncio.run(main())

实际应用示例

异步网络请求

使用aiohttp库实现异步网络请求。

安装aiohttp

pip install aiohttp

示例：异步网络请求

import aiohttp
import asyncio

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

async def main():
    url = "https://www.example.com"
    html = await fetch(url)
    print(html)

asyncio.run(main())

异步文件读写

使用aiofiles库实现异步文件读写。

安装aiofiles

pip install aiofiles

示例：异步文件读写

import aiofiles
import asyncio

async def write_file():
    async with aiofiles.open('example.txt', mode='w') as f:
        await f.write("Hello, asyncio!")

async def read_file():
    async with aiofiles.open('example.txt', mode='r') as f:
        content = await f.read()
        print(content)

async def main():
    await write_file()
    await read_file()

asyncio.run(main())

总结

本文详细介绍了Python中的asyncio模块，涵盖了事件循环、协程、任务等基础知识，以及并发任务、超时处理、异常处理和信号量等高级用法。通过具体的示例，展示了如何使用asyncio实现异步网络请求和异步文件读写。掌握这些异步编程技巧，可以帮助大家在Python编程中高效地处理I/O密集型任务，提高程序的性能和响应速度。