forked from CNSRE/ABTestingGateway
-
Notifications
You must be signed in to change notification settings - Fork 10
/
Copy pathpolicy.lua
236 lines (194 loc) · 5.95 KB
/
policy.lua
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
---
-- @classmod abtesting.adapter.policy
-- @release 0.0.1
local modulename = "abtestingAdminPolicy"
local _M = { _VERSION = "0.0.1" }
local mt = { __index = _M }
local ERRORINFO = require('abtesting.error.errcode').info
local runtimeModule = require('abtesting.adapter.runtime')
local policyModule = require('abtesting.adapter.policy')
local redisModule = require('abtesting.utils.redis')
local systemConf = require('abtesting.utils.init')
local handler = require('abtesting.error.handler').handler
local utils = require('abtesting.utils.utils')
local log = require('abtesting.utils.log')
local ERRORINFO = require('abtesting.error.errcode').info
local cjson = require('cjson.safe')
local doresp = utils.doresp
local dolog = utils.dolog
local doerror = utils.doerror
local redisConf = systemConf.redisConf
local divtypes = systemConf.divtypes
local prefixConf = systemConf.prefixConf
local policyLib = prefixConf.policyLibPrefix
local runtimeLib = prefixConf.runtimeInfoPrefix
local domain_name = prefixConf.domainname
local getPolicyId = function()
local policyID = tonumber(ngx.var.arg_policyid)
if not policyID or policyID < 0 then
local info = ERRORINFO.PARAMETER_TYPE_ERROR
local desc = "policyID invalid"
local response = doresp(info, desc)
log:errlog(dolog(info, desc))
ngx.say(response)
return nil
end
return policyID
end
local getPolicy = function()
local request_body = ngx.var.request_body
local postData = cjson.decode(request_body)
if not request_body then
-- ERRORCODE.PARAMETER_NONE
local errinfo = ERRORINFO.PARAMETER_NONE
local desc = 'request_body or post data'
local response = doresp(errinfo, desc)
log:errlog(dolog(errinfo, desc))
ngx.say(response)
return nil
end
if not postData then
-- ERRORCODE.PARAMETER_ERROR
local errinfo = ERRORINFO.PARAMETER_ERROR
local desc = 'postData is not a json string'
local response = doresp(errinfo, desc)
log:errlog(dolog(errinfo, desc))
ngx.say(response)
return nil
end
local divtype = postData.divtype
local divdata = postData.divdata
if not divtype or not divdata then
-- ERRORCODE.PARAMETER_NONE
local errinfo = ERRORINFO.PARAMETER_NONE
local desc = "policy divtype or policy divdata"
local response = doresp(errinfo, desc)
log:errlog(dolog(errinfo, desc))
ngx.say(response)
return nil
end
if not divtypes[divtype] then
-- ERRORCODE.PARAMETER_TYPE_ERROR
local errinfo = ERRORINFO.PARAMETER_TYPE_ERROR
local desc = "unsupported divtype"
local response = doresp(errinfo, desc)
log:errlog(dolog(errinfo, desc))
ngx.say(response)
return nil
end
return postData
end
_M.check = function(option)
local db = option.db
local policy = getPolicy()
if not policy then
return false
end
local pfunc = function()
local policyMod = policyModule:new(db.redis, policyLib)
return policyMod:check(policy)
end
local status, info = xpcall(pfunc, handler)
if not status then
local response = doerror(info)
ngx.say(response)
return false
end
local chkout = info
local valid = chkout[1]
local err = chkout[2]
local desc = chkout[3]
local response
if not valid then
log:errlog(dolog(err, desc))
response = doresp(err, desc)
else
response = doresp(ERRORINFO.SUCCESS)
end
ngx.say(response)
return true
end
_M.set = function(option)
local db = option.db
local policy = getPolicy()
if not policy then
return false
end
local pfunc = function()
policyMod = policyModule:new(db.redis, policyLib)
return policyMod:check(policy)
end
local status, info = xpcall(pfunc, handler)
if not status then
local response = doerror(info)
ngx.say(response)
return false
end
local chkout = info
local valid = chkout[1]
local err = chkout[2]
local desc = chkout[3]
if not valid then
log:errlog(dolog(err, desc))
local response = doresp(err, desc)
ngx.say(response)
return false
end
local pfunc = function() return policyMod:set(policy) end
local status, info = xpcall(pfunc, handler)
if not status then
local response = doerror(info)
ngx.say(response)
return false
end
local data
if info then
data = ' the id of new policy is '..info
end
local response = doresp(ERRORINFO.SUCCESS, data)
ngx.say(response)
return true
end
_M.del = function(option)
local db = option.db
local policyId = getPolicyId()
if not policyId then
return false
end
local pfunc = function()
local policyMod = policyModule:new(db.redis, policyLib)
return policyMod:del(policyId)
end
local status, info = xpcall(pfunc, handler)
if not status then
local response = doerror(info)
ngx.say(response)
return false
end
local response = doresp(ERRORINFO.SUCCESS)
ngx.say(response)
return true
end
_M.get = function(option)
local db = option.db
local policyId = getPolicyId()
if not policyId then
return false
end
local pfunc = function()
local policyIO = policyModule:new(db.redis, policyLib)
return policyIO:get(policyId)
end
local status, info = xpcall(pfunc, handler)
if not status then
local response = doerror(info)
ngx.say(response)
return false
else
local response = doresp(ERRORINFO.SUCCESS, nil, info)
log:errlog(dolog(ERRORINFO.SUCCESS, nil))
ngx.say(response)
return true
end
end
return _M