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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
module Oxidized
require 'webrick'
require 'json'
module API
class Rest
module Helpers
def send res, json, msg='OK', status=200
res.body = msg
if not json
msg = {:result => msg}
res['Content-Type'] = 'application/json'
res.status = status
res.body = JSON.dump msg
end
end
end
include Oxidized::API::Rest::Helpers
def initialize nodes, listen
@nodes = nodes
addr, port = listen.to_s.split ':'
port, addr = addr, nil if not port
@web = WEBrick::HTTPServer.new :BindAddress=>addr, :Port=>port, :Logger=>Log, :AccessLog=>[]
maps
end
def work
req = select @web.listeners, nil, nil, Config::Sleep
while req
@web.run req.first.first.accept
req = select @web.listeners, nil, nil, 0
end
end
def maps
@web.mount '/nodes/next', Next, @nodes
@web.mount_proc '/nodes' do |req, res|
#script_name, #path_info
case req.path_info[1..-1]
# /nodes/reload - reloads list of nodes
when 'reload'
@nodes.load
send res
# /nodes/list - returns list of nodes
when 'list'
send res, @nodes.list
# /nodes/show/node - returns data about node
when /show\/(.*)/
send res, @nodes.show($1)
# /nodes/fetch/<node> or /nodes/fetch/group/<group>/<node> - returns json formatted configuration file for <node>
when /fetch\/(.*)/
begin
if $1.include? '/'
group, node = $1.split("/")[1..2]
else
group, node = 0, $1
end
json = if node[-4..-1] == '.txt'
node = node[0..-5]
end
send res, json, @nodes.fetch(node, group)
rescue Oxidized::NotSupported => e
send res, e
end
end
end
end
# /nodes/next/node - moves node to head of queue
class Next < WEBrick::HTTPServlet::AbstractServlet
include Oxidized::API::Rest::Helpers
def initialize server, nodes
super server
@nodes = nodes
end
def do_GET req, res
@nodes.next req.path_info[1..-1]
send res
end
def do_PUT req, res
node = req.path_info[1..-1]
begin
opt = JSON.load req.body
Log.debug "before: #{@nodes.list}"
@nodes.next node, opt
Log.debug "after: #{@nodes.list}"
send res
rescue JSON::ParserError
send res, 'broken JSON', 400
end
end
end
end
end
end
|
