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

时间:2011-07-23 14:22:27

标签: python multithreading callback notifications

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

我知道我可以通过Thread.isActive()方法检查所有线程的状态,但轮询很糟糕,所以我想收到通知。

我正在考虑给Threads一个回调方法,并在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?
    ...

谢谢!

3 个答案:

答案 0 :(得分:20)

线程无法调用管理器,除非它具有对管理器的引用。最简单的方法是让管理员在实例化时将它交给线程。

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()

如果您希望能够将任意函数或方法指定为回调,而不是存储对管理器对象的引用,则由于方法包装器等原因,这会变得有点问题。很难设计回调,因此它会获得对管理器线程的引用,这是您想要的。我已经研究了一段时间,并没有想出任何我认为有用或优雅的东西。

答案 1 :(得分:9)

这样做有什么不妥吗?

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"

答案 2 :(得分:3)

如果您希望主线程等待子线程完成执行,那么最好使用某种同步机制。如果只是在一个或多个线程执行完毕后得到通知,则Condition就足够了:

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可能更好,因为它使处理多个工作线程更容易一些。