[BE] 6.02 각 주식 정보 가져오기 API 구현 #54

BE/src/stock/detail/stock-detail.entity.ts

@@ -0,0 +1,13 @@
+import { BaseEntity, Column, Entity, PrimaryColumn } from 'typeorm';
+
+@Entity()
+export class Stocks extends BaseEntity {
+  @PrimaryColumn()
+  code: string;
+
+  @Column()
+  name: string;
+
+  @Column()
+  market: string;
+}

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

@@ -1,11 +1,14 @@
 import { Module } from '@nestjs/common';
+import { TypeOrmModule } from '@nestjs/typeorm';
 import { KoreaInvestmentModule } from '../../koreaInvestment/korea-investment.module';
 import { StockDetailController } from './stock-detail.controller';
 import { StockDetailService } from './stock-detail.service';
+import { StockDetailRepository } from './stock-detail.repository';
+import { Stocks } from './stock-detail.entity';
 
 @Module({
-  imports: [KoreaInvestmentModule],
+  imports: [KoreaInvestmentModule, TypeOrmModule.forFeature([Stocks])],
   controllers: [StockDetailController],
-  providers: [StockDetailService],
+  providers: [StockDetailService, StockDetailRepository],
 })
 export class StockDetailModule {}

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

@@ -0,0 +1,15 @@
+import { InjectDataSource } from '@nestjs/typeorm';
+import { Injectable } from '@nestjs/common';
+import { DataSource, Repository } from 'typeorm';
+import { Stocks } from './stock-detail.entity';
+
+@Injectable()
+export class StockDetailRepository extends Repository<Stocks> {
+  constructor(@InjectDataSource() dataSource: DataSource) {
+    super(Stocks, dataSource.createEntityManager());
+  }
+
+  async findOneByCode(code: string) {
+    return this.findOne({ where: { code } });
+  }
+}

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

@@ -10,12 +10,16 @@ import {
   InquirePriceOutputData,
 } from './interface/stock-detail.interface';
 import { InquirePriceResponseDto } from './dto/stock-detail-response.dto';
+import { StockDetailRepository } from './stock-detail.repository';
 
 @Injectable()
 export class StockDetailService {
   private readonly logger = new Logger();
 
-  constructor(private readonly koreaInvetmentService: KoreaInvestmentService) {}
+  constructor(
+    private readonly koreaInvetmentService: KoreaInvestmentService,
+    private readonly stockDetailRepository: StockDetailRepository,
+  ) {}
 
   /**
    * 주식현재가 시세 데이터를 반환하는 함수
@@ -57,16 +61,23 @@ export class StockDetailService {
    *
    * @author uuuo3o
    */
-  private formatStockData(stock: InquirePriceOutputData) {
-    const stockData = new InquirePriceResponseDto();
-    stockData.stck_shrn_iscd = stock.stck_shrn_iscd;
-    stockData.stck_prpr = stock.stck_prpr;
-    stockData.prdy_vrss = stock.prdy_vrss;
-    stockData.prdy_vrss_sign = stock.prdy_vrss_sign;
-    stockData.prdy_ctrt = stock.prdy_ctrt;
-    stockData.hts_avls = stock.hts_avls;
-    stockData.per = stock.per;
-    return stockData;
+  private async formatStockData(
+    stock: InquirePriceOutputData,
+  ): Promise<InquirePriceResponseDto> {
+    const { name } = await this.stockDetailRepository.findOneByCode(
+      stock.stck_shrn_iscd,
+    );
+
+    return {
+      hts_kor_isnm: name,
+      stck_shrn_iscd: stock.stck_shrn_iscd,
+      stck_prpr: stock.stck_prpr,
+      prdy_vrss: stock.prdy_vrss,
+      prdy_vrss_sign: stock.prdy_vrss_sign,
+      prdy_ctrt: stock.prdy_ctrt,
+      hts_avls: stock.hts_avls,
+      per: stock.per,
+    };
   }
 
   /**