From 239829d669857a929cbd68a25dca3bd7c7b91822 Mon Sep 17 00:00:00 2001 From: Nat Lasseter Date: Wed, 4 Mar 2020 10:38:24 +0000 Subject: Added jobs-path argument --- lib/longboat/jobs.rb | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) (limited to 'lib/longboat/jobs.rb') diff --git a/lib/longboat/jobs.rb b/lib/longboat/jobs.rb index f0ada5a..5166993 100644 --- a/lib/longboat/jobs.rb +++ b/lib/longboat/jobs.rb @@ -4,17 +4,19 @@ module Longboat @collector = collector @jobs = [] @config = config - end - def load! - Dir.entries("./lib/jobs/").each do |file| - next if file =~ /^\./ + @config[:jobs_path].each do |dir| + next unless Dir.exist?(dir) + + Dir.entries(dir).each do |file| + next if file =~ /^\./ - reqname = File.basename(file, ".rb") - cname = reqname.split('_').map(&:capitalize).join + reqname = File.basename(file, ".rb") + cname = reqname.split('_').map(&:capitalize).join - require "jobs/#{reqname}" - @jobs << Kernel.const_get(cname).new(@collector, job_config) + require "jobs/#{reqname}" + @jobs << Kernel.const_get(cname).new(@collector, job_config) + end end end -- cgit v1.2.1