feat: 엣지 서비스 구현

Author: summersummerwhy

backend/src/edge/dtos/createEdge.dto.ts

@@ -1,8 +1,37 @@
 import { ApiProperty } from '@nestjs/swagger';
+import { IsString, IsNumber, IsIn } from 'class-validator';
+// import { Direction } from '../edge.entity';
+
+const Direction = ['N', 'S', 'E', 'W'];
 
 export class CreateEdgeDto {
+  @IsNumber()
+  @ApiProperty({
+    example: 1,
+    description: '출발 노드의 ID',
+  })
   fromNode: number;
+
+  @IsString()
+  @IsIn(Direction)
+  @ApiProperty({
+    example: 'N',
+    description: '출발 노드 지점 방향 (N, S, E, W 중 하나)',
+  })
   fromPoint: string;
+
+  @IsNumber()
+  @ApiProperty({
+    example: 1,
+    description: '도착 노드의 ID',
+  })
   toNode: number;
+
+  @IsString()
+  @IsIn(Direction)
+  @ApiProperty({
+    example: 'N',
+    description: '도착 노드 지점 방향 (N, S, E, W 중 하나)',
+  })
   toPoint: string;
 }

backend/src/edge/edge.entity.ts

@@ -8,12 +8,13 @@ import {
 } from 'typeorm';
 import { Node } from '../node/node.entity';
 
-export enum Direction {
-  NORTH = 'N',
-  WEST = 'W',
-  SOUTH = 'S',
-  EAST = 'E',
-}
+// TODO: frontend, backend가 공유하는 shared에 direction.enum.ts로 분리
+// export enum Direction {
+//   NORTH = 'N',
+//   SOUTH = 'S',
+//   EAST = 'E',
+//   WEST = 'W',
+// }
 
 @Entity()
 export class Edge {
@@ -28,17 +29,11 @@ export class Edge {
   @JoinColumn({ name: 'to_node_id' })
   toNode: Node;
 
-  @Column({
-    type: 'enum',
-    enum: Direction,
-  })
-  fromPoint: Direction;
-
-  @Column({
-    type: 'enum',
-    enum: Direction,
-  })
-  toPoint: Direction;
+  @Column()
+  fromPoint: string;
+
+  @Column()
+  toPoint: string;
 
   @Column({ nullable: true })
   type: string;

backend/src/edge/edge.service.ts

@@ -1,14 +1,66 @@
 import { Injectable } from '@nestjs/common';
 import { EdgeRepository } from './edge.repository';
+import { NodeRepository } from 'src/node/node.repository';
 import { Edge } from './edge.entity';
+import { CreateEdgeDto } from './dtos/createEdge.dto';
+import { EdgeNotFoundException } from 'src/exception/edge.exception';
 
 @Injectable()
 export class EdgeService {
-  constructor(private edgeRepository: EdgeRepository) {}
+  constructor(
+    private readonly edgeRepository: EdgeRepository,
+    private readonly nodeRepository: NodeRepository,
+  ) {}
 
   async createEdge(dto: CreateEdgeDto): Promise<Edge> {
+    const { fromNode, fromPoint, toNode, toPoint } = dto;
 
-    const edge = await this.edgeRepository.save {}
+    // 출발 노드를 조회한다.
+    const existingFromNode = await this.nodeRepository.findOneBy({
+      id: fromNode,
+    });
+    // 도착 노드를 조회한다.
+    const existingToNode = await this.nodeRepository.findOneBy({ id: toNode });
 
+    // 엣지를 생성한다.
+    return await this.edgeRepository.save({
+      fromNode: existingFromNode,
+      fromPoint,
+      toNode: existingToNode,
+      toPoint,
+    });
+  }
+
+  async deleteEdge(id: number): Promise<void> {
+    // 엣지를 삭제한다
+    const deleteResult = await this.edgeRepository.delete(id);
+
+    // 삭제된 엣지가 없으면 노드를 찾지 못한 것
+    if (!deleteResult.affected) {
+      throw new EdgeNotFoundException();
+    }
+  }
+
+  async findEdges(): Promise<Edge[]> {
+    // 모든 엣지들을 조회한다.
+    const edges = await this.edgeRepository.find({
+      relations: ['fromNode', 'toNode'],
+      select: {
+        id: true,
+        fromNode: {
+          id: true,
+        },
+        fromPoint: true,
+        toNode: {
+          id: true,
+        },
+        toPoint: true,
+      },
+    });
+    // 엣지가 없으면 NotFound 에러
+    if (!edges) {
+      throw new EdgeNotFoundException();
+    }
+    return edges;
   }
 }