从APScheduler收集期货

时间:2017-05-25 17:46:57

标签: python apscheduler concurrent.futures

我有以下代码:

import pandas as pd
from concurrent.futures import ThreadPoolExecutor, as_completed
from datetime import datetime
from apscheduler.schedulers.blocking import BlockingScheduler 


class FutureScheduler(object):

    def __init__():
        self.futures = []
        self.scheduler = BlockingScheduler()
        self.pool = ThreadPoolExecutor(5)
        self.full_frame = pd.DataFrame()

    def start(self):
        job = self.scheduler.add_job(self.add_future, 'cron', day_of_week='mon-fri', hour='8-15', minute='*')
        self.scheduler.start()
        self.flush_csvs()

    def add_future(self):
        self.futures.append(self.pool.submit(self.long_running_task))

    def flush_csvs(self):
        for future in as_completed(self.futures):
            results = future.result()
            self.full_frame = pd.concat((self.full_frame, results))
            self.full_frame.to_csv('results.csv')
            print "flushed... Queue size: %s" % len(self.futures)

    def long_running_task(self):
        #takes a while may or may not return before the next one is kicked off

所以我遇到的问题是flush_csvs循环内部的代码永远不会运行。在调用as_completed之前,是否必须将所有期货添加到列表中?有没有办法让BlockingScheduler回归未来?我看到它返回Job,但在这种情况下,我希望它更像未来。

1 个答案:

答案 0 :(得分:1)

这不起作用,因为调度程序阻止主线程继续。这可以防止flush_csvs被执行。

self.scheduler.start()
self.flush_csvs()

但是,这可能不是你想要的。 APScheduler在内部使用线程池,因此回调(self.long_running_task)已经在一个单独的线程中执行。

如果您需要多个核心(使用ProcessPoolExecutor而不是ThreadPoolExecutor)等,您可以通过APScheduler更改此线程池的配置,具体取决于您需要的工作者数量等。您也可以配置每个作业来执行你想要什么。例如,为每分钟运行一次的作业配置策略以合并(仅运行一个),而不是在发生延迟时背靠背运行多次。

http://apscheduler.readthedocs.io/en/latest/userguide.html#choosing-the-right-scheduler-job-store-s-executor-s-and-trigger-s