扭曲-线程运行

时间:2019-05-16 23:39:50

标签: python multithreading twisted

python新手 扭曲的新手

我的团队希望我让一些现有代码在单独的线程中运行。

我想出一个虚构的例子:

from twisted.internet import threads, reactor
from twisted.internet.defer import inlineCallbacks
from time import sleep


class SomeClass(object):
    def __init__(self):
        self.working = False

    def set_working(self, is_working):
        self.working = is_working
        print 'Flag set to {}'.format(is_working)

    @inlineCallbacks
    def do_worker_thread(self):

        # I want to make this call on the main thread
        self.set_working(True)

        # I want to do all this garbage on a separate thread and keep trucking on the main thread
        # This mimics some calls in the real code. There is a call to deferToThread and a try
        # except block there.
        def thread_proc():
            try:
                for i in range(0, 100):
                    print 'Step %d starting'.format(i)
                    self.execute_step(i)
            except Exception:
                print 'An exception happened'
        reactor.callInThread(thread_proc)

        # When the worker thread is done, I want to call back 'on_thread_work_done'

    def execute_step(self, num):
        sleep(num)
        print 'Worker thread: %d'.format(num)

    def on_thread_work_done(self):
        """I want to be called back when the worker thread is done"""
        self.set_working(False)

    @inlineCallbacks
    def do_main_thread(self):
        for c in range(ord('a'), ord('z')+1):
            sleep(c)
            print 'Main thread: {}'.format(c)


if __name__ == "__main__":
    someClass = SomeClass()
    result = someClass.do_worker_thread()
    result.addCallback(someClass.do_main_thread())

    reactor.run()

do_worker_thread上的内容当前在主线程上运行。我在其中添加了注释,希望它在单独的线程中运行。重要的是do_worker_thread立即返回。

我希望输出看起来像:

Flag set to True
Step 0 starting
Main thread: a
Worker thread: 0
Worker thread: 1
Main thread: b
Worker thread: 2
Main thread: c
...

如何更改do_worker_thread中的内容,以使set_working调用位于主线程上,并且在工作线程完成工作之前不会将其设置为False?

2 个答案:

答案 0 :(得分:1)

尝试一下。它使用pure在主线程上安排工作。

callFromThread()

答案 1 :(得分:0)