如何通过引用跨线程和模块传递布尔值

2022-04-17 00:00:00 python python-multithreading

问题描述

我有一个布尔值,我想将其传递给执行来自不同模块的方法的不同线程。此布尔值充当取消令牌,因此如果设置,线程应该退出。它似乎是通过值传递的,因为如果我在另一个线程中设置它,它在其他线程中不会改变。谢谢。

import module2
from threading import Thread

cancellationToken = False

def main:
    thread2 = Thread(target = module2.method2, args (on_input, cancellationToken, ))
    thread2.start()
    ...
    thread2.join()

def on_input(command):
    global cancellationToken
    ...
    if(...):
        cancellationToken = True
    ...

模块2中的方法2只是一个简单的无限While循环,它检查取消令牌并响应用户输入。

def method2(on_input, cancellationToken):
    while(True):
        if(cancellationToken):
            return
        ...
        on_input(...)

解决方案

执行此操作时:

thread2 = Thread(target = module2.method2, args (on_input, cancellationToken, ))

您实际上是将第二个参数的值False传递给线程方法。

但当您在此之后执行此操作时:

cancellationToken = True

您替换的是cancellationToken表示的引用,而不是最初传递给thread2的值。

要实现您想要的操作,您需要为取消状态创建一个可变对象包装:

class CancellationToken:
   def __init__(self):
       self.is_cancelled = False

   def cancel(self):
       self.is_cancelled = True

cancellationToken = CancellationToken()

thread2 = Thread(target = module2.method2, args (on_input, cancellationToken, ))

# then later on
cancellationToken.cancel()

您的线程代码变为:

def method2(on_input, cancellationToken):
    while(True):
        if(cancellationToken.is_cancelled):
            return
        ...
        on_input(...)

相关文章