[BE] 3.02 주식차트 정보 기능 구현 수정 #6

BE/src/koreaInvestment/interface/korea.investment.interface.ts

@@ -0,0 +1,6 @@
+export interface AccessTokenInterface {
+  access_token: string;
+  access_token_token_expired: string;
+  token_type: string;
+  expires_in: number;
+}

BE/src/koreaInvestment/korea.investment.service.ts

@@ -1,4 +1,7 @@
 import axios from 'axios';
+import { UnauthorizedException } from '@nestjs/common';
+import { getFullURL } from '../util/getFullURL';
+import { AccessTokenInterface } from './interface/korea.investment.interface';
 
 export class KoreaInvestmentService {
   private accessToken: string;
@@ -9,19 +12,20 @@ export class KoreaInvestmentService {
     if (this.accessToken && this.tokenExpireTime > new Date()) {
       return this.accessToken;
     }
-    const response = await axios.post(
-      `${process.env.KOREA_INVESTMENT_BASE_URL}/oauth2/tokenP`,
-      {
+    const response = await axios
+      .post<AccessTokenInterface>(getFullURL('/oauth2/tokenP'), {
         grant_type: 'client_credentials',
         appkey: process.env.KOREA_INVESTMENT_APP_KEY,
         appsecret: process.env.KOREA_INVESTMENT_APP_SECRET,
-      },
-    );
+      })
+      .catch(() => {
+        throw new UnauthorizedException('액세스 토큰을 조회하지 못했습니다.');
+      });
 
     const { data } = response;
 
     this.accessToken = data.access_token;
-    this.tokenExpireTime = new Date(Date.now() + +data.expires_in);
+    this.tokenExpireTime = new Date(data.access_token_token_expired);
 
     return this.accessToken;
   }

BE/src/stock/index/dto/stock.index.response.element.dto.ts

@@ -1,16 +1,14 @@
 import { ApiProperty } from '@nestjs/swagger';
 import { StockIndexValueElementDto } from './stock.index.value.element.dto';
-import { StockIndexListElementDto } from './stock.index.list.element.dto';
+import { StockIndexListChartElementDto } from './stock.index.list.chart.element.dto';
 
 export class StockIndexResponseElementDto {
-  @ApiProperty({
-    description: '코스피: 0001, 코스닥: 1001, 코스피200: 2001, KSQ150: 3003',
-  })
-  code: string;
-
   @ApiProperty({ description: '실시간 값', type: StockIndexValueElementDto })
   value: StockIndexValueElementDto;
 
-  @ApiProperty({ description: '실시간 차트', type: StockIndexListElementDto })
-  chart: StockIndexListElementDto;
+  @ApiProperty({
+    description: '실시간 차트',
+    type: [StockIndexListChartElementDto],
+  })
+  chart: StockIndexListChartElementDto[];
 }

BE/src/stock/index/dto/stock.index.value.element.dto.ts

@@ -1,33 +1,21 @@
 import { ApiProperty } from '@nestjs/swagger';
 
 export class StockIndexValueElementDto {
-  constructor(
-    code: string,
-    value: string,
-    diff: string,
-    diffRate: string,
-    sign: string,
-  ) {
-    this.code = code;
-    this.value = value;
+  constructor(value: string, diff: string, diffRate: string, sign: string) {
+    this.curr_value = value;
     this.diff = diff;
-    this.diffRate = diffRate;
+    this.diff_rate = diffRate;
     this.sign = sign;
   }
 
-  @ApiProperty({
-    description: '코스피: 0001, 코스닥: 1001, 코스피200: 2001, KSQ150: 3003',
-  })
-  code: string;
-
   @ApiProperty({ description: '주가 지수' })
-  value: string;
+  curr_value: string;
 
   @ApiProperty({ description: '전일 대비 등락' })
   diff: string;
 
   @ApiProperty({ description: '전일 대비 등락률' })
-  diffRate: string;
+  diff_rate: string;
 
   @ApiProperty({ description: '부호... 인데 추후에 알아봐야 함' })
   sign: string;

BE/src/stock/index/stock.index.controller.ts

@@ -70,22 +70,18 @@ export class StockIndexController {
 
     const stockIndexResponse = new StockIndexResponseDto();
     stockIndexResponse.KOSPI = {
-      code: '0001',
       value: kospiValue,
       chart: kospiChart,
     };
     stockIndexResponse.KOSDAQ = {
-      code: '1001',
       value: kosdaqValue,
       chart: kosdaqChart,
     };
     stockIndexResponse.KOSPI200 = {
-      code: '2001',
       value: kospi200Value,
       chart: kospi200Chart,
     };
     stockIndexResponse.KSQ150 = {
-      code: '3003',
       value: ksq150Value,
       chart: ksq150Chart,
     };
@@ -115,6 +111,11 @@ export class StockIndexController {
       ), // KSQ150
     ]);
 
-    this.socketGateway.sendStockIndexListToClient(stockLists);
+    this.socketGateway.sendStockIndexListToClient({
+      KOSPI: stockLists[0],
+      KOSDAQ: stockLists[1],
+      KOSPI200: stockLists[2],
+      KSQ150: stockLists[3],
+    });
   }
 }

