Class: DispatchServlet
- Inherits:
-
WEBrick::HTTPServlet::AbstractServlet
- Object
- WEBrick::HTTPServlet::AbstractServlet
- DispatchServlet
- Defined in:
- lib/webrick_server.rb
Overview
A custom dispatch servlet for use with WEBrick. It dispatches requests (using the Rails Dispatcher) to the appropriate controller/action. By default, it restricts WEBrick to a managing a single Rails request at a time, but you can change this behavior by setting ActionController::Base.allow_concurrency to true.
Constant Summary collapse
- REQUEST_MUTEX =
Mutex.new
Class Method Summary collapse
-
.dispatch(options = {}) ⇒ Object
Start the WEBrick server with the given options, mounting the DispatchServlet at
/
.
Instance Method Summary collapse
-
#handle_dispatch(req, res, origin = nil) ⇒ Object
:nodoc:.
-
#handle_file(req, res) ⇒ Object
:nodoc:.
-
#initialize(server, options) ⇒ DispatchServlet
constructor
:nodoc:.
-
#service(req, res) ⇒ Object
:nodoc:.
Constructor Details
#initialize(server, options) ⇒ DispatchServlet
:nodoc:
65 66 67 68 69 70 71 72 |
# File 'lib/webrick_server.rb', line 65 def initialize(server, ) #:nodoc: @server_options = @file_handler = WEBrick::HTTPServlet::FileHandler.new(server, [:server_root]) # Change to the RAILS_ROOT, since Webrick::Daemon.start does a Dir::cwd("/") # OPTIONS['working_directory'] is an absolute path of the RAILS_ROOT, set in erailties/lib/commands/servers/webrick.rb Dir.chdir(OPTIONS['working_directory']) if defined?(OPTIONS) && File.directory?(OPTIONS['working_directory']) super end |
Class Method Details
.dispatch(options = {}) ⇒ Object
Start the WEBrick server with the given options, mounting the DispatchServlet at /
.
50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
# File 'lib/webrick_server.rb', line 50 def self.dispatch( = {}) Socket.do_not_reverse_lookup = true # patch for OS X params = { :Port => [:port].to_i, :ServerType => [:server_type], :BindAddress => [:ip] } params[:MimeTypes] = [:mime_types] if [:mime_types] server = WEBrick::HTTPServer.new(params) server.mount('/', DispatchServlet, ) trap("INT") { server.shutdown } server.start end |
Instance Method Details
#handle_dispatch(req, res, origin = nil) ⇒ Object
:nodoc:
110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 |
# File 'lib/webrick_server.rb', line 110 def handle_dispatch(req, res, origin = nil) #:nodoc: 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) set_charset(header) assign_status(res, header) res..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
:nodoc:
89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
# File 'lib/webrick_server.rb', line 89 def handle_file(req, res) #:nodoc: 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
:nodoc:
74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
# File 'lib/webrick_server.rb', line 74 def service(req, res) #:nodoc: unless handle_file(req, res) begin REQUEST_MUTEX.lock unless ActionController::Base.allow_concurrency unless handle_dispatch(req, res) raise WEBrick::HTTPStatus::NotFound, "`#{req.path}' not found." end ensure unless ActionController::Base.allow_concurrency REQUEST_MUTEX.unlock if REQUEST_MUTEX.locked? end end end end |