...
|
...
|
@@ -4,9 +4,11 @@ |
|
|
const cache = global.yoho.cache.master;
|
|
|
const _ = require('lodash');
|
|
|
const logger = global.yoho.logger;
|
|
|
const config = global.yoho.config;
|
|
|
const helpers = global.yoho.helpers;
|
|
|
|
|
|
let ONE_DAY = 60 * 60 * 24;
|
|
|
const MAX_QPS = config.maxQps;
|
|
|
|
|
|
let pages = {
|
|
|
'/product/\\/pro_([\\d]+)_([\\d]+)\\/(.*)/': 5,
|
...
|
...
|
@@ -35,8 +37,6 @@ module.exports = (req, res, next) => { |
|
|
if (remoteIp && !_.get(req.app.locals, 'pc.sys.noLimiter')) { // 判断获取remoteIp成功,并且开关未关闭
|
|
|
let key = `pc:limiter:${remoteIp}`;
|
|
|
|
|
|
logger.debug(`request limiter key=${key}`);
|
|
|
|
|
|
res.on('render', function() {
|
|
|
let route = req.route ? req.route.path : '';
|
|
|
let appPath = req.app.mountpath;
|
...
|
...
|
@@ -53,12 +53,11 @@ module.exports = (req, res, next) => { |
|
|
}
|
|
|
});
|
|
|
|
|
|
function limiterPage() {
|
|
|
const limiterPage = () => {
|
|
|
let refer = req.method === 'GET' ? req.get('Referer') : '';
|
|
|
let limitAPI = helpers.urlFormat('/3party/check', {refer: refer});
|
|
|
let limitPage = helpers.urlFormat('/3party/check', {refer: req.protocol + '://' + req.get('host') + req.originalUrl});
|
|
|
|
|
|
|
|
|
if (_.indexOf(['/3party/check', '/passport/imagesNode', '/passport/cert/headerTip'], req.path) >= 0) {
|
|
|
return next();
|
|
|
}
|
...
|
...
|
@@ -71,24 +70,23 @@ module.exports = (req, res, next) => { |
|
|
}
|
|
|
|
|
|
return res.redirect(limitPage);
|
|
|
}
|
|
|
};
|
|
|
|
|
|
cache.getAsync(key).then(result => {
|
|
|
cache.getAsync(key).then((result) => {
|
|
|
if (result && _.isNumber(result)) {
|
|
|
if (result > 1800) { // 判断 qps
|
|
|
if (result > MAX_QPS) { // 判断 qps
|
|
|
cache.touch(key, ONE_DAY);
|
|
|
console.log('req limit', key);
|
|
|
return limiterPage();
|
|
|
} else {
|
|
|
cache.incrAsync(key, 1); // qps + 1
|
|
|
return next();
|
|
|
}
|
|
|
} else {
|
|
|
cache.setAsync(key, 1, 60); // 设置key,1s失效
|
|
|
cache.setAsync(key, 1, 60); // 设置key,1m失效
|
|
|
return next();
|
|
|
}
|
|
|
}).catch(e => {
|
|
|
logger.error(`request limiter get key[${key}] from cache error.`, e);
|
|
|
}).catch((err) => {
|
|
|
logger.error(`request limiter get key[${key}] from cache error.`, err);
|
|
|
return next();
|
|
|
});
|
|
|
} else {
|
...
|
...
|
|