协程的幕后英雄 - Generator - 简单理解协程工作原理

2024-02-10 01:46:21

协程 - 异步编程的幕后英雄

在现代编程中，异步编程变得越来越重要，因为它允许应用程序在等待输入或执行计算密集型任务时继续执行其他任务。而协程作为异步编程的关键技术之一，在 Python 中通过 Generator 实现，它提供了一种简单且高效的方式来编写异步代码。

Generator - Python 中的协程

Generator 是 Python 中的一个特殊函数，它允许您在函数中暂停执行并稍后恢复执行。通过使用 yield ，您可以将 Generator 函数变成一个协程，以便在需要时暂停执行并返回一个值。

为了更好地理解 Generator 的工作原理，我们来看一个简单的例子：

def my_generator():
    yield 1
    yield 2
    yield 3

for i in my_generator():
    print(i)

在这个例子中，my_generator() 是一个 Generator 函数，它使用 yield 关键字暂停执行并返回一个值。当我们在 for 循环中迭代 my_generator() 时，每次迭代都会调用一次 Generator 函数，并打印出 yield 的值。

Generator 和 asyncio - 协程在 Python 中的应用

在 Python 中，asyncio 是一个流行的异步编程库，它利用协程来实现异步 I/O 操作。asyncio 提供了多种内置函数，允许您使用协程来编写异步代码，例如 asyncio.run() 和 asyncio.create_task()。

以下是一个使用 asyncio 和 Generator 实现简单的 HTTP 服务器的例子：

import asyncio

async def handle_client(reader, writer):
    while True:
        data = await reader.read(100)
        if not data:
            break
        writer.write(data)

async def main():
    server = asyncio.start_server(handle_client, 'localhost', 8888)

    async with server:
        await server.serve_forever()

asyncio.run(main())