python - 记录协程感知变量

Question

我正在遵循 asyncio.Queue 中给出的 worker 示例。在 worker_task() 调用的函数中，我想记录当前工作人员的姓名，而不必在任何地方传递它。

我怀疑通过 LoggerAdapter 和 ContextVar 的某种组合，它可以做我想做的事情。

接下来的事情......

import logging
logging.basicConfig(format="%(worker_name)s %(message)s")
logger = logging.getLogger(__name__)

async def sub_function():
    logger.info("a message") # worker-1: a message

async def worker_task(queue, worker_name):
    logger.do_something(worker_name) # HELP HERE PLEASE
    await sub_function()

queue = asyncio.Queue()

workers = [
    loop.create_task(worker_task(queue, "worker-1"),
    loop.create_task(worker_task(queue, "worker-2"),
]

Answer 1

你可以使用这样的东西:

import logging
import asyncio
from contextvars import ContextVar

WorkerName = ContextVar('worker_name')

logging.basicConfig(format='%(worker_name)s %(message)s', level=logging.INFO)

class WorkerAdapter(logging.LoggerAdapter):
    def process(self, msg, kwargs):
        kwargs.setdefault('extra', {})['worker_name'] = WorkerName.get()
        return msg, kwargs

logger = WorkerAdapter(logging.getLogger(__name__), None)


async def sub_function():
    logger.info('a message')

async def worker_task(worker_name):
    WorkerName.set(worker_name)
    await sub_function()

loop = asyncio.get_event_loop()

workers = [
    loop.create_task(worker_task('worker-1')),
    loop.create_task(worker_task('worker-2')),
]

loop.run_until_complete(asyncio.gather(*workers))