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

README.md

@@ -7,15 +7,31 @@
 - 지하철역 목록 조회 인수 테스트 작성하기
 - 지하철역 삭제 인수 테스트 작성하기
 ## STEP2 지하철 노선 관리
-### 기능 목록  
-- 지하철 노선 생성
-- 지하철 노선 목록 조회
-- 지하철 노선 조회
-- 지하철 노선 수정
-- 지하철 노선 삭제
 ### 프로그래밍 요구사항
 > 아래의 순서로 기능을 구현하세요.  
 > 인수 조건을 검증하는 인수 테스트 작성  
 > 인수 테스트를 충족하는 기능 구현  
 > 인수 테스트의 결과가 다른 인수 테스트에 영향을 끼치지 않도록 인수 테스트를 서로 격리 시키세요.  
-> 인수 테스트의 재사용성과 가독성, 그리고 빠른 테스트 의도 파악을 위해 인수 테스트를 리팩터링 하세요  
+> 인수 테스트의 재사용성과 가독성, 그리고 빠른 테스트 의도 파악을 위해 인수 테스트를 리팩터링 하세요
+### 기능 목록  
+- [x] 지하철 노선 생성
+- [x] 지하철 노선 목록 조회
+- [x] 지하철 노선 조회
+- [x] 지하철 노선 수정
+- [x] 지하철 노선 삭제
+## STEP3 지하철 구간 관리
+### 프로그래밍 요구사항
+> 인수 테스트 주도 개발 프로세스에 맞춰서 기능을 구현하세요.
+- 요구사항 설명을 참고하여 인수 조건을 정의  
+- 인수 조건을 검증하는 인수 테스트 작성  
+- 인수 테스트를 충족하는 기능 구현  
+> 인수 조건은 인수 테스트 메서드 상단에 주석으로 작성하세요.  
+- 뼈대 코드의 인수 테스트를 참고  
+> 인수 테스트의 결과가 다른 인수 테스트에 영향을 끼치지 않도록 인수 테스트를 서로 격리 시키세요.  
+
+>인수 테스트의 재사용성과 가독성, 그리고 빠른 테스트 의도 파악을 위해 인수 테스트를 리팩터링 하세요.  
+### 구간 등록 기능
+- [x] 지하철 노선에 구간을 등록하는 기능을 구현
+- [x] 새로운 구간의 상행역은 해당 노선에 등록되어있는 하행 종점역이어야 한다.
+- [x] 새로운 구간의 하행역은 해당 노선에 등록되어있는 역일 수 없다.
+- [x] 새로운 구간 등록시 위 조건에 부합하지 않는 경우 에러 처리한다

src/main/java/subway/common/ErrorAdvice.java

@@ -12,4 +12,9 @@ public class ErrorAdvice {
     public ResponseEntity<String> notFoundEntity(NotFoundException e) {
         return ResponseEntity.badRequest().body(e.getMessage());
     }
+
+    @ExceptionHandler(IllegalArgumentException.class)
+    public ResponseEntity<String> illegalArgumentException(IllegalArgumentException e) {
+        return ResponseEntity.badRequest().body(e.getMessage());
+    }
 }

src/main/java/subway/common/exception/ErrorMessage.java

@@ -0,0 +1,7 @@
+package subway.common.exception;
+
+public class ErrorMessage {
+
+    public static String SECTION_INTEGRITY_ADD_ERROR_MESSAGE = "지하철구간 추가 조건에 위배됩니다";
+    public static String SECTION_INTEGRITY_REMOVE_ERROR_MESSAGE = "지하철구간 삭제 조건에 위배됩니다";
+}

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

@@ -1,12 +1,16 @@
 package subway.line.domain;
 
+import java.util.List;
 import subway.line.dto.LineRequest;
+import subway.line.dto.SectionRequest;
 
 import javax.persistence.Column;
+import javax.persistence.Embedded;
 import javax.persistence.Entity;
 import javax.persistence.GeneratedValue;
 import javax.persistence.GenerationType;
 import javax.persistence.Id;
