Merge pull request #101 from boostcampwm-2024/develop

v1.3.0 배포

Author: ezcolin2

.github/workflows/cd-pipeline.yml

@@ -54,6 +54,7 @@ jobs:
           docker build -f ./services/backend/Dockerfile.prod -t growth123/octodocs-backend . &
           docker build -f ./services/nginx/Dockerfile.prod -t growth123/octodocs-nginx . &
           docker build -f ./services/websocket/Dockerfile.prod -t growth123/octodocs-websocket . &
+          docker build -f ./services/scheduler/Dockerfile.prod -t growth123/octodocs-scheduler . &
           wait
 
       # Docker 이미지 푸시
@@ -63,6 +64,7 @@ jobs:
           docker push growth123/octodocs-backend &
           docker push growth123/octodocs-nginx &
           docker push growth123/octodocs-websocket &
+          docker push growth123/octodocs-scheduler &
           wait
 
   deploy:

apps/backend/nest-cli.json

@@ -4,5 +4,6 @@
   "sourceRoot": "src",
   "compilerOptions": {
     "deleteOutDir": true
-  }
-}
+  },
+  "entryFile": "./apps/backend/src/main.js"
+}

apps/backend/package.json

@@ -21,10 +21,9 @@
   },
   "dependencies": {
     "@aws-sdk/client-s3": "^3.693.0",
-    "@huggingface/transformers": "^3.3.3",
-    "@langchain/community": "^0.3.28",
-    "@langchain/core": "^0.3.37",
-    "@langchain/openai": "^0.4.2",
+    "@langchain/community": "^0.3.30",
+    "@langchain/core": "^0.3.40",
+    "@langchain/openai": "^0.4.4",
     "@langchain/textsplitters": "^0.1.0",
     "@nestjs/common": "^10.0.0",
     "@nestjs/config": "^3.3.0",
@@ -36,6 +35,7 @@
     "@nestjs/platform-socket.io": "^10.4.8",
     "@nestjs/platform-ws": "^10.4.7",
     "@nestjs/schedule": "^4.1.1",
+    "@nestjs/schematics": "^11.0.0",
     "@nestjs/serve-static": "^4.0.2",
     "@nestjs/swagger": "^8.0.5",
     "@nestjs/typeorm": "^10.0.2",
@@ -107,6 +107,22 @@
       "**/*.(t|j)s"
     ],
     "coverageDirectory": "../coverage",
-    "testEnvironment": "node"
+    "testEnvironment": "node",
+    "moduleDirectories": [
+      "node_modules",
+      "src"
+    ],
+    "moduleNameMapper": {
+      "@root/(.*)$": "<rootDir>/../../$1",
+      "@app/redis/(.*)$": "<rootDir>/../../../libs/redis/src/$1",
+      "@app/page/(.*)$": "<rootDir>/../../../libs/page/src/$1",
+      "@app/node/(.*)$": "<rootDir>/../../../libs/node/src/$1",
+      "@app/edge/(.*)$": "<rootDir>/../../../libs/edge/src/$1",
+      "@app/workspace/(.*)$": "<rootDir>/../../../libs/workspace/src/$1",
+      "@app/user/(.*)$": "<rootDir>/../../../libs/user/src/$1",
+      "@app/role/(.*)$": "<rootDir>/../../../libs/role/src/$1",
+      "@app/token/(.*)$": "<rootDir>/../../../libs/token/src/$1",
+      "@app/exception/(.*)$": "<rootDir>/../../../libs/exception/$1"
+    }
   }
 }

apps/backend/src/abort/chat-abort.controller.spec.ts

@@ -0,0 +1,49 @@
+import { Test, TestingModule } from '@nestjs/testing';
+import { ChatAbortController } from './chat-abort.controller';
+import { ChatAbortService } from './chat-abort.service';
+import { AbortResponseMessage } from './chat-abort.controller';
+
+describe('AbortController', () => {
+  let abortController: ChatAbortController;
+  let abortService: ChatAbortService;
+
+  beforeEach(async () => {
+    const module: TestingModule = await Test.createTestingModule({
+      controllers: [ChatAbortController],
+      providers: [ChatAbortService],
+    }).compile();
+
+    abortController = module.get<ChatAbortController>(ChatAbortController);
+    abortService = module.get<ChatAbortService>(ChatAbortService);
+  });
+
+  it('should be defined', () => {
+    expect(abortController).toBeDefined();
+  });
+
+  it('요청 중단 시 성공 테스트', () => {
+    const requestId = 'test-request';
+    // AbortService의 abortRequest가 true를 반환하도록 spy 설정
+    jest.spyOn(abortService, 'abortRequest').mockReturnValue(true);
+
+    const result = abortController.abort(requestId);
+
+    expect(result).toEqual({
+      success: true,
+      message: AbortResponseMessage.ABORT_SUCCESS,
+    });
+  });
+
+  it('존재하지 않는 requestId로 요청 중단 시 실패 테스트', () => {
+    const requestId = 'non-existent-request';
+    // AbortService의 abortRequest가 false를 반환하도록 spy 설정
+    jest.spyOn(abortService, 'abortRequest').mockReturnValue(false);
+
+    const result = abortController.abort(requestId);
+
+    expect(result).toEqual({
+      success: false,
+      message: AbortResponseMessage.ABORT_FAIL,
+    });
+  });
+});

apps/backend/src/abort/chat-abort.controller.ts

