🚀 1단계 - 지하철 구간 추가 기능 개선

build.gradle

@@ -20,6 +20,10 @@ dependencies {
     // log
     implementation 'net.rakugakibox.spring.boot:logback-access-spring-boot-starter:2.7.1'
 
+    //lombok
+    compileOnly 'org.projectlombok:lombok'
+    annotationProcessor 'org.projectlombok:lombok'
+
     // jgraph
     implementation 'org.jgrapht:jgrapht-core:1.0.1'
 

src/main/java/subway/SubwayApplication.java

@@ -0,0 +1,13 @@
+package subway;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+
+@SpringBootApplication
+public class SubwayApplication {
+
+    public static void main(String[] args) {
+        SpringApplication.run(SubwayApplication.class, args);
+    }
+
+}

src/main/java/subway/common/DatabaseCleaner.java

@@ -0,0 +1,36 @@
+package subway.common;
+
+import java.util.List;
+import java.util.stream.Collectors;
+import javax.persistence.EntityManager;
+import javax.persistence.metamodel.EntityType;
+import javax.transaction.Transactional;
+import org.springframework.stereotype.Component;
+
+@Component
+public class DatabaseCleaner {
+    private final EntityManager entityManager;
+    private final List<String> tableNames;
+
+    public DatabaseCleaner(final EntityManager entityManager) {
+        this.entityManager = entityManager;
+        this.tableNames = entityManager.getMetamodel()
+            .getEntities()
+            .stream()
+            .map(EntityType::getName)
+            .map(String::toLowerCase)
+            .collect(Collectors.toList());
+    }
+
+    @Transactional
+    public void execute() {
+        entityManager.flush();
+        entityManager.createNativeQuery("SET REFERENTIAL_INTEGRITY FALSE").executeUpdate();
+
+        for (String tableName : tableNames) {
+            entityManager.createNativeQuery("TRUNCATE TABLE " + tableName).executeUpdate();
+        }
+
+        entityManager.createNativeQuery("SET REFERENTIAL_INTEGRITY TRUE").executeUpdate();
+    }
+}

src/main/java/subway/controller/LineController.java

@@ -0,0 +1,47 @@
+package subway.controller;
+
+import lombok.RequiredArgsConstructor;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.*;
+import subway.dto.request.LineModifyRequest;
+import subway.dto.request.LineRequest;
+import subway.dto.response.LineResponse;
+import subway.service.LineService;
+
+import java.net.URI;
+import java.util.List;
+
+@RestController
+@RequiredArgsConstructor
+public class LineController {
+    private final LineService lineService;
+
+    @PostMapping("/lines")
+    public ResponseEntity<LineResponse> createLine(@RequestBody LineRequest request) {
+        LineResponse line = lineService.saveLine(request);
+        return ResponseEntity.created(URI.create("/lines/" + line.getId())).body(line);
+    }
+
+    @GetMapping("/lines")
+    public ResponseEntity<List<LineResponse>> findLines() {
+        return ResponseEntity.ok(lineService.findLines());
+    }
+
+    @GetMapping("/lines/{id}")
+    public ResponseEntity<LineResponse> findLine(@PathVariable Long id) {
+        LineResponse line = lineService.findLineResponse(id);
+        return ResponseEntity.created(URI.create("/lines/" + line.getId())).body(line);
+    }
+
+    @DeleteMapping("/lines/{id}")
+    public ResponseEntity<Void> deleteLine(@PathVariable Long id) {
+        lineService.deleteLine(id);
+        return ResponseEntity.noContent().build();
+    }
+
+    @PutMapping("/lines/{id}")
+    public ResponseEntity<Void> modifyStation(@PathVariable Long id, @RequestBody LineModifyRequest request) {
+        lineService.modifyLine(id, request);
+        return ResponseEntity.noContent().build();
+    }
+}

src/main/java/subway/controller/SectionController.java

@@ -0,0 +1,35 @@
+package subway.controller;
+
+import java.net.URI;
+import lombok.RequiredArgsConstructor;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.DeleteMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+import subway.dto.request.SectionRequest;
+import subway.dto.response.LineResponse;
+import subway.service.SectionService;
+
+@RequestMapping("/lines/{id}/sections")
+@RestController
+@RequiredArgsConstructor
+public class SectionController {
+
+    private final SectionService sectionService;
+
+    @PostMapping
+    public ResponseEntity<LineResponse> addSection(@PathVariable Long id, @RequestBody SectionRequest request) {
+        LineResponse line = sectionService.addSection(id, request);
+        return ResponseEntity.created(URI.create("/lines/" + line.getId())).body(line);
+    }
+
+    @DeleteMapping
+    public ResponseEntity<Void> deleteSection(@PathVariable Long id, Long stationId) {
+        sectionService.deleteSection(id, stationId);
+        return ResponseEntity.noContent().build();
+    }
+
+}

src/main/java/subway/controller/StationController.java

@@ -0,0 +1,36 @@
+package subway.controller;
+
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.*;
+import subway.dto.request.StationRequest;
+import subway.dto.response.StationResponse;
+import subway.service.StationService;
+
+import java.net.URI;
+import java.util.List;
+
+@RestController
+public class StationController {
+    private StationService stationService;
+
+    public StationController(StationService stationService) {
+        this.stationService = stationService;
+    }
+
+    @PostMapping("/stations")
+    public ResponseEntity<StationResponse> createStation(@RequestBody StationRequest stationRequest) {
+        StationResponse station = stationService.saveStation(stationRequest);
+        return ResponseEntity.created(URI.create("/stations/" + station.getId())).body(station);
+    }
+
+    @GetMapping(value = "/stations")
+    public ResponseEntity<List<StationResponse>> showStations() {
+        return ResponseEntity.ok().body(stationService.findAllStations());
+    }
+
+    @DeleteMapping("/stations/{id}")
+    public ResponseEntity<Void> deleteStation(@PathVariable Long id) {
+        stationService.deleteStationById(id);
+        return ResponseEntity.noContent().build();
+    }
+}

src/main/java/subway/dto/request/LineModifyRequest.java

@@ -0,0 +1,17 @@
+package subway.dto.request;
+
+
+import lombok.Builder;
+import lombok.Getter;
+
+@Getter
+public class LineModifyRequest {
+    private String name;
+    private String color;
+
+    @Builder
+    public LineModifyRequest(String name, String color) {
+        this.name = name;
+        this.color = color;
+    }
+}

src/main/java/subway/dto/request/LineRequest.java

@@ -0,0 +1,23 @@
+package subway.dto.request;
+
+
+import lombok.Builder;
+import lombok.Getter;
+
+@Getter
+public class LineRequest {
+    private String name;
+    private String color;
+    private Long upStationId;
+    private Long downStationId;
+    private Long distance;
+
+    @Builder
+    public LineRequest(String name, String color, Long upStationId, Long downStationId, Long distance) {
+        this.name = name;
+        this.color = color;
+        this.upStationId = upStationId;
+        this.downStationId = downStationId;
+        this.distance = distance;
+    }
+}

src/main/java/subway/dto/request/SectionRequest.java

@@ -0,0 +1,18 @@
+package subway.dto.request;
+
+import lombok.Builder;
+import lombok.Getter;
+
+@Getter
+public class SectionRequest {
+    private Long upStationId;
+    private Long downStationId;
+    private Long distance;
+
+    @Builder
+    public SectionRequest(Long upStationId, Long downStationId, Long distance) {
+        this.upStationId = upStationId;
+        this.downStationId = downStationId;
+        this.distance = distance;
+    }
+}

src/main/java/subway/dto/request/StationRequest.java

@@ -0,0 +1,9 @@
+package subway.dto.request;
+
+public class StationRequest {
+    private String name;
+
+    public String getName() {
+        return name;
+    }
+}

src/main/java/subway/dto/response/LineResponse.java

@@ -0,0 +1,41 @@
+package subway.dto.response;
+
+
+import java.util.stream.Collectors;
+import lombok.Builder;
+import lombok.Getter;
+import subway.entity.Line;
+import subway.entity.Station;
+
+import java.util.List;
+
+@Getter
+public class LineResponse {
+    private Long id;
+    private String name;
+    private String color;
+    private List<StationResponse> stations;
+
+    private Long totalDistance;
+
+    @Builder
+    public LineResponse(Long id, String name, String color, List<StationResponse> stations, Long totalDistance) {
+        this.id = id;
+        this.name = name;
+        this.color = color;
+        this.stations = stations;
+        this.totalDistance = totalDistance;
+    }
+
+    public static LineResponse from(Line line) {
+        return LineResponse.builder()
+            .id(line.getId())
+            .name(line.getName())
+            .color(line.getColor())
+            .stations(line.getStations().stream()
+                .map(StationResponse::from)
+                .collect(Collectors.toList()))
+            .totalDistance(line.getSections().totalDistance())
+            .build();
+    }
+}

src/main/java/subway/dto/response/StationResponse.java

@@ -0,0 +1,31 @@
+package subway.dto.response;
+
+import lombok.Builder;
+import subway.entity.Station;
+
+public class StationResponse {
+    private Long id;
+    private String name;
+
+    @Builder
+    public StationResponse(Long id, String name) {
+        this.id = id;
+        this.name = name;
+    }
+
+    public Long getId() {
+        return id;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+
+    public static StationResponse from(Station station) {
+        return StationResponse.builder()
+            .id(station.getId())
+            .name(station.getName())
+            .build();
+    }
+}

src/main/java/subway/entity/Line.java

@@ -0,0 +1,47 @@
+package subway.entity;
+
+import java.util.Arrays;
+import java.util.List;
+import javax.persistence.Embedded;
+import javax.persistence.Entity;
+import javax.persistence.GeneratedValue;
+import javax.persistence.GenerationType;
+import javax.persistence.Id;
+import lombok.Builder;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import subway.dto.request.LineModifyRequest;
+
+@Getter
+@Entity
+@NoArgsConstructor
+public class Line {
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
+
+    private String name;
+
+    private String color;
+
+    @Embedded
+    private Sections sections;
+
+    @Builder
+    public Line(String name, String color, Section section) {
+        this.name = name;
+        this.color = color;
+        this.sections = Sections.builder()
+            .sections(Arrays.asList(section))
+            .build();
+    }
+
+    public List<Station> getStations() {
+        return this.sections.stations();
+    }
+
+    public void modify(LineModifyRequest request) {
+        this.name = request.getName();
+        this.color = request.getColor();
+    }
+}