#79 fix: Response 값 추가

Animal Search시에 총 검색된 개수 반환
GDG-on-Campus-KNU · Feb 25, 2024 · ebddcf4 · ebddcf4
1 parent eda4e51
commit ebddcf4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/main/java/com/gdsc/pikpet/dto/AnimalSearchDto.java b/src/main/java/com/gdsc/pikpet/dto/AnimalSearchDto.java
@@ -5,6 +5,7 @@
 public record AnimalSearchDto(
     List<AnimalSimpleDto> animals,
     Integer page,
-    int numberOfElements
+    int numberOfElements,
+    int totalElements
 ) {
 }
diff --git a/src/main/java/com/gdsc/pikpet/service/AnimalService.java b/src/main/java/com/gdsc/pikpet/service/AnimalService.java
@@ -56,6 +56,7 @@ public AnimalSearchDto getAnimals(AnimalFilterCriteria animalFilterCriteria, Use
                 pageable
         );
         int numberOfElements = animals.getNumberOfElements();
+        int totalElements = (int) animals.getTotalElements();
         UserAccount userAccount = userAccountService.getUserAccount(userSecurityDto);
         List<AnimalSimpleDto> animalSimpleDtos = animals.getContent().stream()
                 .map(animal -> {
@@ -64,7 +65,7 @@ public AnimalSearchDto getAnimals(AnimalFilterCriteria animalFilterCriteria, Use
                 })
                 .toList();
 
-        AnimalSearchDto response = new AnimalSearchDto(animalSimpleDtos, animalFilterCriteria.page(), numberOfElements);
+        AnimalSearchDto response = new AnimalSearchDto(animalSimpleDtos, animalFilterCriteria.page(), numberOfElements, totalElements);
         return response;
     }