@@ -0,0 +1,23 @@
+import { Controller, Get, Param } from '@nestjs/common';
+import { ChatAbortService } from './chat-abort.service';
+
+export enum AbortResponseMessage {
+  ABORT_SUCCESS = '웹 요청 중단에 성공했습니다.',
+  ABORT_FAIL = '웹 요청 중단에 실패했습니다.',
+}
+
+@Controller('abort')
+export class ChatAbortController {
+  constructor(private readonly abortService: ChatAbortService) {}
+
+  @Get('/:requestId')
+  abort(@Param('requestId') requestId: string) {
+    const success = this.abortService.abortRequest(requestId);
+    return {
+      success,
+      message: success
+        ? AbortResponseMessage.ABORT_SUCCESS
+        : AbortResponseMessage.ABORT_FAIL,
+    };
+  }
+}

apps/backend/src/abort/chat-abort.module.ts

@@ -0,0 +1,10 @@
+import { Module } from '@nestjs/common';
+import { ChatAbortService } from './chat-abort.service';
+import { ChatAbortController } from './chat-abort.controller';
+
+@Module({
+  providers: [ChatAbortService],
+  controllers: [ChatAbortController],
+  exports: [ChatAbortService],
+})
+export class ChatAbortModule {}

apps/backend/src/abort/chat-abort.service.spec.ts

@@ -0,0 +1,44 @@
+import { Test, TestingModule } from '@nestjs/testing';
+import { ChatAbortService } from './chat-abort.service';
+
+describe('AbortService', () => {
+  let abortService: ChatAbortService;
+
+  beforeEach(async () => {
+    const module: TestingModule = await Test.createTestingModule({
+      providers: [ChatAbortService],
+    }).compile();
+
+    abortService = module.get<ChatAbortService>(ChatAbortService);
+  });
+
+  it('AbortController를 생성 성공 테스트', () => {
+    const requestId = 'test';
+
+    const controller = abortService.createController(requestId);
+
+    expect(controller).toBeDefined(); // AbortController가 생성되었는지 확인
+    expect(controller.signal.aborted).toBe(false); // 아직 중단되지 않았는지 확인
+
+    const storedController = abortService.getController(requestId);
+    expect(storedController).toBe(controller); // 저장된 컨트롤러가 동일한지 확인
+  });
+
+  it('요청을 중단하고 캐시 삭제 성공 테스트', () => {
+    const requestId = 'test';
+    const controller = abortService.createController(requestId);
+
+    expect(controller.signal.aborted).toBe(false); // 초기 상태 확인
+
+    const success = abortService.abortRequest(requestId);
+
+    expect(success).toBe(true); // 요청 중단이 성공했는지 확인
+    expect(controller.signal.aborted).toBe(true); // AbortController가 중단되었는지 확인
+    expect(abortService.getController(requestId)).toBeUndefined(); // 캐시에서 삭제되었는지 확인
+  });
+
+  it('존재하지 않는 requestId로 요청 중단 실패 테스트', () => {
+    const success = abortService.abortRequest('non-existent-id');
+    expect(success).toBe(false); // 존재하지 않는 요청은 중단할 수 없어야 함
+  });
+});

apps/backend/src/abort/chat-abort.service.ts

@@ -0,0 +1,50 @@
+import { Injectable } from '@nestjs/common';
+
+@Injectable()
+export class ChatAbortService {
+  private controllers: Map<
+    string,
+    { controller: AbortController; timestamp: number }
+  > = new Map();
+  private ttl: number = 60000;
+
+  constructor() {
+    setInterval(() => this.cleanup(), 10000);
+  }
+
+  createController(requestId: string): AbortController {
+    const controller = new AbortController();
+    const timestamp = Date.now();
+    this.controllers.set(requestId, { controller, timestamp });
+    return controller;
+  }
+
+  getController(requestId: string): AbortController | undefined {
+    return this.controllers.get(requestId)?.controller;
+  }
+
+  abortRequest(requestId: string): boolean {
+    const entry = this.controllers.get(requestId);
+    if (entry) {
+      if (entry.controller.signal.aborted) {
+        // 이미 취소된 경우, 추가 작업을 하지 않음
+        console.log(`Request with ID ${requestId} is already aborted.`);
+        return false;
+      }
+
+      entry.controller.abort();
+      this.controllers.delete(requestId);
+      return true;
+    }
+    return false;
+  }
+
+  private cleanup(): void {
+    const now = Date.now();
+    this.controllers.forEach((entry, requestId) => {
+      if (now - entry.timestamp > this.ttl) {
+        this.controllers.delete(requestId);
+      }
+    });
+  }
+}

apps/backend/src/abort/dtos/AbortFailResponseDto.ts

@@ -0,0 +1,11 @@
+import { ApiProperty } from '@nestjs/swagger';
+import { IsString } from 'class-validator';
+
+export class AbortFailResponseDto {
+  @ApiProperty({
+    example: '웹 요청 중단에 실패했습니다.',
+    description: '웹 요청 결과 메시지',
+  })
+  @IsString()
+  message: string;
+}

apps/backend/src/abort/dtos/AbortSuccessResponseDto.ts

@@ -0,0 +1,11 @@
+import { ApiProperty } from '@nestjs/swagger';
+import { IsString, IsBoolean } from 'class-validator';
+
+export class AbortSuccessResponseDto {
+  @ApiProperty({
+    example: '웹 요청 중단에 성공했습니다.',
+    description: '웹 요청 결과 메시지',
+  })
+  @IsString()
+  message: string;
+}