104 lines
2.8 KiB
JavaScript
104 lines
2.8 KiB
JavaScript
import s from "apigo.cc/gojs/service"
|
|
import http from "apigo.cc/gojs/http"
|
|
import u from "apigo.cc/gojs/util"
|
|
import co from "apigo.cc/gojs/console"
|
|
|
|
let h2c = http
|
|
let urlPrefix
|
|
function main() {
|
|
s.config({
|
|
app: 'user',
|
|
accessTokens: {
|
|
'testToken1122': 2
|
|
},
|
|
calls: {
|
|
'user': 'testToken1122'
|
|
},
|
|
cpuMonitor: true,
|
|
memoryMonitor: true,
|
|
sessionKey: 'SessionID',
|
|
userIdKey: 'id',
|
|
limitedMessage: { code: 429, message: "访问过于频繁" },
|
|
authFieldMessage: { code: 403, message: "身份验证失败 [{{USER_AUTHLEVEL}}/{{TARGET_AUTHLEVEL}}]" },
|
|
verifyFieldMessage: { code: 400, message: "参数 [{{FAILED_FIELDS}} 验证失败" },
|
|
limiters: {
|
|
ip1s: {
|
|
from: 'ip',
|
|
time: 100,
|
|
times: 10
|
|
}
|
|
},
|
|
static: {
|
|
'/': 'api/',
|
|
},
|
|
rewrite: {
|
|
'/echo2.js': '/echo.js'
|
|
}
|
|
})
|
|
s.register({ path: '/echo', noLog200: true }, ({ args, response }) => {
|
|
// setTimeout(() => {
|
|
// response.end(args.name)
|
|
// }, 1)
|
|
u.sleep(1)
|
|
return args.name
|
|
})
|
|
s.load('api/echo.js', { min: 20, max: 1000, idle: 100 })
|
|
s.load('api/user.js')
|
|
let host = s.start()
|
|
h2c = http.newH2C({
|
|
baseURL: 'http://' + host
|
|
})
|
|
return host
|
|
}
|
|
|
|
function test(name) {
|
|
let r = h2c.get('/echo?name=' + name)
|
|
return r.string()
|
|
}
|
|
|
|
function testStatic() {
|
|
let r = h2c.get('/echo2.js')
|
|
if (r.string().indexOf('/echo2') === -1) return r.string()
|
|
return true
|
|
}
|
|
|
|
function test2(name) {
|
|
let r = h2c.get('/echo2?name=' + name)
|
|
return r.string()
|
|
}
|
|
|
|
function testUser() {
|
|
let r = h2c.get('/userInfo')
|
|
if (r.statusCode !== 403 || r.object().code !== 403) return r
|
|
r = h2c.post('/login', { id: 'a1', name: 'Tom' })
|
|
if (r.statusCode !== 400 || r.object().code !== 400) return r
|
|
r = h2c.post('/login', { id: 123, name: 'Tom\t' })
|
|
if (r.statusCode !== 400 || r.object().code !== 400) return r
|
|
|
|
r = h2c.post('/login', { id: 123, name: 'Tom' }).object()
|
|
if (r.code === 1) {
|
|
// 测试限流器允许的 10 次请求
|
|
for (let i = 0; i < 10; i++) {
|
|
r = h2c.get('/userInfo').object()
|
|
if (r.data.id !== 123 | r.data.name !== 'Tom') {
|
|
return r.data
|
|
}
|
|
}
|
|
// 测试限流器拒绝的 1 次请求
|
|
r = h2c.get('/userInfo')
|
|
if (r.statusCode != 429) {
|
|
return r
|
|
}
|
|
u.sleep(100)
|
|
|
|
// 测试限流器过期后允许的 1 次请求
|
|
r = h2c.get('/userInfo').object()
|
|
if (r.data.id !== 123 | r.data.name !== 'Tom') {
|
|
return r.data
|
|
}
|
|
|
|
return true
|
|
}
|
|
return r
|
|
}
|