BE/src/stock/index/stock.index.service.ts

@@ -1,12 +1,13 @@
-import { Injectable } from '@nestjs/common';
+import { Injectable, InternalServerErrorException } from '@nestjs/common';
 import axios from 'axios';
 import { StockIndexListChartElementDto } from './dto/stock.index.list.chart.element.dto';
-import { StockIndexListElementDto } from './dto/stock.index.list.element.dto';
 import { StockIndexValueElementDto } from './dto/stock.index.value.element.dto';
 import {
   StockIndexChartInterface,
   StockIndexValueInterface,
 } from './interface/stock.index.interface';
+import { getFullURL } from '../../util/getFullURL';
+import { getHeader } from '../../util/getHeader';
 
 @Injectable()
 export class StockIndexService {
@@ -16,18 +17,12 @@ export class StockIndexService {
       accessToken,
     );
 
-    if (result.rt_cd !== '0')
-      throw new Error('데이터를 정상적으로 조회하지 못했습니다.');
-
-    return new StockIndexListElementDto(
-      code,
-      result.output.map((element) => {
-        return new StockIndexListChartElementDto(
-          element.bsop_hour,
-          element.bstp_nmix_prpr,
-        );
-      }),
-    );
+    return result.output.map((element) => {
+      return new StockIndexListChartElementDto(
+        element.bsop_hour,
+        element.bstp_nmix_prpr,
+      );
+    });
   }
 
   async getDomesticStockIndexValueByCode(code: string, accessToken: string) {
@@ -36,40 +31,39 @@ export class StockIndexService {
       accessToken,
     );
 
-    if (result.rt_cd !== '0')
-      throw new Error('데이터를 정상적으로 조회하지 못했습니다.');
+    const data = result.output;
 
     return new StockIndexValueElementDto(
-      code,
-      result.output.bstp_nmix_prpr,
-      result.output.bstp_nmix_prdy_vrss,
-      result.output.bstp_nmix_prdy_ctrt,
-      result.output.prdy_vrss_sign,
+      data.bstp_nmix_prpr,
+      data.bstp_nmix_prdy_vrss,
+      data.bstp_nmix_prdy_ctrt,
+      data.prdy_vrss_sign,
     );
   }
 
   private async requestDomesticStockIndexListApi(
     code: string,
     accessToken: string,
   ) {
-    const response = await axios.get<StockIndexChartInterface>(
-      `${process.env.KOREA_INVESTMENT_BASE_URL}/uapi/domestic-stock/v1/quotations/inquire-index-timeprice`,
-      {
-        headers: {
-          'content-type': 'application/json; charset=utf-8',
-          authorization: `Bearer ${accessToken}`,
-          appkey: process.env.KOREA_INVESTMENT_APP_KEY,
-          appsecret: process.env.KOREA_INVESTMENT_APP_SECRET,
-          tr_id: 'FHPUP02110200',
-          custtype: 'P',
-        },
-        params: {
-          fid_input_hour_1: 300,
-          fid_cond_mrkt_div_code: 'U',
-          fid_input_iscd: code,
+    const response = await axios
+      .get<StockIndexChartInterface>(
+        getFullURL(
+          '/uapi/domestic-stock/v1/quotations/inquire-index-timeprice',
+        ),
+        {
+          headers: getHeader(accessToken, 'FHPUP02110200'),
+          params: {
+            fid_input_hour_1: 300,
+            fid_cond_mrkt_div_code: 'U',
+            fid_input_iscd: code,
+          },
         },
-      },
-    );
+      )
+      .catch(() => {
+        throw new InternalServerErrorException(
+          '주가 지수 차트 정보를 조회하지 못했습니다.',
+        );
+      });
 
     return response.data;
   }
@@ -78,23 +72,22 @@ export class StockIndexService {
     code: string,
     accessToken: string,
   ) {
-    const response = await axios.get<StockIndexValueInterface>(
-      `${process.env.KOREA_INVESTMENT_BASE_URL}/uapi/domestic-stock/v1/quotations/inquire-index-price`,
-      {
-        headers: {
-          'content-type': 'application/json; charset=utf-8',
-          authorization: `Bearer ${accessToken}`,
-          appkey: process.env.KOREA_INVESTMENT_APP_KEY,
-          appsecret: process.env.KOREA_INVESTMENT_APP_SECRET,
-          tr_id: 'FHPUP02100000',
-          custtype: 'P',
-        },
-        params: {
-          fid_cond_mrkt_div_code: 'U',
-          fid_input_iscd: code,
+    const response = await axios
+      .get<StockIndexValueInterface>(
+        getFullURL('/uapi/domestic-stock/v1/quotations/inquire-index-price'),
+        {
+          headers: getHeader(accessToken, 'FHPUP02100000'),
+          params: {
+            fid_cond_mrkt_div_code: 'U',
+            fid_input_iscd: code,
+          },
         },
-      },
-    );
+      )
+      .catch(() => {
+        throw new InternalServerErrorException(
+          '주가 지수 값 정보를 조회하지 못했습니다.',
+        );
+      });
 
     return response.data;
   }

BE/src/websocket/socket.gateway.ts

@@ -6,11 +6,11 @@ export class SocketGateway {
   @WebSocketServer()
   private server: Server;
 
-  sendStockIndexListToClient(stockIndex) {
-    this.server.emit('index', stockIndex);
+  sendStockIndexListToClient(stockChart) {
+    this.server.emit('chart', stockChart);
   }
 
-  sendStockIndexValueToClient(stockIndexValue) {
-    this.server.emit('indexValue', stockIndexValue);
+  sendStockIndexValueToClient(event, stockIndexValue) {
+    this.server.emit(event, stockIndexValue);
   }
 }

BE/src/websocket/socket.service.ts

@@ -4,6 +4,7 @@ import axios from 'axios';
 import { SocketGateway } from './socket.gateway';
 import { StockIndexValueElementDto } from '../stock/index/dto/stock.index.value.element.dto';
 import { SocketConnectTokenInterface } from './interface/socket.interface';
+import { getFullURL } from '../util/getFullURL';
 
 @Injectable()
 export class SocketService implements OnModuleInit {
@@ -12,13 +13,19 @@ export class SocketService implements OnModuleInit {
     H0UPCNT0: this.handleStockIndexValue.bind(this),
   };
 
+  private STOCK_CODE = {
+    '0001': 'KOSPI',
+    '1001': 'KOSDAQ',
+    '2001': 'KOSPI200',
+    '3003': 'KSQ150',
+  };
+
   constructor(private readonly socketGateway: SocketGateway) {}
 
   async onModuleInit() {
     const socketConnectionKey = await this.getSocketConnectionKey();
 
-    const url = 'ws://ops.koreainvestment.com:21000';
-    this.socket = new WebSocket(url);
+    this.socket = new WebSocket(process.env.KOREA_INVESTMENT_SOCKET_URL);
 
     this.socket.onopen = () => {
       this.registerStockIndexByCode('0001', socketConnectionKey); // 코스피
@@ -41,8 +48,8 @@ export class SocketService implements OnModuleInit {
   private handleStockIndexValue(responseData: string) {
     const responseList = responseData.split('^');
     this.socketGateway.sendStockIndexValueToClient(
+      this.STOCK_CODE[responseList[0]],
       new StockIndexValueElementDto(
-        responseList[0],
         responseList[2],
         responseList[4],
         responseList[9],
@@ -53,7 +60,7 @@ export class SocketService implements OnModuleInit {
 
   private async getSocketConnectionKey() {
     const response = await axios.post<SocketConnectTokenInterface>(
-      `${process.env.KOREA_INVESTMENT_BASE_URL}/oauth2/Approval`,
+      getFullURL('/oauth2/Approval'),
       {
         grant_type: 'client_credentials',
         appkey: process.env.KOREA_INVESTMENT_APP_KEY,