Merge pull request #3 from GNU-connect/refactor-project

fix: WarmupService 로직 수정

Author: JangDongHo

src/modules/app/app.module.ts

@@ -13,14 +13,15 @@ import { ReadingRoomsModule } from '../reading-rooms/reading-rooms.module';
 import { ScheduleModule } from '@nestjs/schedule';
 import { HttpModule } from '@nestjs/axios';
 import { AuthModule } from 'src/modules/auth/auth.module';
-
+import { DiscoveryModule } from '@nestjs/core';
 @Module({
   imports: [
     SentryModule.forRoot(),
     ScheduleModule.forRoot(),
     HttpModule,
     SupabaseModule,
     UtilsModule,
+    DiscoveryModule,
     CommonModule,
     UsersModule,
     ReadingRoomsModule,

src/modules/app/app.service.ts

@@ -1,23 +1,61 @@
 import { Injectable, OnModuleInit } from '@nestjs/common';
 import { HttpService } from '@nestjs/axios';
 import { firstValueFrom } from 'rxjs';
+import { DiscoveryService } from '@nestjs/core';
 import { Cron } from '@nestjs/schedule';
+import { Reflector } from '@nestjs/core';
 
 @Injectable()
 export class AppService implements OnModuleInit {
-  constructor(private readonly httpService: HttpService) {}
+  constructor(
+    private readonly httpService: HttpService,
+    private readonly discoverService: DiscoveryService,
+    private readonly reflector: Reflector,
+  ) {}
 
-  onModuleInit() {
+  public onModuleInit() {
     if (process.env.NODE_ENV === 'production') {
       this.warmUpServer();
     }
   }
 
+  // Cron을 이용한 주기적인 warm-up 호출
+  @Cron('*/10 * * * *')
+  public handleCron() {
+    if (process.env.NODE_ENV === 'production') {
+      console.log('Triggering server warm-up...');
+      this.warmUpServer(); // warm-up 메서드를 호출
+    }
+  }
+
   // 서버를 Warm-up하기 위한 메서드
-  async warmUpServer() {
+  private async warmUpServer() {
     console.log('Running warm-up tasks...');
 
-    const requestBody = {
+    const controllers = this.discoverService.getControllers();
+
+    for (const controller of controllers) {
+      const instance = controller.instance;
+      const prototype = Object.getPrototypeOf(instance);
+
+      // 컨트롤러의 모든 메소드를 순회
+      const methods = Object.getOwnPropertyNames(prototype).filter(
+        (method) => method !== 'constructor',
+      );
+
+      for (const method of methods) {
+        const path = this.reflector.get('path', instance[method]);
+        const httpMethod = this.reflector.get('method', instance[method]);
+
+        if (path && httpMethod) {
+          await this.callApi(path, this.getDefaultRequestBody());
+        }
+      }
+    }
+  }
+
+  private getDefaultRequestBody() {
+    return {
       userRequest: {
         user: {
           id: '74f7e7ab2bf19e63bb1ec845b760631259d7615440a2d3db7b344ac48ed1bbcde5',
@@ -27,42 +65,19 @@ export class AppService implements OnModuleInit {
         clientExtra: {
           sys_campus_id: '1',
         },
-        params: {},
-        detailParams: {},
       },
     };
-
-    try {
-      await this.callApi('/api/node/user/get/campus', requestBody);
-      await this.callApi('/api/node/user/get/profile', requestBody);
-      await this.callApi('/api/node/clicker/get/campus', requestBody);
-
-      console.log('All API calls are completed for warm-up.');
-    } catch (error) {
-      console.error('Error occurred during warm-up API calls:', error);
-    }
   }
 
   // HTTP 요청을 보낼 함수
   private async callApi(apiUrl: string, body: any) {
-    const baseUrl = 'https://connectgnu.kro.kr';
+    const baseUrl = 'https://connectgnu.kro.kr/';
 
     try {
-      const response = await firstValueFrom(
-        this.httpService.post(`${baseUrl}${apiUrl}`, body),
-      );
+      await firstValueFrom(this.httpService.post(`${baseUrl}${apiUrl}`, body));
       console.log(`${apiUrl} data fetched successfully`);
     } catch (error) {
       console.error(`${apiUrl} failed:`, error);
     }
   }
-
-  // Cron을 이용한 주기적인 warm-up 호출
-  @Cron('*/10 * * * *')
-  handleCron() {
-    if (process.env.NODE_ENV === 'production') {
-      console.log('Triggering server warm-up...');
-      this.warmUpServer(); // warm-up 메서드를 호출
-    }
-  }
 }