🚀 3단계 - 지하철 구간 관리

README.md

@@ -16,3 +16,14 @@
 - [X] 지하철 노선 조회
 - [X] 지하철 노선 수정
 - [X] 지하철 노선 삭제
+
+# 🚀 3단계 - 지하철 구간 관리
+
+## 요구사항
+
+### 구간 등록 기능
+
+ - [X] 지하철 노선에 구간을 등록
+ - [X] 새로운 구간의 상행역은 해당 노선에 등록되어있는 하행 종점역이어야 한다.
+ - [X] 새로운 구간의 하행역은 해당 노선에 등록되어있는 역일 수 없다.
+ - [X] 새로운 구간 등록시 위 조건에 부합하지 않는 경우 에러 처리한다.

src/main/java/subway/common/execption/CustomExceptionHandler.java

@@ -0,0 +1,16 @@
+package subway.common.execption;
+
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.ControllerAdvice;
+import org.springframework.web.bind.annotation.ExceptionHandler;
+import subway.common.execption.response.ErrorResponse;
+
+@ControllerAdvice
+public class CustomExceptionHandler {
+
+    @ExceptionHandler(IllegalArgumentException.class)
+    public ResponseEntity<ErrorResponse> handleIllegalArgumentException(IllegalArgumentException e) {
+        return ResponseEntity.badRequest().body(ErrorResponse.of(e.getMessage()));
+    }
+
+}

src/main/java/subway/common/execption/response/ErrorResponse.java

@@ -0,0 +1,25 @@
+package subway.common.execption.response;
+
+public class ErrorResponse {
+
+    private final String message;
+
+    public ErrorResponse(String message) {
+        this.message = message;
+    }
+
+    public static ErrorResponse of(String message) {
+        return new ErrorResponse(message);
+    }
+
+    @Override
+    public String toString() {
+        return "ErrorResponse{" +
+                "message='" + message + '\'' +
+                '}';
+    }
+
+    public String getMessage() {
+        return message;
+    }
+}

src/main/java/subway/route/code/RouteValidateTypeCode.java → src/main/java/subway/line/code/LineValidateTypeCode.java

@@ -1,6 +1,6 @@
-package subway.route.code;
+package subway.line.code;
 
