使用 asyncio 和 contextvars 在 python 中的两个异步程序之间共享状态

Question

我目前有两个无限异步任务正在运行，并且希望在它们之间共享状态。一项任务是读取消息然后发送消息的 WebSocket 连接，另一项任务读取传入的光数据。我想在两个任务之间发送一个布尔值，表明 websocket 连接是否成功。

这是我初始化上下文的方式。

client_connect_var = contextvars.ContextVar('client_connect',default = False)
client_connect_var.set(False)
ctx = contextvars.copy_context()
async def main():
  message = json.dumps({'payload': {
                        'payload'})
  loop = asyncio.get_event_loop()
  start_light = asyncio.create_task(calculate_idle(3))
  await asyncio.gather(init_connection(message), start_light)

ctx.run(asyncio.run(main()))

这是我的 init_connection 中的代码：

async def init_connection(message):
  async with websockets.connect(uri) as websocket:
      #This should set the global context variable to true
      client_connect_var.set(True)
      CLIENT_WS = websocket
      client_connect = client_connect_var.get()
    # send init message
    await websocket.send(message)
    print("Connection is open")
    while client_connect:
        await handleMessages(websocket, message)
    await websocket.close()

这是它试图获取灯光代码中当前状态的地方

async def calculate_idle(t):
    orig_time = t
    while True:
        await asyncio.sleep(5)
        #This should be true, but it's false
        client_connect = client_connect_var.get()
        await asyncio.sleep(5)

在calculate_idle中，上下文中的变量仍然设置为false。我不确定如何从 init_connection 内部将其设置为 true。我想将其扩展为更新对象状态中的其他值。任何帮助，将不胜感激！

Answer 1

ContextVar其用途与您尝试使用它的目的相反：它用于任务本地的数据，类似于“任务本地”全局变量。

要在两个任务之间共享数据，您可以使用普通的 Python 共享机制，例如全局变量或普通的可变值（例如字典或您选择的类的实例）。您还可以使用同步asyncio.Event（https://docs.python.org/3/library/asyncio-sync.html#asyncio.Event）在值更改时通知其他任务。