不同模块中的python threading.local()

时间:2018-12-13 15:02:15

标签: python multithreading module local

我正在尝试将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函数位于不同模块中,则标记为答案对我也不起作用。

是否可以在模块之间传递线程本地数据 而不将其作为函数参数传递?

1 个答案:

答案 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", {})

希望它可以帮助下一个寻找相似的东西