-public enum RouteValidateTypeCode {
+public enum LineValidateTypeCode {
 
     SAVE("이름"),
     UPDATE("아이디와 이름"),
@@ -9,7 +9,7 @@ public enum RouteValidateTypeCode {
 
     private final String description;
 
-    RouteValidateTypeCode(String description) {
+    LineValidateTypeCode(String description) {
         this.description = description;
     }
 

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

@@ -0,0 +1,75 @@
+package subway.line.controller;
+
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.*;
+import subway.line.code.LineValidateTypeCode;
+import subway.line.domain.Line;
+import subway.line.dto.request.LineRequest;
+import subway.line.dto.request.SectionRequest;
+import subway.line.dto.response.LineResponse;
+import subway.line.dto.response.SectionResponse;
+import subway.line.service.LineService;
+
+import java.net.URI;
+import java.util.List;
+
+@RestController
+@RequestMapping("/lines")
+public class LineController {
+
+    private final LineService lineService;
+
+    public LineController(LineService LineService) {
+        this.lineService = LineService;
+    }
+
+    @PostMapping()
+    public ResponseEntity<LineResponse> createLine(@RequestBody LineRequest LineRequest) {
+        LineResponse Line = lineService.saveLine(LineValidateTypeCode.SAVE, LineRequest);
+        return ResponseEntity.created(URI.create("/lines/" + Line.getId())).body(Line);
+    }
+
+    @GetMapping("/{id}")
+    public ResponseEntity<LineResponse> inquiryLine(@PathVariable Long id) {
+        LineResponse Line = lineService.inquiryLine(id);
+        return ResponseEntity.ok().body(Line);
+    }
+
+    @GetMapping("")
+    public ResponseEntity<List<LineResponse>> inquiryLines() {
+        List<LineResponse> Lines = lineService.inquiryLines();
+        return ResponseEntity.ok().body(Lines);
+    }
+
+    @PutMapping("/{id}")
+    public ResponseEntity<LineResponse> updateLine(@PathVariable Long id, @RequestBody LineRequest LineRequest) {
+        LineRequest.saveId(id);
+        LineResponse Line = lineService.saveLine(LineValidateTypeCode.UPDATE, LineRequest);
+        return ResponseEntity.ok().body(Line);
+    }
+
+
+    @DeleteMapping("/{id}")
+    public ResponseEntity<LineResponse> deleteLine(@PathVariable Long id) {
+        lineService.deleteLine(id);
+        return ResponseEntity.noContent().build();
+    }
+
+    @PostMapping("/{id}/sections")
+    public ResponseEntity<SectionResponse> createSection(@PathVariable Long id, @RequestBody SectionRequest sectionRequest) {
+        return ResponseEntity.created(URI.create("/lines/" + id + "/sections")).body(lineService.updateSection(id, sectionRequest));
+    }
+
+    @GetMapping("/{id}/sections")
+    public ResponseEntity<List<SectionResponse>> inquirySection(@PathVariable Long id) {
+        return ResponseEntity.ok().body(lineService.inquirySection(id));
+    }
+
+    @DeleteMapping("/{id}/sections")
+    public ResponseEntity<SectionResponse> deleteSection(@PathVariable Long id, @RequestParam Long stationId) {
+        lineService.deleteSection(id, stationId);
+        return ResponseEntity.noContent().build();
+    }
+
+
+}

src/main/java/subway/route/domain/Route.java → src/main/java/subway/line/domain/Line.java

@@ -1,14 +1,12 @@
-package subway.route.domain;
-
-import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
-import com.fasterxml.jackson.annotation.JsonProperty;
-import subway.station.domain.Station;
+package subway.line.domain;
 
 import javax.persistence.*;
+import java.util.ArrayList;
+import java.util.List;
 import java.util.Objects;
 
 @Entity
-public class Route {
+public class Line {
 
     @Id
     @GeneratedValue(strategy = GenerationType.IDENTITY)
@@ -25,24 +23,30 @@ public class Route {
     @Column(length = 10, nullable = false)
     private Integer distance;
 
-    public Route() {
+    @OneToMany(mappedBy = "line", fetch = FetchType.EAGER)
+    private List<Section> sections = new ArrayList<>();
+
+    public Line() {
     }
 
-    public Route(Long id, String name, String color, Integer distance) {
+    public Line(Long id, String name, String color, Integer distance) {
         this.id = id;
         this.name = name;
         this.color = color;
         this.distance = distance;
     }
 
-    public static Route of(Long id, String name, String color, Integer distance) {
-        return new Route(id, name, color, distance);
+    public static Line of(Long id, String name, String color, Integer distance) {
+        return new Line(id, name, color, distance);
     }
 
     public void saveStations(Stations stations) {
         this.stations = stations;
     }
 
+    public void saveSections(List<Section> sections) {
+        this.sections = sections;
+    }
     public Long getId() {
         return id;
     }
@@ -63,12 +67,16 @@ public Integer getDistance() {
         return distance;
     }
 
+    public List<Section> getSections() {
+        return sections;
+    }
+
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;
         if (o == null || getClass() != o.getClass()) return false;
 
-        Route route = (Route) o;
+        Line route = (Line) o;
 
         if (!Objects.equals(id, route.id)) return false;
         return Objects.equals(name, route.name);

src/main/java/subway/line/domain/Section.java

@@ -0,0 +1,51 @@
+package subway.line.domain;
+
+import subway.station.domain.Station;
+
+import javax.persistence.*;
+
+@Entity
+public class Section {
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "line_id", nullable = false)
+    private Line line;
+
+    @ManyToOne
+    @JoinColumn(name = "station_id", nullable = false)
+    private Station station;
+
+    public Section() {
+    }
+
+    public Section(Line line, Station station) {
+        this.line = line;
+        this.station = station;
+    }
+
+    public Long getId() {
+        return id;
+    }
+
+    public Line getLine() {
+        return line;
+    }
+
+    public Station getStation() {
+        return station;
+    }
+
+    @Override
+    public String toString() {
+        return "Section{" +
+                "id=" + id +
+                ", line=" + line +
+                ", station=" + station +
+                '}';
+    }
+
+}

src/main/java/subway/line/domain/Stations.java

@@ -0,0 +1,43 @@
+package subway.line.domain;
+
+import subway.station.domain.Station;
+
+import javax.persistence.Embeddable;
+import javax.persistence.JoinColumn;
+import javax.persistence.OneToOne;
+import java.io.Serializable;
+
+@Embeddable
+public class Stations implements Serializable {
+
+    @OneToOne
+    @JoinColumn(name = "up_station_id", nullable = false)
+    private Station upStation;
+
+    @OneToOne
+    @JoinColumn(name = "down_station_id", nullable = false)
+    private Station downStation;
+
+    public Stations() {}
+
+    public Stations(Station upStation, Station downStation) {
+        this.upStation = upStation;
+        this.downStation = downStation;
+    }
+
+    public Station getUpStation() {
+        return upStation;
+    }
+
+    public Station getDownStation() {
+        return downStation;
+    }
+
+    @Override
+    public String toString() {
+        return "Stations{" +
+                "upStation=" + upStation +
+                ", downStation=" + downStation +
+                '}';
+    }
+}

src/main/java/subway/route/dto/RouteRequest.java → src/main/java/subway/line/dto/request/LineRequest.java

@@ -1,8 +1,8 @@
-package subway.route.dto;
+package subway.line.dto.request;
 
-import subway.route.domain.Route;
+import subway.line.domain.Line;
 
-public class RouteRequest {
+public class LineRequest {
 
     private Long id;
     private String name;
@@ -11,8 +11,8 @@ public class RouteRequest {
     private Long downStationId;
     private Integer distance;
 
-    public Route toEntity() {
-        return Route.of(id, name, color, distance);
+    public Line toEntity() {
+        return Line.of(id, name, color, distance);
     }
 
     public void saveId(Long id) {

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

@@ -0,0 +1,32 @@
+package subway.line.dto.request;
+
+public class SectionRequest {
+
+    private final Long id;
+    private final Long downStationId;
+    private final Long upStationId;
+    private final Long distance;
+
+    public SectionRequest(Long id, Long downStationId, Long upStationId, Long distance) {
+        this.id = id;
+        this.downStationId = downStationId;
+        this.upStationId = upStationId;
+        this.distance = distance;
+    }
+
+    public Long getId() {
+        return id;
+    }
+
+    public Long getDownStationId() {
+        return downStationId;
+    }
+
+    public Long getUpStationId() {
+        return upStationId;
+    }
+
+    public Long getDistance() {
+        return distance;
+    }
+}