add count controller

giangle-mfv · Jun 27, 2024 · 5053f1d · 5053f1d
1 parent 12f2487
commit 5053f1d
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/src/main/java/com/springboot/blog/controller/CommentController.java b/src/main/java/com/springboot/blog/controller/CommentController.java
@@ -30,6 +30,11 @@ public List<CommentDto> getCommentsByPostId(@PathVariable(value = "postId") Long
         return commentService.getCommentsByPostId(postId);
     }
 
+    @GetMapping("/posts/{postId}/comments/count")
+    public int countCommentsByPostId(@PathVariable(value = "postId") Long postId){
+        return commentService.countCommentsByPostId(postId);
+    }
+
     @GetMapping("/posts/{postId}/comments/{id}")
     public ResponseEntity<CommentDto> getCommentById(@PathVariable(value = "postId") Long postId,
                                                      @PathVariable(value = "id") Long commentId){

diff --git a/src/main/java/com/springboot/blog/service/CommentService.java b/src/main/java/com/springboot/blog/service/CommentService.java
@@ -9,6 +9,8 @@ public interface CommentService {
 
     List<CommentDto> getCommentsByPostId(long postId);
 
+    int countCommentsByPostId(long postId);
+
     CommentDto getCommentById(Long postId, Long commentId);
 
     CommentDto updateComment(Long postId, long commentId, CommentDto commentRequest);

diff --git a/src/main/java/com/springboot/blog/service/impl/CommentServiceImpl.java b/src/main/java/com/springboot/blog/service/impl/CommentServiceImpl.java
@@ -54,6 +54,11 @@ public List<CommentDto> getCommentsByPostId(long postId) {
         return comments.stream().map(comment -> mapToDTO(comment)).collect(Collectors.toList());
     }
 
+    @Override
+    public int countCommentsByPostId(long postId) {
+        return commentRepository.findByPostId(postId).size();
+    }
+
     @Override
     public CommentDto getCommentById(Long postId, Long commentId) {
         // retrieve post entity by id