Python异步编程技术详解:async、await、yield和anext

engchina

于 2024-07-03 10:56:24 发布

阅读量998

点赞数 9

分类专栏： LINUX 文章标签： python 开发语言异步 async yield anext await

本文链接：https://blog.csdn.net/engchina/article/details/140146566

版权

LINUX 专栏收录该内容

585 篇文章 67 订阅

订阅专栏

Python异步编程技术详解:async、await、yield和anext

异步编程是Python中一种强大的并发编程模式,可以显著提高I/O密集型应用的性能。本文将详细介绍Python中的几种重要的异步编程技术,包括async/await、yield和anext等,并通过一个实际的代码示例来展示它们的使用。

1. async和await

async和await是Python 3.5引入的语法,用于定义和使用协程(coroutine)。

async def用于定义一个协程函数
await用于等待一个协程完成

示例:

import asyncio

async def fetch_data():
    print("开始获取数据...")
    await asyncio.sleep(2)  # 模拟I/O操作
    print("数据获取完成!")
    return {"data": "some_value"}

async def main():
    result = await fetch_data()
    print(f"获取到的数据: {result}")

asyncio.run(main())

2. yield

yield关键字用于定义生成器函数。在异步编程中,它常用于创建异步生成器。

示例:

async def async_generator():
    for i in range(3):
        await asyncio.sleep(1)
        yield i

async def main():
    async for number in async_generator():
        print(f"生成的数字: {number}")

asyncio.run(main())

3. anext

anext()是Python 3.10引入的函数,用于获取异步迭代器的下一个值。它返回一个协程对象,可以使用await等待结果。

示例:

import asyncio

async def async_counter(stop):
    count = 0
    while count < stop:
        yield count
        count += 1
        await asyncio.sleep(0.1)

async def main():
    counter = async_counter(3)
    try:
        while True:
            value = await anext(counter)
            print(f"计数: {value}")
    except StopAsyncIteration:
        print("计数结束")

asyncio.run(main())

4. StopAsyncIteration

StopAsyncIteration是一个异常,用于标识异步迭代的结束。当异步迭代器没有更多的值可以产生时,会抛出这个异常。

5. 综合示例:异步聊天机器人

下面是一个结合了上述所有概念的异步聊天机器人示例:

import asyncio
import random

class ChatBot:
    def __init__(self, name):
        self.name = name

    async def generate_response(self, message):
        await asyncio.sleep(random.uniform(0.5, 2.0))  # 模拟思考时间
        responses = [
            f"你说'{message}'是什么意思?",
            f"嗯,关于'{message}'我需要想想...",
            f"'{message}'很有趣!我们来聊点别的吧。",
            "我明白了,请继续。",
            "这个话题真深奥,能具体解释一下吗?"
        ]
        return f"{self.name}: {random.choice(responses)}"

async def chat_stream(bot, messages):
    for message in messages:
        response = await bot.generate_response(message)
        yield response

async def main():
    bot = ChatBot("AI助手")
    user_messages = [
        "你好!",
        "今天天气真不错",
        "你对人工智能的发展有什么看法?",
        "谢谢你的回答"
    ]

    chat_generator = chat_stream(bot, user_messages)

    try:
        while True:
            response = await anext(chat_generator)
            print(f"用户: {user_messages.pop(0)}")
            print(response)
            print()
    except StopAsyncIteration:
        print("对话结束")

if __name__ == "__main__":
    asyncio.run(main())