feat:add maxPage in getting my posts

PoolC · Apr 11, 2024 · 02ba4b6 · 02ba4b6
1 parent 714667a
commit 02ba4b6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/src/main/java/org/poolc/api/post/controller/PostController.java b/src/main/java/org/poolc/api/post/controller/PostController.java
@@ -52,7 +52,7 @@ public ResponseEntity<GetBoardResponse> viewPostsByBoard(@AuthenticationPrincipa
     }
 
     @GetMapping("/my_posts")
-    public ResponseEntity<List<PostResponse>> viewMyPosts(@AuthenticationPrincipal Member member, @RequestParam int page) {
+    public ResponseEntity<GetBoardResponse> viewMyPosts(@AuthenticationPrincipal Member member, @RequestParam int page) {
         return ResponseEntity.status(HttpStatus.OK).body(postService.findPostsByMember(member, page));
     }
 

diff --git a/src/main/java/org/poolc/api/post/service/PostService.java b/src/main/java/org/poolc/api/post/service/PostService.java
@@ -42,14 +42,16 @@ public Post findPostById(Member member, Long postId) {
     }
 
     @Transactional(readOnly = true)
-    public List<PostResponse> findPostsByMember(Member member, int page) {
+    public GetBoardResponse findPostsByMember(Member member, int page) {
         PageRequest pr = PageRequest.of(page, size,Sort.by("createdAt").descending());
         Page<Post> posts = postRepository.findByMember(member, pr);
         if (posts.getNumberOfElements() == 0) return null;
-        return posts.stream()
-                //.sorted(Comparator.comparing(Post::getCreatedAt).reversed())
-                .map(PostResponse::of)
-                .collect(Collectors.toList());
+        return new GetBoardResponse(
+                posts.getTotalPages(),
+                posts.stream()
+                        .map(GetPostsResponse::of)
+                        .collect(Collectors.toList())
+        );
     }
 
     @Transactional(readOnly = true)