Class: APN::SenderDaemon

Inherits:
Object
  • Object
show all
Defined in:
lib/apn/sender_daemon.rb

Overview

A wrapper designed to daemonize an APN::Sender instance to keep in running in the background. Connects worker’s output to a custom logger, if available. Creates a pid file suitable for monitoring with monit.

Based off delayed_job’s great example, except we can be much lighter by not loading the entire Rails environment. To use in a Rails app, script/generate apn_sender.

Instance Method Summary collapse

Constructor Details

#initialize(args) ⇒ SenderDaemon

Returns a new instance of SenderDaemon.



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
 
# File 'lib/apn/sender_daemon.rb', line 17

def initialize(args)
  @options = {worker_count: 1, delay: 5}

  optparse = 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('--cert-path=PATH', 'Path to directory containing apn .pem certificates.') do |path|
      @options[:cert_root] = path
    end
    opts.on('c', '--full-cert-path=PATH', 'Full path to desired .pem certificate.') do |path|
      @options[:full_cert_path] = path
    end
    opts.on('--cert-pass=PASSWORD', 'Password for the apn .pem certificates.') do |pass|
      @options[:cert_pass] = pass
    end
    opts.on('--cert-name=NAME', 'Certificate file name. Default: apn_production.pem') do |certificate_name|
      @options[:certificate_name] = certificate_name
    end
    opts.on('-n', '--number-of-workers=WORKERS', "Number of unique workers to spawn") do |worker_count|
      @options[:worker_count] = worker_count.to_i rescue 1
    end
    opts.on('-d', '--delay=D', "Delay between rounds of work (seconds)") do |d|
      @options[:delay] = d
    end
  end

  # If no arguments, give help screen
  @args = optparse.parse!(args.empty? ? ['-h'] : args)
end

Instance Method Details

#daemonize ⇒ Object 



51
52
53
54
55
56
57
58
59
 
# File 'lib/apn/sender_daemon.rb', line 51

def daemonize
  @options[:worker_count].times do |worker_index|
    process_name = @options[:worker_count] == 1 ? "apn_sender" : "apn_sender.#{worker_index}"
    pids_dir = defined?(Rails) ? "#{::RAILS_ROOT}/tmp/pids" : "tmp/pids"
    Daemons.run_proc(process_name, :dir => pids_dir, :dir_mode => :normal, :ARGV => @args) do |*args|
      run(process_name)
    end
  end
end

#run(worker_name = nil) ⇒ Object 



61
62
63
64
65
66
67
68
69
70
71
72
 
# File 'lib/apn/sender_daemon.rb', line 61

def run(worker_name = nil)
  APN.password = @options[:cert_pass]
  APN.full_certificate_path = @options[:full_cert_path]
  APN.root = @options[:cert_root]
  APN.certificate_name = @options[:certificate_name]

  worker = ::Resque::Worker.new(APN::Jobs::QUEUE_NAME)
  worker.work(@options[:delay])
rescue => e
  STDERR.puts e.message
  exit 1
end