fix: Throttler 테스트를 위해 오픈

boostcampwm2023 · Dec 13, 2023 · defee79 · defee79
1 parent d9b645b
commit defee79
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 11 deletions.
diff --git a/back/src/app.module.ts b/back/src/app.module.ts
@@ -6,12 +6,12 @@ import { AuthModule } from './modules/auth/auth.module';
 import typeOrmConfig from './config/orm-config';
 import { JwtModule } from '@nestjs/jwt';
 import { UserModule } from './modules/user/user.module';
-import { APP_INTERCEPTOR } from '@nestjs/core';
+import { APP_GUARD, APP_INTERCEPTOR } from '@nestjs/core';
 import { ControllerLoggerInterceptor } from './common/interceptors/log.interceptor';
 import { ThrottlerModule } from '@nestjs/throttler';
 import throttlerConfig from './config/throttler-config';
 import { MongooseModule } from '@nestjs/mongoose';
-//import { ThrottlerBehindProxyGuard } from './common/guards/throttler.guard';
+import { ThrottlerBehindProxyGuard } from './common/guards/throttler.guard';
 
 @Module({
   imports: [
@@ -27,8 +27,8 @@ import { MongooseModule } from '@nestjs/mongoose';
   controllers: [AppController],
   providers: [
     AppService,
-    { provide: APP_INTERCEPTOR, useClass: ControllerLoggerInterceptor }
-    //{ provide: APP_GUARD, useClass: ThrottlerBehindProxyGuard }
+    { provide: APP_INTERCEPTOR, useClass: ControllerLoggerInterceptor },
+    { provide: APP_GUARD, useClass: ThrottlerBehindProxyGuard }
   ]
 })
 export class AppModule {}
diff --git a/back/src/config/throttler-config.ts b/back/src/config/throttler-config.ts
@@ -4,8 +4,8 @@ const throttlerConfig: ThrottlerModuleOptions = {
   throttlers: [
     {
       name: 'api',
-      ttl: 1000, // 5 seconds
-      limit: 5
+      ttl: 10000, // 1 seconds
+      limit: 3
     }
   ],
   errorMessage: 'Too Many API Requests'

diff --git a/back/src/modules/message/message.controller.ts b/back/src/modules/message/message.controller.ts
@@ -33,8 +33,7 @@ import { JWTRequest } from 'src/common/interface/request.interface';
 import { ClovaService } from './clova.service';
 import { UpdateMessageLocationsDto } from './dto/update-message-locations.dto';
 import { IPLoggerInterceptor } from 'src/common/interceptors/ip-log.interceptor';
-//import { Throttle } from '@nestjs/throttler';
-
+import { Throttle } from '@nestjs/throttler';
 @ApiTags('Message API')
 @Controller('message')
 export class MessageController {
@@ -125,7 +124,7 @@ export class MessageController {
   }
 
   @UseGuards(JWTGuard)
-  //@Throttle({ api: { limit: 10, ttl: 1000 } })
+  @Throttle({ api: { limit: 10, ttl: 1000 } })
   @ApiCookieAuth('access_token')
   @Put('/:message_id/open')
   @HttpCode(200)

diff --git a/back/src/modules/snowball/snowball.controller.ts b/back/src/modules/snowball/snowball.controller.ts
@@ -28,7 +28,7 @@ import { JWTGuard } from '../../common/guards/jwt.guard';
 import { UpdateMainDecoDto } from './dto/update-main-decoration.dto';
 import { JWTRequest } from '../../common/interface/request.interface';
 import { hasJWTInterceptor } from '../../common/interceptors/hasJwt.interceptor';
-//import { Throttle } from '@nestjs/throttler';
+import { Throttle } from '@nestjs/throttler';
 
 @ApiTags('Snowball API')
 @Controller('snowball')
@@ -87,7 +87,7 @@ export class SnowballController {
   }
 
   @Get('/:snowball_id')
-  //@Throttle({ api: { limit: 10, ttl: 1000 } })
+  @Throttle({ api: { limit: 10, ttl: 1000 } })
   @UseInterceptors(hasJWTInterceptor)
   @HttpCode(200)
   @ApiResponse({