экспресс -лимит Redis
import { Request, Response, NextFunction } from 'express'
import IORedis from 'ioredis'
import ip from 'request-ip'
let io = new IORedis({
host: process.env.REDIS_HOST || 'localhost',
port: parseInt(process.env.REDIS_PORT || '')
})
export async function rateLimiterById(req: Request, res: Response, next: NextFunction): Promise<any> {
// store id to redis
await io.set(`redis-id:${req.payload.uid}`, req.payload.uid)
// get request by id
const getId = await io.get(`redis-id:${req.payload.uid}`)
// counter count request
const maxCounterRequest = await io.incrby(`counter-id:${req.payload.uid}`, 1)
if (getId === req.payload.uid && maxCounterRequest <= 50) {
await io.expire(`counter-id:${req.payload.uid}`, 10)
} else {
await io.del(`redis-id:${req.payload.uid}`)
return res.status(429).json({
status: 'ERROR TO MANY REQUEST',
code: 'AX2AC5R',
message: 'cannot access this endpoint, after 10 second is over'
})
}
next()
}
export async function rateLimiterByIp(req: Response, res: Response, next: NextFunction): Promise<any> {
const getIp = ip.getClientIp(req)
// store id to redis
await io.set(`redis-ip:${getIp}`, `${getIp}`)
// get request by id
const getStoreIp = await io.get(`redis-ip:${getIp}`)
// counter count request
const maxCounterRequest = await io.incrby(`counter-ip:${getIp}`, 1)
if (getStoreIp === getIp && maxCounterRequest <= 50) {
await io.expire(`counter-ip:${getIp}`, 10)
} else {
await io.del(`redis-ip:${getIp}`)
return res.status(429).json({
status: 'ERROR TO MANY REQUEST',
code: 'AX2AC5R',
message: 'cannot access this endpoint, after 10 second is over'
})
}
next()
}
Restu Wahyu Saputra