+import subway.station.domain.Station;
 
 @Entity
 public class Line {
@@ -21,11 +25,17 @@ public class Line {
     @Column(length = 20, nullable = false)
     private String color;
 
-    public Line() {}
+    @Embedded
+    private Sections sections;
+
+    public Line() {
+        this.sections = new Sections();
+    }
 
     public Line(String name, String color) {
         this.name = name;
         this.color = color;
+        this.sections = new Sections();
     }
 
     public Long getId() {
@@ -44,4 +54,16 @@ public void update(LineRequest lineRequest) {
         this.name = lineRequest.getName();
         this.color = lineRequest.getColor();
     }
+
+    public void addSection(Section section) {
+        this.sections.addSection(section);
+    }
+
+    public List<Station> getStations() {
+        return sections.getStations();
+    }
+
+    public void deleteSection(Station station) {
+        sections.deleteSection(station);
+    }
 }

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

@@ -0,0 +1,63 @@
+package subway.line.domain;
+
+import subway.station.domain.Station;
+
+import javax.persistence.Entity;
+import javax.persistence.FetchType;
+import javax.persistence.GeneratedValue;
+import javax.persistence.GenerationType;
+import javax.persistence.Id;
+import javax.persistence.JoinColumn;
+import javax.persistence.ManyToOne;
+
+@Entity
+public class Section {
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
+
+    @ManyToOne(fetch = FetchType.LAZY)
+    @JoinColumn(name = "lineId")
+    private Line line;
+
+    @ManyToOne
+    @JoinColumn(name = "upStationId")
+    private Station upStation;
+
+    @ManyToOne
+    @JoinColumn(name = "downStationId")
+    private Station downStation;
+
+    private int distance;
+
+    public Section(Line line, Station upStation, Station downStation, int distance) {
+        this.line = line;
+        this.upStation = upStation;
+        this.downStation = downStation;
+        this.distance = distance;
+    }
+
+    public Section() {
+    }
+
+    public Long getId() {
+        return id;
+    }
+
+    public Line getLine() {
+        return line;
+    }
+
+    public Station getDownStation() {
+        return downStation;
+    }
+
+    public Station getUpStation() {
+        return upStation;
+    }
+
+    public int getDistance() {
+        return distance;
+    }
+}

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

@@ -0,0 +1,80 @@
+package subway.line.domain;
+
+import java.util.Optional;
+import java.util.stream.Collectors;
+import javax.persistence.CascadeType;
+import javax.persistence.OneToMany;
+
+import subway.common.exception.ErrorMessage;
+
+import javax.persistence.Embeddable;
+import java.util.ArrayList;
+import java.util.List;
+
+import subway.station.domain.Station;
+
+@Embeddable
+public class Sections {
+
+    @OneToMany(mappedBy = "line", orphanRemoval = true, cascade = CascadeType.ALL)
+    private List<Section> sections;
+
+    public Sections() {
+        this.sections = new ArrayList<>();
+    }
+
+    public void addSection(Section section) {
+        if (notValidSection(section) && alreadyValidSection(section)) {
+           throw new IllegalArgumentException(ErrorMessage.SECTION_INTEGRITY_ADD_ERROR_MESSAGE);
+        }
+        sections.add(section);
+    }
+
+    public List<Station> getStations() {
+        List<Station> stations = sections.stream()
+                .map(Section::getUpStation)
+                .collect(Collectors.toList());
+        setLastStations(stations);
+
+        return stations;
+    }
+
+    public void deleteSection(Station station) {
+        if(isNotLastStation(station) || sections.size() < 2) {
+            throw new IllegalArgumentException(ErrorMessage.SECTION_INTEGRITY_REMOVE_ERROR_MESSAGE);
+        }
+        sections.remove(sections.size() - 1);
+    }
+
+    private boolean notValidSection(Section section) {
+        if (sections.size() > 0) {
+            Station endStation = sections.get(sections.size() - 1).getDownStation();
+            return !endStation.equals(section.getUpStation());
+        }
+        return false;
+    }
+
+    private boolean alreadyValidSection(Section section) {
+        return sections.stream()
+                .map(Section::getDownStation)
+                .anyMatch(station -> station.getId().equals(section.getDownStation().getId()));
+    }
+
+    private void setLastStations(List<Station> stations) {
+        getLastSection().ifPresent(section -> stations.add(section.getDownStation()));
+    }
+
+    private Optional<Section> getLastSection() {
+        if (sections.size() > 0) {
+            return Optional.of(sections.get(sections.size() - 1));
+        }
+        return Optional.empty();
+    }
+
+    private boolean isNotLastStation(Station station) {
+
+        return getLastSection()
+                .filter(section -> section.getDownStation().getId().equals(station.getId()))
+                .isEmpty();
+    }
+}

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

@@ -1,12 +1,35 @@
 package subway.line.dto;
 
+import java.util.List;
+import java.util.stream.Collectors;
 import subway.line.domain.Line;
+import subway.station.domain.Station;
+import subway.station.dto.StationResponse;
 
 public class LineResponse {
 
-    private final long id;
-    private final String name;
-    private final String color;
+    private long id;
+    private String name;
+    private String color;
+    private List<StationResponse> stations;
+
+    public LineResponse() {
+    }
+
+
+    public LineResponse(long id, String name, String color, List<StationResponse> stations) {
+        this.id = id;
+        this.name = name;
+        this.color = color;
+        this.stations = stations;
+    }
+
+    public static LineResponse from(Line line) {
+        return new LineResponse(line.getId(), line.getName(),
+            line.getColor(), line.getStations().stream()
+            .map(StationResponse::new)
+            .collect(Collectors.toList()));
+    }
 
     public long getId() {
         return id;
@@ -20,13 +43,7 @@ public String getColor() {
         return color;
     }
 
-    public LineResponse(long id, String name, String color) {
-        this.id = id;
-        this.name = name;
-        this.color = color;
-    }
-
-    public LineResponse(Line line) {
-        this(line.getId(), line.getName(), line.getColor());
+    public List<StationResponse> getStations() {
+        return stations;
     }
 }

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

@@ -0,0 +1,22 @@
+package subway.line.dto;
+
+import subway.line.domain.Section;
+
+public class SectionRequest {
+
+    private long upStationId;
+    private long downStationId;
+    private int distance;
+
+    public long getUpStationId() {
+        return upStationId;
+    }
+
+    public long getDownStationId() {
+        return downStationId;
+    }
+
+    public int getDistance() {
+        return distance;
+    }
+}

src/main/java/subway/line/dto/SectionResponse.java

@@ -0,0 +1,29 @@
+package subway.line.dto;
+
+import subway.station.dto.StationResponse;
+
+public class SectionResponse {
+
+    private Long id;
+    private StationResponse upStation;
+    private StationResponse downStation;
+    private int distance;
+
+    public Long getId() {
+        return id;
+    }
+
+    public StationResponse getUpStation() {
+        return upStation;
+    }
+
+    public StationResponse getDownStation() {
+        return downStation;
+    }
+
+    public int getDistance() {
+        return distance;
+    }
+
+
+}