🔧 fix: 프로그램 시작 시 종목 중복 구독 안되도록 로직 수정 #53

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

@@ -11,6 +11,12 @@ export class StockOrderRepository extends Repository<Order> {
     super(Order, dataSource.createEntityManager());
   }
 
+  async findAllCodeByStatus() {
+    return this.createQueryBuilder('orders')
+      .select('DISTINCT orders.stock_code')
+      .getRawMany();
+  }
+
   async updateOrderAndAssetWhenBuy(order, realPrice) {
     const queryRunner = this.dataSource.createQueryRunner();
     await queryRunner.startTransaction();

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

@@ -24,10 +24,8 @@ export class StockOrderService {
 
   private readonly logger = new Logger();
 
-  async findAllPendingOrder() {
-    return this.stockOrderRepository.findBy({
-      status: StatusType.PENDING,
-    });
+  async findAllPendingOrderCode() {
+    return this.stockOrderRepository.findAllCodeByStatus();
   }
 
   async buy(userId: number, stockOrderRequest: StockOrderRequestDto) {

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

@@ -7,6 +7,7 @@ import {
 import { BaseSocketService } from '../../base-socket.service';
 import { SocketGateway } from '../../socket.gateway';
 import { StockOrderService } from '../../../stock/order/stock-order.service';
+import { Order } from '../../../stock/order/stock-order.entity';
 
 @Injectable()
 export class StockPriceSocketService {
@@ -19,7 +20,7 @@ export class StockPriceSocketService {
     private readonly stockOrderService: StockOrderService,
   ) {
     baseSocketService.registerSocketOpenHandler(async () => {
-      const orders = await stockOrderService.findAllPendingOrder();
+      const orders: Order[] = await stockOrderService.findAllPendingOrderCode();
       orders.forEach((order) => {
         baseSocketService.registerCode(this.TR_ID, order.stock_code);
       });