3단계 - 수강신청(DB 적용)

src/main/java/nextstep/courses/domain/Capacity.java

@@ -1,5 +1,7 @@
 package nextstep.courses.domain;
 
+import java.util.Objects;
+
 public class Capacity {
     private final int value;
 
@@ -17,4 +19,17 @@ public int getValue() {
     public boolean isFull(int registeredCount) {
         return registeredCount >= value;
     }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (!(o instanceof Capacity)) return false;
+        Capacity capacity = (Capacity) o;
+        return value == capacity.value;
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(value);
+    }
 }

src/main/java/nextstep/courses/domain/Image.java

@@ -1,16 +1,16 @@
 package nextstep.courses.domain;
 
 public class Image {
+    private static final int MAX_IMAGE_SIZE = 1024 * 1024;
+    private static final int WIDTH_LIMIT = 300;
+    private static final int HEIGHT_LIMIT = 200;
+    private static final double REQUIRED_ASPECT_RATIO = 1.5;
+    private static final double ASPECT_RATIO_TOLERANCE = 0.01;
     private String fileName;
     private String contentType;
     private long sizeInBytes;
     private int width;
     private int height;
-    private static final int IMAGE_SIZE_1MB = 1024 * 1024;
-    private static final int WIDTH_LIMIT = 300;
-    private static final int HEIGHT_LIMIT = 200;
-    private static final double REQUIRED_ASPECT_RATIO = 1.5;
-    private static final double ASPECT_RATIO_TOLERANCE = 0.01;
 
     public Image(String fileName, String contentType, long sizeInBytes, int width, int height) {
         validate(contentType, sizeInBytes, width, height);
@@ -21,8 +21,24 @@ public Image(String fileName, String contentType, long sizeInBytes, int width, i
         this.height = height;
     }
 
+    public String getFileName() {
+        return fileName;
+    }
+    public String getContentType() {
+        return contentType;
+    }
+    public long getSizeInBytes() {
+        return sizeInBytes;
+    }
+    public int getWidth() {
+        return width;
+    }
+    public int getHeight() {
+        return height;
+    }
+
     private void validate(String contentType, long sizeInBytes, int width, int height) {
-        if (sizeInBytes > IMAGE_SIZE_1MB) {
+        if (sizeInBytes > MAX_IMAGE_SIZE) {
             throw new IllegalArgumentException("이미지 크기는 1MB 이하여야 합니다.");
         }
 

src/main/java/nextstep/courses/domain/PaidSession.java

@@ -13,6 +13,14 @@ public PaidSession(Long id, String name, Period period, Image coverImage,
         this.tuitionFee = new TuitionFee(tuitionFee);
     }
 
+    public Capacity getCapacity() {
+        return maxCapacity;
+    }
+
+    public TuitionFee getTuitionFee() {
+        return tuitionFee;
+    }
+
     @Override
     protected void validateRegistration(Long studentId, Payment payment) {
         if (registeredStudents.size() >= maxCapacity.getValue()) {
@@ -27,7 +35,7 @@ protected void validateRegistration(Long studentId, Payment payment) {
         if (!studentId.equals(payment.getNsUserId())) {
             throw new IllegalArgumentException("결제한 사용자와 일치하지 않습니다.");
         }
-        if (this.tuitionFee.getAmount() != payment.getAmount()) {
+        if (!this.tuitionFee.isSameAmount(payment.getAmount())) {
             throw new IllegalArgumentException("결제 금액과 일치하지 않습니다.");
         }
     }

src/main/java/nextstep/courses/domain/Period.java

@@ -1,6 +1,7 @@
 package nextstep.courses.domain;
 
 import java.time.LocalDate;
+import java.util.Objects;
 
 public class Period {
     private final LocalDate startDate;
@@ -21,4 +22,18 @@ public LocalDate getStartDate() {
     public LocalDate getEndDate() {
         return endDate;
     }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (!(o instanceof Period)) return false;
+        Period period = (Period) o;
+        return Objects.equals(startDate, period.startDate) &&
+                Objects.equals(endDate, period.endDate);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(startDate, endDate);
+    }
 }

src/main/java/nextstep/courses/domain/Session.java

@@ -2,22 +2,45 @@
 
 import nextstep.payments.domain.Payment;
 
-import java.util.HashSet;
-import java.util.Set;
 public abstract class Session {
     protected Long id;
     protected String name;
     protected Period period;
     protected Image coverImage;
     protected SessionStatus status;
-    protected Set<Long> registeredStudents = new HashSet<>();
+    protected Students registeredStudents;
 
     public Session(Long id, String name, Period period, Image coverImage, SessionStatus status) {
         this.id = id;
         this.name = name;
         this.period = period;
         this.coverImage = coverImage;
         this.status = status;
+        this.registeredStudents = new Students();
+    }
+
+    public Students getRegisteredStudent() {
+        return registeredStudents;
+    }
+
+    public Image getCoverImage() {
+        return coverImage;
+    }
+
+    public String getName() {
+        return name;
+    }
+
+    public Period getPeriod() {
+        return period;
+    }
+
+    public SessionStatus getStatus() {
+        return status;
+    }
+
+    public Long getId() {
+        return id;
     }
 
     public void register(Long studentId, Payment payment) {
@@ -26,11 +49,7 @@ public void register(Long studentId, Payment payment) {
         }
 
         validateRegistration(studentId, payment);
-        registeredStudents.add(studentId);
-    }
-
-    public boolean isRegistered(Long studentId) {
-        return registeredStudents.contains(studentId);
+        registeredStudents.addStudent(new Student(studentId));
     }
 
     protected abstract void validateRegistration(Long studentId, Payment payment);

src/main/java/nextstep/courses/domain/Student.java

@@ -0,0 +1,30 @@
+package nextstep.courses.domain;
+
+public class Student {
+    Long id;
+    public Student() {
+    }
+    public Student(Long id) {
+        this.id = id;
+    }
+
+    public Long getId() {
+        return id;
+    }
+
+    public boolean contains(Long studentId) {
+        return id.equals(studentId);
+    }
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (o == null || getClass() != o.getClass()) return false;
+        Student student = (Student) o;
+        return id != null && id.equals(student.id);
+    }
+
+    @Override
+    public int hashCode() {
+        return id != null ? id.hashCode() : 0;
+    }
+}

src/main/java/nextstep/courses/domain/Students.java

@@ -0,0 +1,28 @@
+package nextstep.courses.domain;
+
+import java.util.HashSet;
+import java.util.Set;
+
+public class Students {
+    private Set<Student> students = new HashSet<>();
+
+    public void addStudent(Student student) {
+        if (contains(student)) {
+            throw new IllegalArgumentException("이미 등록된 학생입니다.");
+        }
+        students.add(student);
+    }
+
+    public void removeStudent(Student student) {
+        students.remove(student);
+    }
+
+    public boolean contains(Student student) {
+        return students.stream()
+                .anyMatch(s -> s.equals(student));
+    }
+
+    public int size() {
+        return students.size();
+    }
+}

src/main/java/nextstep/courses/domain/TuitionFee.java

@@ -1,5 +1,7 @@
 package nextstep.courses.domain;
 
+import java.util.Objects;
+
 public class TuitionFee {
     private final int amount;
 
@@ -10,7 +12,20 @@ public TuitionFee(int amount) {
         this.amount = amount;
     }
 
-    public int getAmount() {
-        return amount;
+    public boolean isSameAmount(int amount) {
+        return this.amount == amount;
+    }
+
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) return true;
+        if (!(o instanceof TuitionFee)) return false;
+        TuitionFee that = (TuitionFee) o;
+        return amount == that.amount;
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(amount);
     }
 }

src/main/java/nextstep/courses/infrastructure/JdbcSessionRepository.java

@@ -0,0 +1,118 @@
+package nextstep.courses.infrastructure;
+
+import nextstep.courses.domain.Image;
+import nextstep.courses.domain.Period;
+import nextstep.courses.domain.Session;
+import nextstep.courses.domain.SessionStatus;
+import nextstep.courses.domain.PaidSession;
+import nextstep.courses.domain.FreeSession;
+
+import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.jdbc.core.RowMapper;
+import org.springframework.jdbc.core.simple.SimpleJdbcInsert;
+import org.springframework.stereotype.Repository;
+
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+@Repository
+public class JdbcSessionRepository {
+    private final JdbcTemplate jdbcTemplate;
+
+    public JdbcSessionRepository(JdbcTemplate jdbcTemplate) {
+        this.jdbcTemplate = jdbcTemplate;
+    }
+
+    public void save(Session session, Long courseId) {
+        Long imageId = null;
+        if (session.getCoverImage() != null) {
+            imageId = insertImage(session.getCoverImage());
+        }
+
+        String sql = "insert into session (name, type, start_date, end_date, status, capacity, tuition_fee, image_id, course_id) " +
+                "values (?, ?, ?, ?, ?, ?, ?, ?, ?)";
+
+        jdbcTemplate.update(sql,
+                session.getName(),
+                session.getType().name(),
+                session.getPeriod().getStartDate(),
+                session.getPeriod().getEndDate(),
+                session.getStatus().name(),
+                session instanceof PaidSession ? ((PaidSession) session).getCapacity() : null,
+                session instanceof PaidSession ? ((PaidSession) session).getTuitionFee() : null,
+                imageId,
+                courseId
+        );
+    }
+
+    public List<Session> findAll() {
+        String sql = "select s.*, i.file_name, i.content_type, i.size_in_bytes, i.width, i.height " +
+                "from session s left join image i on s.image_id = i.id";
+        return jdbcTemplate.query(sql, sessionRowMapper());
+    }
+
+    public Session findById(Long id) {
+        String sql = "select s.*, i.file_name, i.content_type, i.size_in_bytes, i.width, i.height " +
+                "from session s left join image i on s.image_id = i.id where s.id = ?";
+        return jdbcTemplate.queryForObject(sql, sessionRowMapper(), id);
+    }
+
+    public void update(Session session) {
+        String sql = "update session set name = ?, start_date = ?, end_date = ?, status = ?, capacity = ?, tuition_fee = ? where id = ?";
+        jdbcTemplate.update(sql,
+                session.getName(),
+                session.getPeriod().getStartDate(),
+                session.getPeriod().getEndDate(),
+                session.getStatus().name(),
+                session instanceof PaidSession ? ((PaidSession) session).getCapacity() : null,
+                session instanceof PaidSession ? ((PaidSession) session).getTuitionFee() : null,
+                session.getId()
+        );
+    }
+
+    public void deleteById(Long id) {
+        jdbcTemplate.update("delete from session where id = ?", id);
+    }
+
+    private Long insertImage(Image image) {
+        SimpleJdbcInsert insert = new SimpleJdbcInsert(jdbcTemplate)
+                .withTableName("image")
+                .usingGeneratedKeyColumns("id");
+
+        Map<String, Object> params = new HashMap<>();
+        params.put("file_name", image.getFileName());
+        params.put("content_type", image.getContentType());
+        params.put("size_in_bytes", image.getSizeInBytes());
+        params.put("width", image.getWidth());
+        params.put("height", image.getHeight());
+
+        return insert.executeAndReturnKey(params).longValue();
+    }
+
+    private RowMapper<Session> sessionRowMapper() {
+        return (rs, rowNum) -> {
+            SessionStatus status = SessionStatus.valueOf(rs.getString("status"));
+            Period period = new Period(rs.getDate("start_date").toLocalDate(), rs.getDate("end_date").toLocalDate());
+
+            Image image = null;
+            if (rs.getString("file_name") != null) {
+                image = new Image(
+                        rs.getString("file_name"),
+                        rs.getString("content_type"),
+                        rs.getLong("size_in_bytes"),
+                        rs.getInt("width"),
+                        rs.getInt("height")
+                );
+            }
+
+            String type = rs.getString("type");
+            if (type.equals("FREE")) {
+                return new FreeSession(rs.getLong("id"), rs.getString("name"), period, image, status);
+            } else {
+                return new PaidSession(rs.getLong("id"), rs.getString("name"), period, image, status,
+                        rs.getInt("capacity"), rs.getInt("tuition_fee"));
+            }
+        };
+    }
+}