我正在开发一个Luigi管道,用于检查手动创建的文件是否存在,如果存在,继续执行下一个任务:
import luigi, os
class ExternalFileChecker(luigi.ExternalTask):
task_namespace='MyTask'
path = luigi.Parameter()
def output(self):
return luigi.LocalTarget(os.path.join(self.path, 'externalfile.txt'))
class ProcessExternalFile(luigi.Task):
task_namespace='MyTask'
path = luigi.Parameter()
def requires(self):
return ExternalFileChecker(path=self.path)
def output(self):
dirname = self.path
outfile = os.path.join(dirname, 'processedfile.txt')
return luigi.LocalTarget(outfile)
def run(self):
#do processing
if __name__ == '__main__':
path = r'D:\MyPath\luigi'
luigi.run(['MyTask.ProcessExternalFile','--path', path,\
'--worker-retry-external-tasks','--scheduler-retry-delay', '20',\
'--worker-keep-alive'])
我想要的是,在创建手册文件并将其粘贴到路径中后,luigi会继续。当我这样做时,它不是找到文件并继续执行任务,而是每隔几秒重新检查一个新任务:
DEBUG: There are no more tasks to run at this time
DEBUG: There are 2 pending tasks possibly being run by other workers
DEBUG: There are 2 pending tasks unique to this worker
DEBUG: Sleeping for 1.536391 seconds
DEBUG: Asking scheduler for work...
DEBUG: Done
DEBUG: There are no more tasks to run at this time
DEBUG: There are 2 pending tasks possibly being run by other workers
DEBUG: There are 2 pending tasks unique to this worker
DEBUG: Sleeping for 5.669132 seconds
DEBUG: Asking scheduler for work...
DEBUG: Done
(...)
经过相当长的时间(15-20分钟左右),luigi将找到该文件,然后它可以按照需要继续。我该怎么做才能防止这种延迟?我希望luigi在文件存在后立即继续。
答案 0 :(得分:3)
要记住的一些事项:
keep_alive = True
,在这种情况下,当没有更多待处理任务时它将退出。)retry_external_tasks
部分中的[worker]
配置设置控制。我认为你所观察到的是这样的。您的管道正在运行,任务ProcessExternalFile
失败,然后您添加文件,任务在retry_delay
期间保持FAILED,然后最终变为PENDING并且工作人员再次获得此任务,指出它发现文件并且任务变得完整。
这是否是您想要的行为取决于您。如果希望更快地找到该文件,则可以更改重试间隔。或者,您可以在while
方法中执行无限run
循环,并定期检查文件,并在找到时跳出循环。您还可以将Luigi配置为完全禁用重试逻辑。