Class: Rack::Handler::Mongrel2

Inherits:
Object
  • Object
show all
Defined in:
lib/rack/handler/mongrel2.rb

Class Method Summary collapse

Class Method Details

.run(app, options = {}) ⇒ Object 



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
 
# File 'lib/rack/handler/mongrel2.rb', line 8

def run(app, options = {})
  options = {
    :recv => ENV['RACK_MONGREL2_RECV'] || 'tcp://127.0.0.1:9997',
    :send => ENV['RACK_MONGREL2_SEND'] || 'tcp://127.0.0.1:9996',
    :uuid => ENV['RACK_MONGREL2_UUID']
  }.merge(options)

  raise ArgumentError.new('Must specify an :uuid or set RACK_MONGREL2_UUID') if options[:uuid].nil?

  conn = ::Mongrel2::Connection.new(options[:uuid], options[:recv], options[:send])

  running = true

  # This doesn't work at all until zmq fixes their shit (in 2.1.x I think), but trap it now anyway.
  %w(INT TERM KILL).each do |sig|
    trap(sig) do
      conn.close
      running = false
    end
  end

  while running
    req = conn.recv rescue nil
    next if req.nil? || req.disconnect?
    break if !running

    script_name = ENV['RACK_RELATIVE_URL_ROOT'] || req.headers['PATTERN'].split('(', 2).first.gsub(/\/$/, '')
    env = {
      'rack.version' => Rack::VERSION,
      'rack.url_scheme' => 'http', # Only HTTP for now
      'rack.input' => StringIO.new(req.body),
      'rack.errors' => $stderr,
      'rack.multithread' => true,
      'rack.multiprocess' => true,
      'rack.run_once' => false,
      'mongrel2.pattern' => req.headers['PATTERN'],
      'REQUEST_METHOD' => req.headers['METHOD'],
  'CONTENT_TYPE' => req.headers['content-type'],
      'SCRIPT_NAME' => script_name,
      'PATH_INFO' => req.headers['PATH'].gsub(script_name, ''),
      'QUERY_STRING' => req.headers['QUERY'] || ''
    }

    env['SERVER_NAME'], env['SERVER_PORT'] = req.headers['host'].split(':', 2)
    req.headers.each do |key, val|
      unless key =~ /content_(type|length)/i
        key = "HTTP_#{key.upcase.gsub('-', '_')}"
      end
      env[key] = val
    end

    status, headers, rack_response = app.call(env)
    body = ''
    rack_response.each { |b| body << b }
    conn.reply(req, body, status, headers)
  end
ensure
  conn.close if conn.respond_to?(:close)
end