blob: 17ad4412fd46fe3cfac79598e2c90eaf41f8009b (
plain)
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
|
silence_warnings do
require 'reel'
end
require 'logger'
require 'nickserver/config'
require 'nickserver/adapters/celluloid_http'
require 'nickserver/dispatcher'
require 'nickserver/logging_responder'
require 'nickserver/client_error'
module Nickserver
class ReelServer < Reel::Server::HTTP
def self.start(options = {})
new(options[:host], options[:port])
end
def initialize(host = "127.0.0.1", port = 3000)
Celluloid.logger = logger
super(host, port, &method(:on_connection))
end
def handle_connection(*args)
silence_warnings do
super
end
end
def on_connection(connection)
connection.each_request do |request|
handle_request(request)
end
end
protected
def handle_request(request)
log_request(request)
handler = handler_for(request)
handler.respond_to params(request), request.headers
rescue ClientError => e
logger.warn e
request.respond 400, JSON.generate(error: e.message)
rescue StandardError => e
logger.error e
request.respond 500, "{}"
end
def log_request(request)
logger.info "#{request.method} #{request.uri}"
logger.debug " #{params(request)}"
rescue URI::Error => e
raise ClientError, e.message
end
def handler_for(request)
# with reel the request is the responder
responder = LoggingResponder.new(request, logger)
Dispatcher.new(responder)
end
def params(request)
if request.query_string
CGI.parse request.query_string
else
CGI.parse request.body.to_s
end
end
def logger
@logger ||= ::Logger.new Config.log_file
end
end
end
|
