зеркало из https://github.com/github/ruby.git
121 строка
2.5 KiB
Ruby
121 строка
2.5 KiB
Ruby
require 'webrick'
|
|
require 'drb/drb'
|
|
require_relative 'http0'
|
|
|
|
module DRb
|
|
module HTTP0
|
|
|
|
def self.open_server(uri, config)
|
|
unless /^http:/ =~ uri
|
|
raise(DRbBadScheme, uri) unless uri =~ /^http:/
|
|
raise(DRbBadURI, 'can\'t parse uri:' + uri)
|
|
end
|
|
Server.new(uri, config)
|
|
end
|
|
|
|
class Callback < WEBrick::HTTPServlet::AbstractServlet
|
|
def initialize(config, drb)
|
|
@config = config
|
|
@drb = drb
|
|
@queue = Thread::Queue.new
|
|
end
|
|
|
|
def do_POST(req, res)
|
|
@req = req
|
|
@res = res
|
|
@drb.push(self)
|
|
@res.body = @queue.pop
|
|
@res['content-type'] = 'application/octet-stream;'
|
|
end
|
|
|
|
def req_body
|
|
@req.body
|
|
end
|
|
|
|
def reply(body)
|
|
@queue.push(body)
|
|
end
|
|
|
|
def close
|
|
@queue.push('')
|
|
end
|
|
end
|
|
|
|
class Server
|
|
def initialize(uri, config)
|
|
@uri = uri
|
|
@config = config
|
|
@queue = Thread::Queue.new
|
|
setup_webrick(uri)
|
|
end
|
|
attr_reader :uri
|
|
|
|
def close
|
|
@server.shutdown if @server
|
|
@server = nil
|
|
end
|
|
|
|
def push(callback)
|
|
@queue.push(callback)
|
|
end
|
|
|
|
def accept
|
|
client = @queue.pop
|
|
ServerSide.new(uri, client, @config)
|
|
end
|
|
|
|
def setup_webrick(uri)
|
|
logger = WEBrick::Log::new($stderr, WEBrick::Log::FATAL)
|
|
u = URI.parse(uri)
|
|
s = WEBrick::HTTPServer.new(:Port => u.port,
|
|
:AddressFamily => Socket::AF_INET,
|
|
:BindAddress => u.host,
|
|
:Logger => logger,
|
|
:ServerType => Thread)
|
|
s.mount(u.path, Callback, self)
|
|
@server = s
|
|
s.start
|
|
end
|
|
end
|
|
|
|
class ServerSide
|
|
def initialize(uri, callback, config)
|
|
@uri = uri
|
|
@callback = callback
|
|
@config = config
|
|
@msg = DRbMessage.new(@config)
|
|
@req_stream = StrStream.new(@callback.req_body)
|
|
end
|
|
attr_reader :uri
|
|
|
|
def close
|
|
@callback.close if @callback
|
|
@callback = nil
|
|
end
|
|
|
|
def alive?; false; end
|
|
|
|
def recv_request
|
|
begin
|
|
@msg.recv_request(@req_stream)
|
|
rescue
|
|
close
|
|
raise $!
|
|
end
|
|
end
|
|
|
|
def send_reply(succ, result)
|
|
begin
|
|
return unless @callback
|
|
stream = StrStream.new
|
|
@msg.send_reply(stream, succ, result)
|
|
@callback.reply(stream.buf)
|
|
rescue
|
|
close
|
|
raise $!
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|