Class: DispatchServlet

Inherits:
WEBrick::HTTPServlet::AbstractServlet
  • Object
show all
Defined in:
lib/webrick_server.rb

Constant Summary collapse

REQUEST_MUTEX = 
Mutex.new

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(server, options) ⇒ DispatchServlet

Returns a new instance of DispatchServlet.



60
61
62
63
64
65
 
# File 'lib/webrick_server.rb', line 60

def initialize(server, options)
  @server_options = options
  @file_handler = WEBrick::HTTPServlet::FileHandler.new(server, options[:server_root])
  Dir.chdir(ABSOLUTE_RAILS_ROOT)
  super
end

Class Method Details

.dispatch(options = {}) ⇒ Object 



46
47
48
49
50
51
52
53
54
55
56
57
58
 
# File 'lib/webrick_server.rb', line 46

def self.dispatch(options = {})
  Socket.do_not_reverse_lookup = true # patch for OS X

  server = WEBrick::HTTPServer.new(:Port => options[:port].to_i, :ServerType => options[:server_type], :BindAddress => options[:ip])
  server.mount('/', DispatchServlet, options)

  trap("INT") { server.shutdown }

  require File.join(@server_options[:server_root], "..", "config", "environment") unless defined?(RAILS_ROOT)
  require "dispatcher"

  server.start
end

Instance Method Details

#handle_dispatch(req, res, origin = nil) ⇒ Object 



103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
 
# File 'lib/webrick_server.rb', line 103

def handle_dispatch(req, res, origin = nil)    
  data = StringIO.new
  Dispatcher.dispatch(
    CGI.new("query", create_env_table(req, origin), StringIO.new(req.body || "")), 
    ActionController::CgiRequest::DEFAULT_SESSION_OPTIONS, 
    data
  )

  header, body = extract_header_and_body(data)
  assign_status(res, header)
  res.cookies.concat(header.delete('set-cookie'))
  header.each { |key, val| res[key] = val.join(", ") }
  
  res.body = body
  return true
rescue => err
  p err, err.backtrace
  return false
end

#handle_file(req, res) ⇒ Object 



82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
 
# File 'lib/webrick_server.rb', line 82

def handle_file(req, res)
  begin
    req = req.dup
    path = req.path.dup

    # Add .html if the last path piece has no . in it
    path << '.html' if path != '/' && (%r{(^|/)[^./]+$} =~ path) 
    path.gsub!('+', ' ') # Unescape + since FileHandler doesn't do so.

    req.instance_variable_set(:@path_info, path) # Set the modified path...
    
    @file_handler.send(:service, req, res)
    return true
  rescue HTTPStatus::PartialContent, HTTPStatus::NotModified => err
    res.set_error(err)
    return true
  rescue => err
    return false
  end
end

#service(req, res) ⇒ Object 



67
68
69
70
71
72
73
74
75
76
77
78
79
80
 
# File 'lib/webrick_server.rb', line 67

def service(req, res)
  begin
    unless handle_file(req, res)
      REQUEST_MUTEX.lock unless ActionController::Base.allow_concurrency
      unless handle_dispatch(req, res)
        raise WEBrick::HTTPStatus::NotFound, "`#{req.path}' not found."
      end
    end
  ensure
    unless ActionController::Base.allow_concurrency
      REQUEST_MUTEX.unlock if REQUEST_MUTEX.locked?
    end
  end
end