aboutsummaryrefslogtreecommitdiffstats
path: root/railties/lib/commands/process/spawner.rb
blob: fd09daa55bac0163877c6ab4911f7ab0a2205857 (plain) (blame)
1
2
3
4
5
6
7
8
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
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
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
require 'active_support'
require 'optparse'
require 'socket'
require 'fileutils'

def daemonize #:nodoc:
  exit if fork                   # Parent exits, child continues.
  Process.setsid                 # Become session leader.
  exit if fork                   # Zap session leader. See [1].
  Dir.chdir "/"                  # Release old working directory.
  File.umask 0000                # Ensure sensible umask. Adjust as needed.
  STDIN.reopen "/dev/null"       # Free file descriptors and
  STDOUT.reopen "/dev/null", "a" # point them somewhere sensible.
  STDERR.reopen STDOUT           # STDOUT/ERR should better go to a logfile.
end

class Spawner
  def self.record_pid(name = "#{OPTIONS[:process]}.spawner", id = Process.pid)
    FileUtils.mkdir_p(OPTIONS[:pids])
    File.open(File.expand_path(OPTIONS[:pids] + "/#{name}.pid"), "w+") { |f| f.write(id) }
  end

  def self.spawn_all
    OPTIONS[:instances].times do |i|
      port = OPTIONS[:port] + i
      print "Checking if something is already running on #{OPTIONS[:address]}:#{port}..."

      begin
        srv = TCPServer.new(OPTIONS[:address], port)
        srv.close
        srv = nil

        puts "NO"
        puts "Starting dispatcher on port: #{OPTIONS[:address]}:#{port}"

        FileUtils.mkdir_p(OPTIONS[:pids])
        spawn(port)
      rescue
        puts "YES"
      end
    end
  end
end

class FcgiSpawner < Spawner
  def self.spawn(port)
    cmd = "#{OPTIONS[:spawner]} -f #{OPTIONS[:dispatcher]} -p #{port} -P #{OPTIONS[:pids]}/#{OPTIONS[:process]}.#{port}.pid"
    cmd << " -a #{OPTIONS[:address]}" if can_bind_to_custom_address?
    system(cmd)
  end

  def self.can_bind_to_custom_address?
    @@can_bind_to_custom_address ||= /^\s-a\s/.match `#{OPTIONS[:spawner]} -h`
  end
end

class MongrelSpawner < Spawner
  def self.spawn(port)
    cmd =
      "mongrel_rails start -d " +
      "-a #{OPTIONS[:address]} " +
      "-p #{port} " +
      "-P #{OPTIONS[:pids]}/#{OPTIONS[:process]}.#{port}.pid " +
      "-e #{OPTIONS[:environment]} " +
      "-c #{OPTIONS[:rails_root]} " +
      "-l #{OPTIONS[:rails_root]}/log/mongrel.log"

    # Add prefix functionality to spawner's call to mongrel_rails
    # Digging through monrel's project subversion server, the earliest
    # Tag that has prefix implemented in the bin/mongrel_rails file
    # is 0.3.15 which also happens to be the earilest tag listed.
    # References: http://mongrel.rubyforge.org/svn/tags
    if Mongrel::Const::MONGREL_VERSION.to_f >=0.3 && !OPTIONS[:prefix].nil?
      cmd = cmd + " --prefix #{OPTIONS[:prefix]}"
    end
    system(cmd)
  end
  
  def self.can_bind_to_custom_address?
    true
  end
end


begin
  require_library_or_gem 'fcgi'
rescue Exception
  # FCGI not available
end

begin
  require_library_or_gem 'mongrel'
rescue Exception
  # Mongrel not available
end

server = case ARGV.first
  when "fcgi", "mongrel"
    ARGV.shift
  else
    if defined?(Mongrel)
      "mongrel"
    elsif RUBY_PLATFORM !~ /(:?mswin|mingw)/ && !silence_stderr { `spawn-fcgi -version` }.blank? && defined?(FCGI)
      "fcgi"
    end
end

case server
  when "fcgi"
    puts "=> Starting FCGI dispatchers"
    spawner_class = FcgiSpawner
  when "mongrel"
    puts "=> Starting mongrel dispatchers"
    spawner_class = MongrelSpawner
  else
    puts "Neither FCGI (spawn-fcgi) nor Mongrel was installed and available!"
    exit(0)
end



