upstream.lua
2.32 KB
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
94
local upstream = require "ngx.upstream"
local json=require "cjson"
local get_servers = upstream.get_servers
local get_upstreams = upstream.get_upstreams
local cache=ngx.shared.upstream
-- get all peers for upstram: u
function list(u)
local d={}
d["name"]=u
d["value"]={}
-- get primary peers
local peers,err = upstream.get_primary_peers(u)
if err then
ngx.say("failed to get primary servers in upstream ", u)
return
end
for _,p in ipairs(peers) do
local s={}
s["id"]=p.id
s["down"]=p.down and p.down or false
s["name"]=p.name
s["backup"]=false
table.insert(d["value"],s)
end
-- get backup peers
peers,err = upstream.get_backup_peers(u)
if err then
ngx.say("failed to get backup servers in upstream ", u)
return
end
for _,p in ipairs(peers) do
local s={}
s["id"]=p.id
s["down"]=p.down and p.down or false
s["name"]=p.name
s["backup"]=true
table.insert(d["value"],s)
end
ngx.header["Content-type"]="application/json;charset=utf-8"
ngx.say(json.encode(d))
end
function upordown(upstream_name,is_backup,peer_id,down_value)
local t={}
t["upstream"]=upstream_name
t["backup"]=is_backup
t["id"]=peer_id
t["value"]=down_value
local rKey=upstream_name .. ":" .. tostring(peer_id) .. ":" .. tostring(is_backup)
local key="d:" .. rKey
local vKey="v:" .. rKey
cache:add(vKey,0)
local v,err=cache:incr(vKey,1)
if not v then
return false
end
local suc=cache:set(key,json.encode(t))
return suc
end
local args=ngx.req.get_uri_args()
local method=args["method"]
-- make sure upstream exist
local upstream = args["upstream"]
if upstream == nil or upstream == '' then
ngx.exit(400)
ngx.log(ngx.ERR, "request params is error. upstream is null")
end
if method == "list" then
list(upstream)
elseif(method=="down" or method=="up") then
local backup=args["backup"]=="true" and true or false
local id=tonumber(args["id"])
local down= method=="down" and true or false
local t={}
ngx.header["Content-type"]="application/json;charset=utf-8"
if not id then
ngx.exit(400)
ngx.log(ngx.ERR, "request params is error. upstream or id is null")
else
local suc=upordown(upstream,backup,id,down)
t["suc"]=suc
ngx.say(json.encode(t))
end
end