6

下面的代码似乎没有同时运行,我不确定为什么:

def run_normalizers(config, debug, num_threads, name=None):

    def _run():
        print('Started process for normalizer')
        sqla_engine = init_sqla_from_config(config)
        image_vfs = create_s3vfs_from_config(config, config.AWS_S3_IMAGE_BUCKET)
        storage_vfs = create_s3vfs_from_config(config, config.AWS_S3_STORAGE_BUCKET)

        pp = PipedPiper(config, image_vfs, storage_vfs, debug=debug)

        if name:
            pp.run_pipeline_normalizers(name)
        else:
            pp.run_all_normalizers()
        print('Normalizer process complete')

    threads = []
    for i in range(num_threads):
        threads.append(multiprocessing.Process(target=_run))
    [t.start() for t in threads]
    [t.join() for t in threads]


run_normalizers(...)

config变量只是在_run()函数之外定义的字典。所有的进程似乎都被创建了——但它并不比我用一个进程创建的快。基本上,函数中发生的事情run_**_normalizers()是从数据库(SQLAlchemy)中的队列表中读取数据,然后发出一些 HTTP 请求,然后运行规范化器的“管道”来修改数据,然后将其保存回数据库。我来自 JVM 领域,那里的线程“繁重”并且经常用于并行性 - 我对此有点困惑,因为我认为多进程模块应该绕过 Python 的 GIL 的限制。

4

2 回答 2

3

修复了我的多处理问题-实际上切换了线程。不知道实际上是什么修复了它的想法——我只是重新设计了所有东西,制作了工人和任务,还有什么不是,现在一切都在飞翔。这是我所做的基础知识:

import abc
from Queue import Empty, Queue
from threading import Thread

class AbstractTask(object):
    """
        The base task
    """
    __metaclass__ = abc.ABCMeta

    @abc.abstractmethod
    def run_task(self):
        pass

class TaskRunner(object):

    def __init__(self, queue_size, num_threads=1, stop_on_exception=False):
        super(TaskRunner, self).__init__()
        self.queue              = Queue(queue_size)
        self.execute_tasks      = True
        self.stop_on_exception  = stop_on_exception

        # create a worker
        def _worker():
            while self.execute_tasks:

                # get a task
                task = None
                try:
                    task = self.queue.get(False, 1)
                except Empty:
                    continue

                # execute the task
                failed = True
                try:
                    task.run_task()
                    failed = False
                finally:
                    if failed and self.stop_on_exception:
                        print('Stopping due to exception')
                        self.execute_tasks = False
                    self.queue.task_done()

        # start threads
        for i in range(0, int(num_threads)):
            t = Thread(target=_worker)
            t.daemon = True
            t.start()


    def add_task(self, task, block=True, timeout=None):
        """
            Adds a task
        """
        if not self.execute_tasks:
            raise Exception('TaskRunner is not accepting tasks')
        self.queue.put(task, block, timeout)


    def wait_for_tasks(self):
        """
            Waits for tasks to complete
        """
        if not self.execute_tasks:
            raise Exception('TaskRunner is not accepting tasks')
        self.queue.join()

我所做的就是创建一个 TaskRunner 并向其中添加任务(数千个),然后调用 wait_for_tasks()。所以,显然在我所做的重新架构中,我“修复”了我遇到的其他一些问题。不过很奇怪。

于 2013-07-26T20:01:50.040 回答
1

如果您仍在寻找多处理解决方案,您可能首先想了解如何使用工作人员池,然后您不必自己管理 num_threads 进程:http: //docs.python.org/ 2/library/multiprocessing.html#using-a-pool-of-workers

对于减速问题,您是否尝试过将配置对象作为参数传递给 _run 函数?我不知道这是否/如何在内部做出改变,但猜测它可能会改变一些东西。

于 2013-07-29T20:40:36.847 回答