🚀 2단계 - 지하철 노선 관리 (지하철 노선 생성)

build.gradle

@@ -19,6 +19,7 @@ dependencies {
 
     // log
     implementation 'net.rakugakibox.spring.boot:logback-access-spring-boot-starter:2.7.1'
+    implementation 'org.projectlombok:lombok:1.18.22'
 
     testImplementation 'org.springframework.boot:spring-boot-starter-test'
     testImplementation 'io.rest-assured:rest-assured:4.5.1'
@@ -28,4 +29,4 @@ dependencies {
 
 test {
     useJUnitPlatform()
-}
+}

src/main/java/subway/SubwayService.java

@@ -0,0 +1,12 @@
+package subway;
+
+public interface SubwayService<T, V> {
+
+    V create(T request);
+
+    // SubwayResponse get(SubwayRequest subwayRequest);
+    //
+    // SubwayResponse update(SubwayRequest subwayRequest);
+    //
+    // SubwayResponse delete(SubwayRequest subwayRequest);
+}

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

@@ -0,0 +1,30 @@
+package subway.line;
+
+import java.net.URI;
+
+import org.springframework.http.MediaType;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RestController;
+
+import lombok.RequiredArgsConstructor;
+import subway.line.request.LineRequest;
+import subway.line.response.LineResponse;
+
+@RequiredArgsConstructor
+@RestController
+public class LineController {
+
+    private final LineService lineService;
+
+    @PostMapping(
+        value = "/lines",
+        consumes = MediaType.APPLICATION_JSON_VALUE,
+        produces = MediaType.APPLICATION_JSON_VALUE
+    )
+    public ResponseEntity<LineResponse> createLine(@RequestBody LineRequest lineRequest) {
+        LineResponse line = lineService.create(lineRequest);
+        return ResponseEntity.created(URI.create("/lines/" + line.getId())).body(line);
+    }
+}

src/main/java/subway/line/LineRepository.java

@@ -0,0 +1,8 @@
+package subway.line;
+
+import org.springframework.data.jpa.repository.JpaRepository;
+
+import subway.line.domain.Line;
+
+public interface LineRepository extends JpaRepository<Line, Long> {
+}

src/main/java/subway/line/LineService.java

@@ -0,0 +1,57 @@
+package subway.line;
+
+import java.util.Arrays;
+
+import javax.transaction.Transactional;
+
+import org.springframework.stereotype.Service;
+
+import lombok.RequiredArgsConstructor;
+import subway.SubwayService;
+import subway.line.domain.Line;
+import subway.line.request.LineRequest;
+import subway.line.response.LineResponse;
+import subway.station.StationRepository;
+import subway.station.domain.Station;
+
+@RequiredArgsConstructor
+@Service
+public class LineService implements SubwayService<LineRequest, LineResponse> {
+
+    private final LineRepository lineRepository;
+    private final StationRepository stationRepository;
+
+    @Transactional
+    @Override
+    public LineResponse create(LineRequest lineRequest) {
+        return saveLine(lineRequest);
+    }
+
+    private LineResponse saveLine(LineRequest lineRequest) {
+        Line line = lineRepository.save(new Line(
+            lineRequest.getName(),
+            lineRequest.getColor(),
+            lineRequest.getUpStationId(),
+            lineRequest.getDownStationId(),
+            lineRequest.getDistance()
+        ));
+        return createLineResponse(line);
+    }
+
+    private LineResponse createLineResponse(Line line) {
+        Station upStation = getStation(line.getUpStationId());
+        Station downStation = getStation(line.getDownStationId());
+
+        return LineResponse.builder()
+            .id(line.getId())
+            .name(line.getName())
+            .color(line.getColor())
+            .stations(Arrays.asList(upStation, downStation))
+            .build();
+    }
+
+    private Station getStation(Long id) {
+        return stationRepository.findById(id)
+            .orElseGet(() -> stationRepository.save(new Station("지하철역" + id)));
+    }
+}

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

@@ -0,0 +1,38 @@
+package subway.line.domain;
+
+import javax.persistence.Column;
+import javax.persistence.Entity;
+import javax.persistence.GeneratedValue;
+import javax.persistence.GenerationType;
+import javax.persistence.Id;
+
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+
+@Getter
+@NoArgsConstructor
+@Entity
+public class Line {
+
+    @Id
+    @GeneratedValue(strategy = GenerationType.IDENTITY)
+    private Long id;
+    @Column(length = 20, nullable = false)
+    private String name;
+    @Column(length = 20, nullable = false)
+    private String color;
+    @Column(nullable = false)
+    private Long upStationId;
+    @Column(nullable = false)
+    private Long downStationId;
+    @Column(nullable = false)
+    private Long distance;
+
+    public Line(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/line/request/LineRequest.java

@@ -0,0 +1,16 @@
+package subway.line.request;
+
+import lombok.Builder;
+import lombok.Getter;
+
+@Getter
+@Builder
+public class LineRequest {
+
+    private final String name;
+    private final String color;
+    private final Long upStationId;
+    private final Long downStationId;
+    private final Long distance;
+
+}

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

@@ -0,0 +1,18 @@
+package subway.line.response;
+
+import java.util.List;
+
+import lombok.Builder;
+import lombok.Getter;
+import subway.station.domain.Station;
+
+@Getter
+@Builder
+public class LineResponse {
+
+    private final Long id;
+    private final String name;
+    private final String color;
+    private final List<Station> stations;
+
+}

src/main/java/subway/StationController.java → src/main/java/subway/station/StationController.java

@@ -1,22 +1,29 @@
-package subway;
-
-import org.springframework.http.ResponseEntity;
-import org.springframework.web.bind.annotation.*;
+package subway.station;
 
 import java.net.URI;
 import java.util.List;
 
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.DeleteMapping;
+import org.springframework.web.bind.annotation.GetMapping;
+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.RestController;
+
+import lombok.RequiredArgsConstructor;
+import subway.station.request.StationRequest;
+import subway.station.response.StationResponse;
+
 @RestController
+@RequiredArgsConstructor
 public class StationController {
-    private StationService stationService;
 
-    public StationController(StationService stationService) {
-        this.stationService = stationService;
-    }
+    private final StationService stationService;
 
     @PostMapping("/stations")
     public ResponseEntity<StationResponse> createStation(@RequestBody StationRequest stationRequest) {
-        StationResponse station = stationService.saveStation(stationRequest);
+        StationResponse station = stationService.create(stationRequest);
         return ResponseEntity.created(URI.create("/stations/" + station.getId())).body(station);
     }
 
@@ -30,4 +37,4 @@ public ResponseEntity<Void> deleteStation(@PathVariable Long id) {
         stationService.deleteStationById(id);
         return ResponseEntity.noContent().build();
     }
-}
+}

src/main/java/subway/StationRepository.java → src/main/java/subway/station/StationRepository.java

@@ -1,6 +1,8 @@
-package subway;
+package subway.station;
 
 import org.springframework.data.jpa.repository.JpaRepository;
 
+import subway.station.domain.Station;
+
 public interface StationRepository extends JpaRepository<Station, Long> {
 }