我正在關(guān)注asyncio.Queue. 在由worker_task()我調(diào)用的函數(shù)中，我想記錄當前的工作人員姓名，而不必將其傳遞到任何地方。我經(jīng)歷的某種組合懷疑LoggerAdapter，并ContextVar可以做我想做的。以下內(nèi)容...import logginglogging.basicConfig(format="%(worker_name)s %(message)s")logger = logging.getLogger(__name__)async def sub_function(): logger.info("a message") # worker-1: a messageasync 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"),]
1 回答
心有法竹
TA貢獻1866條經(jīng)驗 獲得超5個贊
你可以使用這樣的東西：
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))
添加回答
舉報
0/150
提交
取消