如何在 Python 中传递并运行回调方法

Question

我有一个管理器（主线程），它创建其他线程来处理各种操作。 我希望我的经理在它创建的线程结束时（当 run() 方法执行完成时）收到通知。

我知道我可以通过使用 Thread.isActive() 方法检查所有线程的状态来做到这一点，但轮询很糟糕，所以我想收到通知。

我正在考虑为线程提供一个回调方法，并在 run() 方法的末尾调用此函数：

class Manager():
    ...
    MyThread(self.on_thread_finished).start() # How do I pass the callback

    def on_thread_finished(self, data):
        pass
    ...

class MyThread(Thread):
    ...
    def run(self):
        ....
        self.callback(data) # How do I call the callback?
    ...

谢谢！

Answer 1

线程不能调用管理器，除非它有对管理器的引用。实现这一点的最简单方法是管理器在实例化时将其交给线程。

class Manager(object):
    def new_thread(self):
        return MyThread(parent=self)
    def on_thread_finished(self, thread, data):
        print thread, data

class MyThread(Thread):

    def __init__(self, parent=None):
        self.parent = parent
        super(MyThread, self).__init__()

    def run(self):
        # ...
        self.parent and self.parent.on_thread_finished(self, 42)

mgr    = Manager()
thread = mgr.new_thread()
thread.start()

如果您希望能够分配任意函数或方法作为回调，而不是存储对管理器对象的引用，那么由于方法包装器等，这会变得有点问题。设计回调很困难，因此它会获取对管理器和线程的引用，这正是您想要的。我为此工作了一段时间，但没有想出任何我认为有用或优雅的东西。

Answer 2

这样做有什么问题吗？

from threading import Thread

class Manager():
    def Test(self):
        MyThread(self.on_thread_finished).start()

    def on_thread_finished(self, data):
        print "on_thread_finished:", data

class MyThread(Thread):
    def __init__(self, callback):
        Thread.__init__(self)
        self.callback = callback

    def run(self):
        data = "hello"
        self.callback(data)

m = Manager()
m.Test() # prints "on_thread_finished: hello"

Answer 3

如果您希望主线程等待子线程完成执行，那么最好使用某种同步机制。如果只是在一个或多个线程完成执行时收到通知，则 条件就足够了：

import threading

class MyThread(threading.Thread):
    def __init__(self, condition):
        threading.Thread.__init__(self)
        self.condition = condition

    def run(self):
        print "%s done" % threading.current_thread()
        with self.condition:
            self.condition.notify()


condition = threading.Condition()
condition.acquire()

thread = MyThread(condition)
thread.start()

condition.wait()

但是，使用 Queue 可能更好，因为它可以处理多个工作线程容易一点。