[FIX] 답글 좋아요 수 반영

Author: SeoYeomm

crud/comment.py

@@ -1,5 +1,5 @@
 from sqlalchemy.orm import Session
-from models.comment import Comment
+from models.comment import Comment, CommentLike
 from schemas.comment import CommentCreate
 from datetime import datetime
 from typing import List
@@ -22,23 +22,33 @@ def create_comment(db: Session, user_id: int, discussion_id: int, comment_data:
 def add_like_to_comment(db: Session, comment_id: int, user_id: int):
     comment = db.query(Comment).filter(Comment.comment_id == comment_id).first()
 
-    if not comment:
-        return None  # 댓글이 존재하지 않음
-
-    # 좋아요 토글: 0이면 +1, 1 이상이면 -1
-    if comment.like == 0:
-        comment.like += 1  # 좋아요 추가
+    # 사용자가 이미 좋아요를 눌렀는지 확인
+    existing_like = db.query(CommentLike).filter(
+        CommentLike.comment_id == comment_id,
+        CommentLike.user_id == user_id
+    ).first()
+
+    if existing_like:
+        # 이미 눌렀다면 삭제 (좋아요 취소)
+        db.delete(existing_like)
+        comment.like -= 1
+        liked = False
     else:
-        comment.like -= 1  # 좋아요 취소
+        # 좋아요 추가
+        new_like = CommentLike(comment_id=comment_id, user_id=user_id)
+        db.add(new_like)
+        comment.like += 1
+        liked = True
 
     db.commit()
     db.refresh(comment)
 
-    return comment
+    return {"comment_id": comment_id, "like": comment.like, "liked": liked}
 
 # 사용자가 좋아요 누른 답글 조회
 def get_liked_comments_by_user(db: Session, user_id: int):
-    return db.query(Comment).filter(Comment.user_id == user_id, Comment.like > 0).all()
+    return db.query(Comment).join(CommentLike, Comment.comment_id == CommentLike.comment_id)\
+        .filter(CommentLike.user_id == user_id).all()
 
 # 토론에 대한 모든 답글 조회
 def get_comments_by_discussion_id(db: Session, discussion_id: int) -> List[Comment]:

models/comment.py

@@ -12,3 +12,10 @@ class Comment(Base):
     like = Column(Integer, default=0, nullable=False)
     created_date = Column(DateTime, server_default=func.now(), nullable=False)
     updated_date = Column(DateTime, onupdate=func.now(), nullable=True)
+
+class CommentLike(Base):
+    __tablename__ = "comment_likes"
+
+    comment_likes_id = Column(BigInteger, primary_key=True, autoincrement=True, nullable=False)
+    comment_id = Column(BigInteger, ForeignKey("comments.comment_id", ondelete="CASCADE"), nullable=False)
+    user_id = Column(BigInteger, ForeignKey("users.user_id", ondelete="CASCADE"), nullable=False)