SHOOT - API 요청 횟수 제한

민경찬·2023년 2월 11일

SHOOT

백엔드

목록 보기

2/19

서론

프론트엔드에서 무한 스크롤 개발 도중 한 번에 많은 양의 요청을 던졌고 서버에서는 DB 커넥션 수 제한이라는 에러가 발생했다.

한 명의 사용자로부터 너무 반복적인 요청이 오는 것을 어떻게 예외처리 해줘야하는지 알아보자.

본론

1. express-rate-limit 사용

npm i express-rate-limit

express-rate-limit 라이브러리를 다운로드 받아준다.

const rateLimit = require('express-rate-limit');

module.exports = rateLimit({
    windowMs: 60 * 1000,
    max: 3,
    handler(req, res) {
        res.status(423).send({
            message : 'too many request'
        });
    },
})

우선 테스트를 위해 분당 3번의 요청만 할 수 있도록 설정한다.

postman에서 빠르게 세 번의 요청을 넣어주면
잘 나온다.

이제 10초에 10개의 api를 요청할 수 있도록 셋팅하고 마무리!

2. 게시글 업로드 쿨타임 설정

같은 유저가 같은 제목의 게시글을 1분 내로 연속적으로 올릴 수 없도록 하는 코드를 만들어 보려고 한다.

postCoolDown.js

const redis = require('redis').createClient();

module.exports = (req, res, next) => {
    (async () => {
        const title = req.body.title || '';
        const loginUserEmail = req.email || '';
        
        try{
            await redis.connect();

            const existState = await redis.exists(`post_upload_${loginUserEmail}_${title}`);
            if(existState){
                await redis.disconnect();

                res.status(423).send({
                    message : 'too many request'
                });
            }else{
                await redis.set(`post_upload_${loginUserEmail}_${title}`, 1);
                await redis.expire(`post_upload_${loginUserEmail}_${title}`, 60); // 1 minutes

                await redis.disconnect();

                next();
            }
        }catch(err){
            console.log(err);

            if(redis.isOpen){
                await redis.disconnect();
            }
            
            res.status(409).send({
                message : 'unexpected error occured'
            });
        }
    })();
}