[BE] 7.03 매수 API 구현 #49

BE/src/app.module.ts

@@ -9,6 +9,7 @@ import { StockIndexModule } from './stock/index/stock-index.module';
 import { StockTopfiveModule } from './stock/topfive/stock-topfive.module';
 import { KoreaInvestmentModule } from './koreaInvestment/korea-investment.module';
 import { SocketModule } from './websocket/socket.module';
+import { StockOrderModule } from './stock/order/stock-order.module';
 import { typeOrmConfig } from './configs/typeorm.config';
 
 @Module({
@@ -21,6 +22,7 @@ import { typeOrmConfig } from './configs/typeorm.config';
     StockIndexModule,
     StockTopfiveModule,
     SocketModule,
+    StockOrderModule,
   ],
   controllers: [AppController],
   providers: [AppService],

BE/src/auth/jwt-auth-guard.ts

@@ -0,0 +1,5 @@
+import { AuthGuard } from '@nestjs/passport';
+import { Injectable } from '@nestjs/common';
+
+@Injectable()
+export class JwtAuthGuard extends AuthGuard('jwt') {}

BE/src/stock/order/dto/stock-order-request.dto.ts

@@ -0,0 +1,19 @@
+import { ApiProperty } from '@nestjs/swagger';
+import { IsInt, IsNumber, IsPositive } from 'class-validator';
+
+export class StockOrderRequestDto {
+  @ApiProperty({ description: '주식 id' })
+  @IsInt()
+  @IsPositive()
+  stock_id: number;
+
+  @ApiProperty({ description: '매수/매도 희망 가격' })
+  @IsNumber()
+  @IsPositive()
+  price: number;
+
+  @ApiProperty({ description: '매수/매도 희망 수량' })
+  @IsInt()
+  @IsPositive()
+  amount: number;
+}

BE/src/stock/order/enum/status-type.ts

@@ -0,0 +1,4 @@
+export enum StatusType {
+  PENDING = 'PENDING',
+  COMPLETE = 'COMPLETE',
+}

BE/src/stock/order/enum/trade-type.ts

@@ -0,0 +1,4 @@
+export enum TradeType {
+  SELL = 'SELL',
+  BUY = 'BUY',
+}

BE/src/stock/order/interface/request.interface.ts

@@ -0,0 +1,9 @@
+export interface RequestInterface {
+  user: {
+    id: number;
+    email: string;
+    password: string;
+    tutorial: boolean;
+    kakaoId: number;
+  };
+}

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

@@ -0,0 +1,42 @@
+import {
+  Body,
+  Controller,
+  Post,
+  Req,
+  UseGuards,
+  ValidationPipe,
+} from '@nestjs/common';
+import {
+  ApiBearerAuth,
+  ApiOperation,
+  ApiResponse,
+  ApiTags,
+} from '@nestjs/swagger';
+import { StockOrderService } from './stock-order.service';
+import { StockOrderRequestDto } from './dto/stock-order-request.dto';
+import { JwtAuthGuard } from '../../auth/jwt-auth-guard';
+import { RequestInterface } from './interface/request.interface';
+
+@Controller('/api/stocks/trade')
+@ApiTags('주식 매수/매도 API')
+export class StockOrderController {
+  constructor(private readonly stockTradeService: StockOrderService) {}
+
+  @Post('/buy')
+  @ApiBearerAuth()
+  @UseGuards(JwtAuthGuard)
+  @ApiOperation({
+    summary: '주식 매수 API',
+    description: '주식 id, 매수 가격, 수량으로 주식을 매수한다.',
+  })
+  @ApiResponse({
+    status: 201,
+    description: '주식 매수 성공',
+  })
+  async buy(
+    @Req() request: RequestInterface,
+    @Body(ValidationPipe) stockOrderRequest: StockOrderRequestDto,
+  ) {
+    await this.stockTradeService.buy(request.user.id, stockOrderRequest);
+  }
+}

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

@@ -0,0 +1,46 @@
+import {
+  Column,
+  CreateDateColumn,
+  Entity,
+  PrimaryGeneratedColumn,
+} from 'typeorm';
+import { TradeType } from './enum/trade-type';
+import { StatusType } from './enum/status-type';
+
+@Entity('orders')
+export class Order {
+  @PrimaryGeneratedColumn()
+  id: number;
+
+  @Column({ nullable: false })
+  user_id: number;
+
+  @Column({ nullable: false })
+  stock_id: number;
+
+  @Column({
+    type: 'enum',
+    enum: TradeType,
+    nullable: false,
+  })
+  trade_type: TradeType;
+
+  @Column({ nullable: false })
+  amount: number;
+
+  @Column({ nullable: false })
+  price: number;
+
+  @Column({
+    type: 'enum',
+    enum: StatusType,
+    nullable: false,
+  })
+  status: StatusType;
+
+  @CreateDateColumn()
+  created_at: Date;
+
+  @Column({ nullable: true })
+  completed_at?: Date;
+}

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

@@ -0,0 +1,13 @@
+import { Module } from '@nestjs/common';
+import { TypeOrmModule } from '@nestjs/typeorm';
+import { StockOrderController } from './stock-order.controller';
+import { StockOrderService } from './stock-order.service';
+import { Order } from './stock-order.entity';
+import { StockOrderRepository } from './stock-order.repository';
+
+@Module({
+  imports: [TypeOrmModule.forFeature([Order])],
+  controllers: [StockOrderController],
+  providers: [StockOrderService, StockOrderRepository],
+})
+export class StockOrderModule {}

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

@@ -0,0 +1,11 @@
+import { DataSource, Repository } from 'typeorm';
+import { InjectDataSource } from '@nestjs/typeorm';
+import { Injectable } from '@nestjs/common';
+import { Order } from './stock-order.entity';
+
+@Injectable()
+export class StockOrderRepository extends Repository<Order> {
+  constructor(@InjectDataSource() dataSource: DataSource) {
+    super(Order, dataSource.createEntityManager());
+  }
+}

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

@@ -0,0 +1,23 @@
+import { Injectable } from '@nestjs/common';
+import { StockOrderRequestDto } from './dto/stock-order-request.dto';
+import { StockOrderRepository } from './stock-order.repository';
+import { TradeType } from './enum/trade-type';
+import { StatusType } from './enum/status-type';
+
+@Injectable()
+export class StockOrderService {
+  constructor(private readonly stockOrderRepository: StockOrderRepository) {}
+
+  async buy(userId: number, stockOrderRequest: StockOrderRequestDto) {
+    const order = this.stockOrderRepository.create({
+      user_id: userId,
+      stock_id: stockOrderRequest.stock_id,
+      trade_type: TradeType.BUY,
+      amount: stockOrderRequest.amount,
+      price: stockOrderRequest.price,
+      status: StatusType.PENDING,
+    });
+
+    await this.stockOrderRepository.save(order);
+  }
+}

BE/src/util/swagger.ts

@@ -13,6 +13,7 @@ export function setupSwagger(app: INestApplication): void {
     .setTitle('Juga API')
     .setDescription('Juga API 문서입니다.')
     .setVersion('1.0.0')
+    .addBearerAuth()
     .build();
 
   const document = SwaggerModule.createDocument(app, options);