fix: query wihout parent_id all result will come out

2025-01-15 15:16:58 +08:00 · 2025-01-15 15:16:58 +08:00 · efff638ab5
parent 56145d71af
commit efff638ab5
1 changed files with 5 additions and 0 deletions
--- a/pkg/repository/comment.go
+++ b/pkg/repository/comment.go
@ -110,7 +110,12 @@ func (repo *CommentRepository) ListComments(ctx context.Context, req repository.
 	filter := bson.M{}
 	if req.ParentID != nil {
 		filter["parent_comment_id"] = req.ParentID
+	} else {
+		filter["parent_comment_id"] = bson.M{
+			"$in": []interface{}{"", nil},
+		}
 	}
+
 	if req.ReferenceID != nil {
 		filter["reference_id"] = req.ReferenceID
 	}