2020-12-30 14:25:14 +08:00

17 lines
396 B
JavaScript

const throttleMap = new Map();
export default async (ctx, next) => {
if (ctx.config.throttle) {
if (throttleMap.get(ctx.key) >= Date.now()) {
ctx.error = {
type: 'FREQUENTLY',
msg: '请求过于频繁'
};
return;
}
}
await next();
throttleMap.set(ctx.key, Date.now() + ctx.config.throttle);
};