OPTIONS = {
  :environment => "production",
  :spawner     => '/usr/bin/env spawn-fcgi',
  :dispatcher  => File.expand_path(RELATIVE_RAILS_ROOT + '/public/dispatch.fcgi'),
  :pids        => File.expand_path(RELATIVE_RAILS_ROOT + "/tmp/pids"),
  :rails_root  => File.expand_path(RELATIVE_RAILS_ROOT),
  :process     => "dispatch",
  :port        => 8000,
  :address     => '0.0.0.0',
  :instances   => 3,
  :repeat      => nil,
  :prefix      => nil
}

ARGV.options do |opts|
  opts.banner = "Usage: spawner [platform] [options]"

  opts.separator ""

  opts.on <<-EOF
  Description:
    The spawner is a wrapper for spawn-fcgi and mongrel that makes it
    easier to start multiple processes running the Rails dispatcher. The
    spawn-fcgi command is included with the lighttpd web server, but can
    be used with both Apache and lighttpd (and any other web server
    supporting externally managed FCGI processes). Mongrel automatically
    ships with with mongrel_rails for starting dispatchers.

    The first choice you need to make is whether to spawn the Rails
    dispatchers as FCGI or Mongrel. By default, this spawner will prefer
    Mongrel, so if that's installed, and no platform choice is made,
    Mongrel is used.

    Then decide a starting port (default is 8000) and the number of FCGI
    process instances you'd like to run. So if you pick 9100 and 3
    instances, you'll start processes on 9100, 9101, and 9102.

    By setting the repeat option, you get a protection loop, which will
    attempt to restart any FCGI processes that might have been exited or
    outright crashed.

    You can select bind address for started processes. By default these
    listen on every interface. For single machine installations you would
    probably want to use 127.0.0.1, hiding them form the outside world.

     Examples:
       spawner               # starts instances on 8000, 8001, and 8002
                             # using Mongrel if available.
       spawner fcgi          # starts instances on 8000, 8001, and 8002
                             # using FCGI.
       spawner mongrel -i 5  # starts instances on 8000, 8001, 8002,
                             # 8003, and 8004 using Mongrel.
       spawner -p 9100 -i 10 # starts 10 instances counting from 9100 to
                             # 9109 using Mongrel if available.
       spawner -p 9100 -r 5  # starts 3 instances counting from 9100 to
                             # 9102 and attempts start them every 5
                             # seconds.
       spawner -a 127.0.0.1  # starts 3 instances binding to localhost
  EOF

  opts.on("  Options:")

  opts.on("-p", "--port=number",      Integer, "Starting port number (default: #{OPTIONS[:port]})")                { |OPTIONS[:port]| }

  if spawner_class.can_bind_to_custom_address?
    opts.on("-a", "--address=ip",     String,  "Bind to IP address (default: #{OPTIONS[:address]})")                { |OPTIONS[:address]| }
  end

  opts.on("-p", "--port=number",      Integer, "Starting port number (default: #{OPTIONS[:port]})")                { |v| OPTIONS[:port] = v }
  opts.on("-i", "--instances=number", Integer, "Number of instances (default: #{OPTIONS[:instances]})")            { |v| OPTIONS[:instances] = v }
  opts.on("-r", "--repeat=seconds",   Integer, "Repeat spawn attempts every n seconds (default: off)")             { |v| OPTIONS[:repeat] = v }
  opts.on("-e", "--environment=name", String,  "test|development|production (default: #{OPTIONS[:environment]})")  { |v| OPTIONS[:environment] = v }
  opts.on("-P", "--prefix=path",      String,  "URL prefix for Rails app. [Used only with Mongrel > v0.3.15]: (default: #{OPTIONS[:prefix]})")         { |v| OPTIONS[:prefix] = v }
  opts.on("-n", "--process=name",     String,  "default: #{OPTIONS[:process]}")                                    { |v| OPTIONS[:process] = v }
  opts.on("-s", "--spawner=path",     String,  "default: #{OPTIONS[:spawner]}")                                    { |v| OPTIONS[:spawner] = v }
  opts.on("-d", "--dispatcher=path",  String,  "default: #{OPTIONS[:dispatcher]}") { |dispatcher| OPTIONS[:dispatcher] = File.expand_path(dispatcher) }

  opts.separator ""

  opts.on("-h", "--help", "Show this help message.") { puts opts; exit }

  opts.parse!
end

ENV["RAILS_ENV"] = OPTIONS[:environment]

if OPTIONS[:repeat]
  daemonize
  trap("TERM") { exit }
  spawner_class.record_pid

  loop do
    spawner_class.spawn_all
    sleep(OPTIONS[:repeat])
  end
else
  spawner_class.spawn_all
end