如何在异步任务未完成时赋值变量？无限循环更新变量问题

核心思路 #

你的需求本质是在无限循环中，确保所有异步任务执行完毕后，再统一更新变量x。原代码的问题在于异步任务直接修改全局变量，导致更新时机不可控，且无法保证所有任务完成后再操作。我们可以通过让任务返回结果、利用TaskGroup的等待特性来解决。

重构后的代码 #

import asyncio, random

async def coro1():
    await asyncio.sleep(random.randint(1,5))
    return 5  # 任务返回结果，而非直接修改全局变量

async def coro2():
    await asyncio.sleep(random.randint(1,5))
    return 10  # 任务返回结果，而非直接修改全局变量

async def main():
    x = 1
    print(f"初始x值: {x}")
    while True:
        async with asyncio.TaskGroup() as tg:
            # 提交任务并保存任务对象
            task1 = tg.create_task(coro1())
            task2 = tg.create_task(coro2())
        
        # 所有任务完成后，根据需求更新x
        # 这里可以自定义更新逻辑，比如取最大值、指定任务结果等
        x = task2.result()
        print(f"更新后的x值: {x}")
        # 可选：添加循环间隔，避免无停顿循环
        # await asyncio.sleep(2)

asyncio.run(main())

关键细节说明 #

• 任务返回结果：将coro1和coro2改为返回计算值，而非直接修改全局变量，让变量更新的逻辑集中在main函数中，更可控。
• TaskGroup的自动等待：async with asyncio.TaskGroup()代码块会阻塞直到所有提交的任务完成，确保只有当两个异步任务都执行完毕后，才会进入变量更新步骤。
• 无限循环中的更新：每次循环都会创建新的任务组，重复“执行任务→等待完成→更新变量”的流程，完全匹配你的场景。

自定义更新逻辑示例 #

如果需要根据两个任务的结果综合调整x，可以修改赋值逻辑：

# 示例1：取两个结果的最大值
x = max(task1.result(), task2.result())
# 示例2：随机选择一个任务的结果
x = task1.result() if random.choice([True, False]) else task2.result()
# 示例3：合并结果（如果业务需要）
x = f"{task1.result()}-{task2.result()}"

内容的提问来源于stack exchange，提问作者Naadiyaar