[BE] socket 분리 로직 수정

BE/src/stock/index/stock-index.module.ts

@@ -8,6 +8,5 @@ import { SocketModule } from '../../websocket/socket.module';
   imports: [KoreaInvestmentModule, SocketModule],
   controllers: [StockIndexController],
   providers: [StockIndexService],
-  exports: [StockIndexService],
 })
 export class StockIndexModule {}

BE/src/stock/order/stock-order.controller.ts

@@ -22,7 +22,7 @@ import { RequestInterface } from './interface/request.interface';
 @Controller('/api/stocks/trade')
 @ApiTags('주식 매수/매도 API')
 export class StockOrderController {
-  constructor(private readonly stockTradeService: StockOrderService) {}
+  constructor(private readonly stockOrderService: StockOrderService) {}
 
   @Post('/buy')
   @ApiBearerAuth()
@@ -39,7 +39,7 @@ export class StockOrderController {
     @Req() request: RequestInterface,
     @Body(ValidationPipe) stockOrderRequest: StockOrderRequestDto,
   ) {
-    await this.stockTradeService.buy(request.user.id, stockOrderRequest);
+    await this.stockOrderService.buy(request.user.id, stockOrderRequest);
   }
 
   @Post('/sell')
@@ -57,7 +57,7 @@ export class StockOrderController {
     @Req() request: RequestInterface,
     @Body(ValidationPipe) stockOrderRequest: StockOrderRequestDto,
   ) {
-    await this.stockTradeService.sell(request.user.id, stockOrderRequest);
+    await this.stockOrderService.sell(request.user.id, stockOrderRequest);
   }
 
   @Delete('/:order_id')
@@ -75,6 +75,6 @@ export class StockOrderController {
     @Req() request: RequestInterface,
     @Param('order_id') orderId: number,
   ) {
-    await this.stockTradeService.cancel(request.user.id, orderId);
+    await this.stockOrderService.cancel(request.user.id, orderId);
   }
 }

BE/src/websocket/base-socket.service.ts

@@ -1,25 +1,35 @@
 import { WebSocket } from 'ws';
-import { Injectable, OnModuleInit } from '@nestjs/common';
-import { SocketGateway } from './socket.gateway';
+import {
+  Injectable,
+  InternalServerErrorException,
+  OnModuleInit,
+} from '@nestjs/common';
 import { SocketTokenService } from './socket-token.service';
 
 @Injectable()
-export abstract class BaseSocketService implements OnModuleInit {
-  protected socket: WebSocket;
-  protected socketConnectionKey: string;
+export class BaseSocketService implements OnModuleInit {
+  private socket: WebSocket;
+  private socketConnectionKey: string;
+  private socketOpenHandlers: (() => void | Promise<void>)[] = [];
+  private socketDataHandlers: {
+    [key: string]: (data) => void;
+  } = {};
 
-  constructor(
-    protected readonly socketTokenService: SocketTokenService,
-    protected readonly socketGateway: SocketGateway,
-  ) {}
+  constructor(private readonly socketTokenService: SocketTokenService) {}
 
   async onModuleInit() {
     this.socketConnectionKey =
       await this.socketTokenService.getSocketConnectionKey();
     this.socket = new WebSocket(process.env.KOREA_INVESTMENT_SOCKET_URL);
 
     this.socket.onopen = () => {
-      this.handleSocketOpen();
+      Promise.all(
+        this.socketOpenHandlers.map(async (socketOpenHandler) => {
+          await socketOpenHandler();
+        }),
+      ).catch(() => {
+        throw new InternalServerErrorException();
+      });
     };
 
     this.socket.onmessage = (event) => {
@@ -30,14 +40,11 @@ export abstract class BaseSocketService implements OnModuleInit {
       if (data.length < 2) return;
 
       const dataList = data[3].split('^');
-      this.handleSocketData(dataList);
+      this.socketDataHandlers[data[1]](dataList);
     };
   }
 
-  protected abstract handleSocketOpen(): void;
-  protected abstract handleSocketData(dataList): void;
-
-  protected registerCode(trId: string, trKey: string) {
+  registerCode(trId: string, trKey: string) {
     this.socket.send(
       JSON.stringify({
         header: {
@@ -55,4 +62,12 @@ export abstract class BaseSocketService implements OnModuleInit {
       }),
     );
   }
+
+  registerSocketOpenHandler(handler: () => void | Promise<void>) {
+    this.socketOpenHandlers.push(handler);
+  }
+
+  registerSocketDataHandler(tradeCode: string, handler: (data) => void) {
+    this.socketDataHandlers[tradeCode] = handler;
+  }
 }

BE/src/websocket/socket.module.ts

@@ -2,11 +2,17 @@ import { Module } from '@nestjs/common';
 import { StockIndexSocketService } from './stock-index-socket.service';
 import { SocketGateway } from './socket.gateway';
 import { SocketTokenService } from './socket-token.service';
+import { BaseSocketService } from './base-socket.service';
 
 @Module({
   imports: [],
   controllers: [],
-  providers: [SocketTokenService, StockIndexSocketService, SocketGateway],
+  providers: [
+    SocketTokenService,
+    StockIndexSocketService,
+    SocketGateway,
+    BaseSocketService,
+  ],
   exports: [SocketGateway],
 })
 export class SocketModule {}

BE/src/websocket/stock-index-socket.service.ts

@@ -1,32 +1,42 @@
 import { Injectable } from '@nestjs/common';
 import { StockIndexValueElementDto } from '../stock/index/dto/stock-index-value-element.dto';
 import { BaseSocketService } from './base-socket.service';
+import { SocketGateway } from './socket.gateway';
 
 @Injectable()
-export class StockIndexSocketService extends BaseSocketService {
+export class StockIndexSocketService {
+  private TR_ID = 'H0UPCNT0';
   private STOCK_CODE = {
     '0001': 'KOSPI',
     '1001': 'KOSDAQ',
     '2001': 'KOSPI200',
     '3003': 'KSQ150',
   };
 
-  protected handleSocketOpen() {
-    this.registerCode('H0UPCNT0', '0001'); // 코스피
-    this.registerCode('H0UPCNT0', '1001'); // 코스닥
-    this.registerCode('H0UPCNT0', '2001'); // 코스피200
-    this.registerCode('H0UPCNT0', '3003'); // KSQ150
-  }
+  constructor(
+    private readonly socketGateway: SocketGateway,
+    private readonly baseSocketService: BaseSocketService,
+  ) {
+    baseSocketService.registerSocketOpenHandler(() => {
+      this.baseSocketService.registerCode(this.TR_ID, '0001'); // 코스피
+      this.baseSocketService.registerCode(this.TR_ID, '1001'); // 코스닥
+      this.baseSocketService.registerCode(this.TR_ID, '2001'); // 코스피200
+      this.baseSocketService.registerCode(this.TR_ID, '3003'); // KSQ150
+    });
 
-  protected handleSocketData(dataList: string[]) {
-    this.socketGateway.sendStockIndexValueToClient(
-      this.STOCK_CODE[dataList[0]],
-      new StockIndexValueElementDto(
-        dataList[2],
-        dataList[4],
-        dataList[9],
-        dataList[3],
-      ),
+    baseSocketService.registerSocketDataHandler(
+      this.TR_ID,
+      (data: string[]) => {
+        this.socketGateway.sendStockIndexValueToClient(
+          this.STOCK_CODE[data[0]],
+          new StockIndexValueElementDto(
+            data[2], // 주가 지수
+            data[4], // 전일 대비 등락
+            data[9], // 전일 대비 등락률
+            data[3], // 부호
+          ),
+        );
+      },
     );
   }
 }