Skip to content

Commit 00292d2

Browse files
fix comment and post bug for user
1 parent 0ec6db9 commit 00292d2

File tree

2 files changed

+16
-4
lines changed

2 files changed

+16
-4
lines changed

internal/repository/comment.go

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -79,13 +79,19 @@ func (r *commentRepository) CountLikes(commentID uint) (int64, error) {
7979
}
8080

8181
func (r *commentRepository) ListCommentLikes(commentID uint) ([]models.CommentLikeResponse, error) {
82-
var likes []models.CommentLikeResponse
82+
var likes []models.CommentLike
8383
if err := r.db.
84+
Preload("User").
8485
Where("comment_id = ?", commentID).
8586
Find(&likes).Error; err != nil {
8687
return nil, err
8788
}
88-
return likes, nil
89+
90+
res := make([]models.CommentLikeResponse, 0, len(likes))
91+
for _, l := range likes {
92+
res = append(res, l.ToResponse())
93+
}
94+
return res, nil
8995
}
9096

9197
func (r *commentRepository) HasUserLiked(userID, commentID uint) (bool, error) {

internal/repository/post.go

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,13 +100,19 @@ func (r *postRepository) CountLikes(postID uint) (int64, error) {
100100
}
101101

102102
func (r *postRepository) ListLikesByPostID(postID uint) ([]models.PostLikeResponse, error) {
103-
var likes []models.PostLikeResponse
103+
var likes []models.PostLike
104104
if err := r.db.
105+
Preload("User").
105106
Where("post_id = ?", postID).
106107
Find(&likes).Error; err != nil {
107108
return nil, err
108109
}
109-
return likes, nil
110+
111+
res := make([]models.PostLikeResponse, 0, len(likes))
112+
for _, l := range likes {
113+
res = append(res, l.ToResponse())
114+
}
115+
return res, nil
110116
}
111117

112118
func (r *postRepository) HasUserLiked(userID, postID uint) (bool, error) {

0 commit comments

Comments
 (0)