Ref: 엔티티 개선

Author: joyewon0705

src/main/java/com/back/domain/board/comment/entity/Comment.java

@@ -9,63 +9,80 @@
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.Objects;
 
 @Entity
 @Getter
 @NoArgsConstructor
 public class Comment extends BaseEntity {
     @ManyToOne(fetch = FetchType.LAZY)
-    @JoinColumn(name = "post_id")
+    @JoinColumn(name = "post_id", nullable = false)
     private Post post;
 
     @ManyToOne(fetch = FetchType.LAZY)
-    @JoinColumn(name = "user_id")
+    @JoinColumn(name = "user_id", nullable = false)
     private User user;
 
+    @Column(nullable = false, columnDefinition = "TEXT")
     private String content;
 
-    // TODO: 추후 CommentRepositoryImpl#getCommentsByPostId 로직 개선 필요, ERD에도 반영할 것
     @Column(nullable = false)
     private Long likeCount = 0L;
 
-    // 해당 댓글의 부모 댓글
     @ManyToOne(fetch = FetchType.LAZY)
     @JoinColumn(name = "parent_comment_id")
     private Comment parent;
 
-    // 해당 댓글에 달린 대댓글 목록
     @OneToMany(mappedBy = "parent", cascade = CascadeType.ALL, orphanRemoval = true)
     private List<Comment> children = new ArrayList<>();
 
     @OneToMany(mappedBy = "comment", cascade = CascadeType.ALL, orphanRemoval = true)
     private List<CommentLike> commentLikes = new ArrayList<>();
 
     // -------------------- 생성자 --------------------
-    public Comment(Post post, User user, String content) {
-        this.post = post;
-        this.user = user;
-        this.content = content;
-    }
-
     public Comment(Post post, User user, String content, Comment parent) {
         this.post = post;
         this.user = user;
         this.content = content;
         this.parent = parent;
+        post.addComment(this);
+        user.addComment(this);
+    }
+
+    // -------------------- 정적 팩토리 메서드 --------------------
+    /** 루트 댓글 생성 */
+    public static Comment createRoot(Post post, User user, String content) {
+        return new Comment(post, user, content, null);
+    }
+
+    /** 대댓글 생성 */
+    public static Comment createChild(Post post, User user, String content, Comment parent) {
+        Comment comment = new Comment(post, user, content, parent);
+        parent.getChildren().add(comment);
+        return comment;
+    }
+
+    // -------------------- 연관관계 편의 메서드 --------------------
+    public void addLike(CommentLike like) {
+        this.commentLikes.add(like);
+    }
+
+    public void removeLike(CommentLike like) {
+        this.commentLikes.remove(like);
     }
 
     // -------------------- 비즈니스 메서드 --------------------
-    // 댓글 업데이트
+    /** 댓글 내용 수정 */
     public void update(String content) {
         this.content = content;
     }
 
-    // 좋아요 수 증가
+    /** 좋아요 수 증가 */
     public void increaseLikeCount() {
         this.likeCount++;
     }
 
-    // 좋아요 수 감소
+    /** 좋아요 수 감소 */
     public void decreaseLikeCount() {
         if (this.likeCount > 0) {
             this.likeCount--;

src/main/java/com/back/domain/board/comment/entity/CommentLike.java

@@ -2,24 +2,31 @@
 
 import com.back.domain.user.entity.User;
 import com.back.global.entity.BaseEntity;
-import jakarta.persistence.Entity;
-import jakarta.persistence.FetchType;
-import jakarta.persistence.JoinColumn;
-import jakarta.persistence.ManyToOne;
+import jakarta.persistence.*;
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
 
 @Entity
 @Getter
 @NoArgsConstructor
-@AllArgsConstructor
+@Table(
+        uniqueConstraints = @UniqueConstraint(columnNames = {"comment_id", "user_id"})
+)
 public class CommentLike extends BaseEntity {
     @ManyToOne(fetch = FetchType.LAZY)
-    @JoinColumn(name = "comment_id")
+    @JoinColumn(name = "comment_id", nullable = false)
     private Comment comment;
 
     @ManyToOne(fetch = FetchType.LAZY)
-    @JoinColumn(name = "user_id")
+    @JoinColumn(name = "user_id", nullable = false)
     private User user;
+
+    // -------------------- 생성자 --------------------
+    public CommentLike(Comment comment, User user) {
+        this.comment = comment;
+        this.user = user;
+        comment.addLike(this);
+        user.addCommentLike(this);
+    }
 }

src/main/java/com/back/domain/board/comment/service/CommentLikeService.java

@@ -70,6 +70,10 @@ public CommentLikeResponse cancelLikeComment(Long commentId, Long userId) {
         CommentLike commentLike = commentLikeRepository.findByUserIdAndCommentId(userId, commentId)
                 .orElseThrow(() -> new CustomException(ErrorCode.COMMENT_LIKE_NOT_FOUND));
 
+        // 연관관계 제거
+        comment.removeLike(commentLike);
+        user.removeCommentLike(commentLike);
+
         // CommentLike 삭제
         commentLikeRepository.delete(commentLike);
 

src/main/java/com/back/domain/board/comment/service/CommentService.java

@@ -38,12 +38,15 @@ public class CommentService {
     private final PostRepository postRepository;
     private final ApplicationEventPublisher eventPublisher;
 
+    // TODO: 연관관계 고려, 메서드 명, 중복 코드 제거, 주석 통일
+    // TODO: comment 끝나면 post도 해야 함.. entity > DTO > Repo > Service > Controller > Docs 순으로..
     /**
      * 댓글 생성 서비스
      * 1. User 조회
      * 2. Post 조회
-     * 3. Comment 생성
-     * 4. Comment 저장 및 CommentResponse 반환
+     * 3. Comment 생성 및 저장
+     * 4. 댓글 작성 이벤트 발행
+     * 5. CommentResponse 반환
      */
     public CommentResponse createComment(Long postId, CommentRequest request, Long userId) {
         // User 조회
@@ -54,10 +57,11 @@ public CommentResponse createComment(Long postId, CommentRequest request, Long u
         Post post = postRepository.findById(postId)
                 .orElseThrow(() -> new CustomException(ErrorCode.POST_NOT_FOUND));
 
-        // Comment 생성
-        Comment comment = new Comment(post, user, request.content());
+        // CommentCount 증가
+        post.increaseCommentCount();
 
-        // Comment 저장 및 응답 반환
+        // Comment 생성 및 저장
+        Comment comment = Comment.createRoot(post, user, request.content());
         commentRepository.save(comment);
 
         // 댓글 작성 이벤트 발행
@@ -164,6 +168,10 @@ public CommentResponse updateComment(Long postId, Long commentId, CommentRequest
      * 4. Comment 삭제
      */
     public void deleteComment(Long postId, Long commentId, Long userId) {
+        // User 조회
+        User user = userRepository.findById(userId)
+                .orElseThrow(() -> new CustomException(ErrorCode.USER_NOT_FOUND));
+
         // Post 조회
         Post post = postRepository.findById(postId)
                 .orElseThrow(() -> new CustomException(ErrorCode.POST_NOT_FOUND));
@@ -177,6 +185,11 @@ public void deleteComment(Long postId, Long commentId, Long userId) {
             throw new CustomException(ErrorCode.COMMENT_NO_PERMISSION);
         }
 
+        // 연관관계 제거
+        post.removeComment(comment);
+        user.removeComment(comment);
+
+        // Comment 삭제
         commentRepository.delete(comment);
     }
 

src/main/java/com/back/domain/board/post/entity/Post.java

@@ -9,17 +9,16 @@
 
 import java.util.ArrayList;
 import java.util.List;
-import java.util.regex.Matcher;
-import java.util.regex.Pattern;
 
 @Entity
 @Getter
 @NoArgsConstructor
 public class Post extends BaseEntity {
     @ManyToOne(fetch = FetchType.LAZY)
-    @JoinColumn(name = "user_id")
+    @JoinColumn(name = "user_id", nullable = false)
     private User user;
 
+    @Column(nullable = false)
     private String title;
 
     @Column(nullable = false, columnDefinition = "TEXT")
@@ -28,7 +27,6 @@ public class Post extends BaseEntity {
     @Column(length = 500)
     private String thumbnailUrl;
 
-    // TODO: 추후 PostRepositoryImpl#searchPosts 로직 개선 필요, ERD에도 반영할 것
     @Column(nullable = false)
     private Long likeCount = 0L;
 
@@ -51,73 +49,94 @@ public class Post extends BaseEntity {
     private List<Comment> comments = new ArrayList<>();
 
     // -------------------- 생성자 --------------------
-    public Post(User user, String title, String content) {
-        this.user = user;
-        this.title = title;
-        this.content = content;
-        this.thumbnailUrl = null;
-    }
-
     public Post(User user, String title, String content, String thumbnailUrl) {
         this.user = user;
         this.title = title;
         this.content = content;
         this.thumbnailUrl = thumbnailUrl;
+        user.addPost(this);
+    }
+
+    // -------------------- 연관관계 편의 메서드 --------------------
+    public void addPostCategoryMapping(PostCategoryMapping mapping) {
+        this.postCategoryMappings.add(mapping);
+    }
+
+    public void addLike(PostLike like) {
+        this.postLikes.add(like);
+    }
+
+    public void removeLike(PostLike like) {
+        this.postLikes.remove(like);
+    }
+
+    public void addBookmark(PostBookmark bookmark) {
+        this.postBookmarks.add(bookmark);
+    }
+
+    public void removeBookmark(PostBookmark bookmark) {
+        this.postBookmarks.remove(bookmark);
+    }
+
+    public void addComment(Comment comment) {
+        this.comments.add(comment);
+    }
+
+    public void removeComment(Comment comment) {
+        this.comments.remove(comment);
     }
 
     // -------------------- 비즈니스 메서드 --------------------
-    // 게시글 업데이트
+    /** 게시글 내용 수정 */
     public void update(String title, String content) {
         this.title = title;
         this.content = content;
     }
 
-    // 카테고리 업데이트
+    /** 카테고리 일괄 업데이트 */
     public void updateCategories(List<PostCategory> categories) {
         this.postCategoryMappings.clear();
-        categories.forEach(category ->
-                this.postCategoryMappings.add(new PostCategoryMapping(this, category))
-        );
+        categories.forEach(category -> new PostCategoryMapping(this, category));
     }
 
-    // 좋아요 수 증가
+    /** 좋아요 수 증가 */
     public void increaseLikeCount() {
         this.likeCount++;
     }
 
-    // 좋아요 수 감소
+    /** 좋아요 수 감소 */
     public void decreaseLikeCount() {
         if (this.likeCount > 0) {
             this.likeCount--;
         }
     }
 
-    // 북마크 수 증가
+    /** 북마크 수 증가 */
     public void increaseBookmarkCount() {
         this.bookmarkCount++;
     }
 
-    // 북마크 수 감소
+    /** 북마크 수 감소 */
     public void decreaseBookmarkCount() {
         if (this.bookmarkCount > 0) {
             this.bookmarkCount--;
         }
     }
 
-    // 댓글 수 증가
+    /** 댓글 수 증가 */
     public void increaseCommentCount() {
         this.commentCount++;
     }
 
-    // 댓글 수 감소
+    /** 댓글 수 감소 */
     public void decreaseCommentCount() {
         if (this.commentCount > 0) {
             this.commentCount--;
         }
     }
 
     // -------------------- 헬퍼 메서드 --------------------
-    // 게시글에 연결된 카테고리 목록 조회
+    /** 게시글에 연결된 카테고리 목록 조회 */
     public List<PostCategory> getCategories() {
         return postCategoryMappings.stream()
                 .map(PostCategoryMapping::getCategory)

src/main/java/com/back/domain/board/post/entity/PostBookmark.java

@@ -10,18 +10,23 @@
 @Entity
 @Getter
 @NoArgsConstructor
-@AllArgsConstructor
 @Table(
-        uniqueConstraints = {
-                @UniqueConstraint(columnNames = {"post_id", "user_id"})
-        }
+        uniqueConstraints = @UniqueConstraint(columnNames = {"post_id", "user_id"})
 )
 public class PostBookmark extends BaseEntity {
     @ManyToOne(fetch = FetchType.LAZY)
-    @JoinColumn(name = "post_id")
+    @JoinColumn(name = "post_id", nullable = false)
     private Post post;
 
     @ManyToOne(fetch = FetchType.LAZY)
-    @JoinColumn(name = "user_id")
+    @JoinColumn(name = "user_id", nullable = false)
     private User user;
+
+    // -------------------- 생성자 --------------------
+    public PostBookmark(Post post, User user) {
+        this.post = post;
+        this.user = user;
+        post.addBookmark(this);
+        user.addPostBookmark(this);
+    }
 }