我正在尝试将threading.local()中的数据传递给不同模块中的函数。 代码是这样的:
other_module.py:
import threading
# 2.1
ll = threading.local()
def other_fn():
# 2.2
ll = threading.local()
v = getattr(ll, "v", None)
print(v)
main_module.py:
import threading
import other_module
# 1.1
ll = threading.local()
def main_fn(v):
# 1.2
ll = threading.local()
ll.v = v
other_fn()
for i in [1,2]:
t = threading.Thread(target=main_fn, args=(i,))
t.start()
但是组合1.x-2.x都不适合我。 我发现了类似的问题-Access thread local object in different module - Python,但回答,如果print_message函数位于不同模块中,则标记为答案对我也不起作用。
是否可以在模块之间传递线程本地数据 而不将其作为函数参数传递?
答案 0 :(得分:0)
在类似情况下,我最终在单独的模块中执行以下操作:
import threading
from collections import defaultdict
tls = defaultdict(dict)
def get_thread_ctx():
""" Get thread-local, global context"""
return tls[threading.get_ident()]
这实际上会创建一个称为tls的global
变量。然后,每个线程(基于其身份)都会在全局字典中获取一个密钥。我也将其作为命令处理。示例:
class Test(Thread):
def __init__(self):
super().__init__()
# note: we cannot initialize thread local here, since thread
# is not running yet
def run(self):
# Get thread context
tmp = get_thread_ctx()
# Create an app-specific entry
tmp["probe"] = {}
self.ctx = tmp["probe"]
while True:
...
现在,在另一个模块中:
def get_thread_settings():
ctx = get_thread_ctx()
probe_ctx = ctx.get("probe", None)
# Get what you need from the app-specific region of this thread
return probe_ctx.get("settings", {})
希望它可以帮助下一个寻找相似的东西