从我所看到和理解的情况来看,当同时运行几个Luigi工作流程时,工人数量总和。这意味着如果我一起运行两个工作流并且工作者数量设置为 n ,则在luigi.cfg文件中并且假设工作流同时使用多于n个工作者,则scheduler将使用 2xn worker。
在Luigi的手册中,即使我同时运行十几个工作流程,我找不到任何方法将工作人员数量限制为 n 。
这是我的luigi.cfg文件
[core]
workers: 3
这是我正在使用的示例脚本(它实际上使用了sciluigi(luigi上面的一个层),但我认为它不会对任务和调度程序配置产生影响)。我喜欢这样,当我一起运行几次时,最后3个工作流程等待在开始之前完成三个第一个工作流程。
import optparse
import luigi
import sciluigi
import random
import time
import sys
import os
import subprocess
class MyFooWriter(sciluigi.Task):
# We have no inputs here
# Define outputs:
outdir = sciluigi.Parameter();
def out_foo(self):
return sciluigi.TargetInfo(self, os.path.join(self.outdir,'foo.txt'))
def run(self):
with self.out_foo().open('w') as foofile:
foofile.write('foo\n')
class MyFooReplacer(sciluigi.Task):
replacement = sciluigi.Parameter() # Here, we take as a parameter
# what to replace foo with.
outFile = sciluigi.Parameter();
outdir = sciluigi.Parameter();
# Here we have one input, a "foo file":
in_foo = None
# ... and an output, a "bar file":
def out_replaced(self):
return sciluigi.TargetInfo(self, os.path.join(self.outdir, self.outFile))
def run(self):
replacement = ""
with open(self.in_foo().path, 'r') as content_file:
content = content_file.read()
replacement = content.replace('foo', self.replacement)
for i in range(1,30):
sys.stderr.write(str(i)+"\n")
time.sleep(1)
with open(self.out_replaced().path,'w') as out_f:
out_f.write(replacement)
class MyWorkflow(sciluigi.WorkflowTask):
outdir = luigi.Parameter()
def workflow(self):
#rdint = random.randint(1,1000)
rdint = 100
barfile = "foobar_" + str(rdint) +'.bar.txt'
foowriter = self.new_task('foowriter', MyFooWriter, outdir = self.outdir)
fooreplacer = self.new_task('fooreplacer', MyFooReplacer, replacement='bar', outFile = barfile, outdir = self.outdir)
fooreplacer.in_foo = foowriter.out_foo
return fooreplacer
# End of script ....
if __name__ == '__main__':
parser = optparse.OptionParser()
parser.add_option('-d', dest = "outdir", action="store", default=".")
options, remainder = parser.parse_args()
params = {"outdir" : options.outdir}
wf = [MyWorkflow(outdir = options.outdir)]
luigi.build(wf)
这是我用来同时运行脚本的一个小perl脚本(用Perl,我最喜欢的语言: - ))。
#! /usr/bin/perl
use strict;
for (my $i = 0; $i < 6; $i++) {
my $testdir = "test".$i;
system("mkdir -p $testdir");
system("python run_sciluigi.py -d $testdir&");
sleep (2)
}
答案 0 :(得分:2)
虽然不完全是工作者限制,但可以使用resources
概念对并发执行施加全局限制。
在luigi.cfg
[resources]
max_workers=5
在你的所有任务中:
class MyFooReplacer(sciluigi.Task):
resources = {'max_workers': 1}
http://luigi.readthedocs.io/en/stable/configuration.html#resources