feat: comment status, converter 추가 #63

codestates-seb · Feb 20, 2023 · c3c732a · c3c732a
1 parent f203347
commit c3c732a
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 0 deletions.
diff --git a/server/src/main/java/com/main36/pikcha/domain/comment/entity/Comment.java b/server/src/main/java/com/main36/pikcha/domain/comment/entity/Comment.java
@@ -3,10 +3,16 @@
 import com.main36.pikcha.domain.member.entity.Member;
 import com.main36.pikcha.domain.post.entity.Post;
 import com.main36.pikcha.global.audit.Auditable;
+import com.main36.pikcha.global.exception.BusinessLogicException;
+import com.main36.pikcha.global.exception.ExceptionCode;
+import com.main36.pikcha.global.utils.CommentStatusConverter;
 import lombok.*;
+import org.springframework.batch.item.validator.SpringValidator;
 
 import javax.persistence.*;
+import java.net.http.HttpClient;
 import java.util.ArrayList;
+import java.util.Arrays;
 import java.util.List;
 
 @Getter
@@ -29,6 +35,9 @@ public class Comment extends Auditable {
     @JoinColumn(name = "member_id")
     private Member member;
 
+    @Convert(converter = CommentStatusConverter.class)
+    private CommentStatus status = CommentStatus.Alive;
+
     @ManyToOne(fetch = FetchType.LAZY)
     @JoinColumn(name = "post_id")
     private Post post;
@@ -47,4 +56,25 @@ public void updateParent(Comment parent){
     public List<Comment> getChildren(){
         return this.children;
     }
+
+    @Getter
+    public enum CommentStatus {
+        Alive("생존", "1"),
+        Dead("사망", "2");
+
+        private String status;
+        private String code;
+
+        CommentStatus(String status, String code) {
+            this.status = status;
+            this.code = code;
+        }
+
+        public static CommentStatus ofCode(String code) {
+            return Arrays.stream(CommentStatus.values())
+                    .filter(v-> v.getCode().equals(code))
+                    .findAny()
+                    .orElseThrow(()-> new BusinessLogicException(ExceptionCode.COMMENT_STATUS_INVALID));
+        }
+    }
 }
diff --git a/server/src/main/java/com/main36/pikcha/global/exception/ExceptionCode.java b/server/src/main/java/com/main36/pikcha/global/exception/ExceptionCode.java
@@ -14,6 +14,7 @@ public enum ExceptionCode {
     ATTRACTION_IMAGE_EXISTS(409, "Attraction image exists"),
 
     COMMENT_NOT_FOUND(404, "Comment not found"),
+    COMMENT_STATUS_INVALID(404, "Comment status is inValid"),
     COMMENT_EXISTS(409, "Comment exists"),
 
     NOT_AUTHOR(403, "This Member didn't write the Comment"),

diff --git a/server/src/main/java/com/main36/pikcha/global/utils/CommentStatusConverter.java b/server/src/main/java/com/main36/pikcha/global/utils/CommentStatusConverter.java
@@ -0,0 +1,19 @@
+package com.main36.pikcha.global.utils;
+
+import com.main36.pikcha.domain.comment.entity.Comment;
+
+import javax.persistence.AttributeConverter;
+import javax.persistence.Converter;
+
+@Converter
+public class CommentStatusConverter implements AttributeConverter<Comment.CommentStatus, String> {
+    @Override
+    public String convertToDatabaseColumn(Comment.CommentStatus attribute) {
+        return attribute.getCode();
+    }
+
+    @Override
+    public Comment.CommentStatus convertToEntityAttribute(String dbData) {
+        return Comment.CommentStatus.ofCode(dbData);
+    }
+}