Class: Delayed::Command

Inherits:
Object show all
Defined in:
lib/delayed/command.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(args) ⇒ Command

Returns a new instance of Command.



9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
 
# File 'lib/delayed/command.rb', line 9

def initialize(args)
  @files_to_reopen = []
  @options = {
    :quiet => true,
    :pid_dir => "#{RAILS_ROOT}/tmp/pids",
    :log_file => "#{RAILS_ROOT}/log/delayed_job.log"
  }
  
  @worker_count = 1
  
  opts = OptionParser.new do |opts|
    opts.banner = "Usage: #{File.basename($0)} [options] start|stop|restart|run"

    opts.on('-h', '--help', 'Show this message') do
      puts opts
      exit 1
    end
    opts.on('-e', '--environment=NAME', 'Specifies the environment to run this delayed jobs under (test/development/production).') do |e|
      STDERR.puts "The -e/--environment option has been deprecated and has no effect. Use RAILS_ENV and see http://github.com/collectiveidea/delayed_job/issues/#issue/7"
    end
    opts.on('--min-priority N', 'Minimum priority of jobs to run.') do |n|
      @options[:min_priority] = n
    end
    opts.on('--max-priority N', 'Maximum priority of jobs to run.') do |n|
      @options[:max_priority] = n
    end
    opts.on('-n', '--number_of_workers=workers', "Number of unique workers to spawn") do |worker_count|
      @worker_count = worker_count.to_i rescue 1
    end
    opts.on('--pid-dir=DIR', 'Specifies an alternate directory in which to store the process ids.') do |dir|
      @options[:pid_dir] = dir
    end
    opts.on('--log-file=FILE', 'Specified an alternate log file.') do |file|
      @options[:log_file] = file
    end
    opts.on('-i', '--identifier=n', 'A numeric identifier for the worker.') do |n|
      @options[:identifier] = n
    end
    opts.on('--max-attempts=N', 'Maximum number of times a job is retried on error') do |n|
      @options[:max_attempts] = n
    end
  end
  @args = opts.parse!(args)
end

Instance Attribute Details

#worker_count ⇒ Object

Returns the value of attribute worker_count.



7
8
9
 
# File 'lib/delayed/command.rb', line 7

def worker_count
  @worker_count
end

Instance Method Details

#daemonize ⇒ Object 



54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
 
# File 'lib/delayed/command.rb', line 54

def daemonize
  Delayed::Worker.backend.before_fork

  ObjectSpace.each_object(File) do |file|
    @files_to_reopen << file unless file.closed?
  end
  
  dir = @options[:pid_dir]
  Dir.mkdir(dir) unless File.exists?(dir)
  
  if @worker_count > 1 && @options[:identifier]
    raise ArgumentError, 'Cannot specify both --number-of-workers and --identifier'
  elsif @worker_count == 1 && @options[:identifier]
    process_name = "delayed_job.#{@options[:identifier]}"
    run_process(process_name, dir)
  else
    worker_count.times do |worker_index|
      process_name = worker_count == 1 ? "delayed_job" : "delayed_job.#{worker_index}"
      run_process(process_name, dir)
    end
  end
end

#run(worker_name = nil) ⇒ Object 



83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
 
# File 'lib/delayed/command.rb', line 83

def run(worker_name = nil)
  Dir.chdir(RAILS_ROOT)
  
  # Re-open file handles
  @files_to_reopen.each do |file|
    begin
      file.reopen file.path
      file.sync = true
    rescue ::Exception
    end
  end

  Delayed::Worker.logger = Logger.new(@options[:log_file])
  Delayed::Worker.backend.after_fork

  Delayed::Worker.max_attempts = @options[:max_attempts].to_i if @options[:max_attempts] && @options[:max_attempts].to_i > 0

  worker = Delayed::Worker.new(@options)
  worker.name_prefix = "#{worker_name} "
  worker.start
rescue => e
  Rails.logger.fatal e
  STDERR.puts e.message
  exit 1
end

#run_process(process_name, dir) ⇒ Object 



77
78
79
80
81
 
# File 'lib/delayed/command.rb', line 77

def run_process(process_name, dir)
  Daemons.run_proc(process_name, :dir => dir, :dir_mode => :normal, :ARGV => @args) do |*args|
    run process_name
  end
end