feat: 유저 펫 조회 api 추가

Author: whale22

src/main/java/cmf/commitField/domain/noti/noti/controller/ApiV1NotiController.java

@@ -30,7 +30,7 @@
 public class ApiV1NotiController {
     private final NotiService notiService;
     private final UserRepository userRepository;
-    private final NotiWebSocketHandler notiWebSocketHandler;
+//    private final NotiWebSocketHandler notiWebSocketHandler;
     private final ApplicationEventPublisher eventPublisher;
 
     @GetMapping("")

src/main/java/cmf/commitField/domain/pet/controller/PetController.java

@@ -13,8 +13,6 @@
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
-import java.util.List;
-
 @RestController
 @RequiredArgsConstructor
 @RequestMapping("/api/pets")
@@ -47,11 +45,6 @@ public Pet createPet(
         return petService.createPet(name, imageFile, user);
     }
 
-    // 모든 펫 조회
-    @GetMapping
-    public List<Pet> getAllPets() {
-        return petService.getAllPets();
-    }
 
     // 특정 펫 조회
     @GetMapping("/{petId}")

src/main/java/cmf/commitField/domain/pet/controller/UserPetController.java

@@ -1,20 +1,18 @@
 package cmf.commitField.domain.pet.controller;
 
 
-import cmf.commitField.domain.pet.entity.Pet;
+import cmf.commitField.domain.pet.dto.UserPetListDto;
 import cmf.commitField.domain.pet.service.PetService;
 import cmf.commitField.domain.pet.service.UserPetService;
-import cmf.commitField.domain.user.entity.User;
+import cmf.commitField.domain.user.entity.CustomOAuth2User;
 import cmf.commitField.domain.user.service.CustomOAuth2UserService;
-import cmf.commitField.global.globalDto.GlobalResponse;
 import lombok.RequiredArgsConstructor;
+import org.springframework.http.ResponseEntity;
+import org.springframework.security.core.annotation.AuthenticationPrincipal;
 import org.springframework.web.bind.annotation.GetMapping;
-import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.RequestMapping;
 import org.springframework.web.bind.annotation.RestController;
 
-import java.util.List;
-
 @RestController
 @RequestMapping("/user-pets")
 @RequiredArgsConstructor
@@ -24,11 +22,11 @@ public class UserPetController {
     private final CustomOAuth2UserService customOAuth2UserService;
     private final PetService petService;
 
-    // TODO: 기능 확장시 추가 예정
     // 유저의 도감 조회 (보유한 펫 목록)
-    @GetMapping("/collection/{userId}")
-    public GlobalResponse<List<Pet>> getUserPetCollection(@PathVariable Long userId) {
-        User user = customOAuth2UserService.getUserById(userId).orElse(null);
-        return GlobalResponse.success(userPetService.getUserPetCollection(user));
+    @GetMapping("/collection")
+    public ResponseEntity<UserPetListDto> getUserPetCollection(@AuthenticationPrincipal CustomOAuth2User oAuth2User) {
+        String username = oAuth2User.getName();
+        UserPetListDto userPetListDto = petService.getAllPets(username);
+        return ResponseEntity.ok(userPetListDto);
     }
 }

src/main/java/cmf/commitField/domain/pet/dto/UserPetListDto.java

@@ -0,0 +1,17 @@
+package cmf.commitField.domain.pet.dto;
+
+import lombok.AllArgsConstructor;
+import lombok.Builder;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+
+import java.util.Map;
+
+@Builder
+@Getter
+@NoArgsConstructor
+@AllArgsConstructor
+public class UserPetListDto {
+    private String username;
+    private Map<Integer, Integer> petList; // 아이디, 소유 갯수
+}

src/main/java/cmf/commitField/domain/pet/service/PetService.java

@@ -1,6 +1,7 @@
 package cmf.commitField.domain.pet.service;
 
 import cmf.commitField.domain.pet.dto.UserPetDto;
+import cmf.commitField.domain.pet.dto.UserPetListDto;
 import cmf.commitField.domain.pet.entity.Pet;
 import cmf.commitField.domain.pet.entity.PetGrow;
 import cmf.commitField.domain.pet.repository.PetRepository;
@@ -13,9 +14,7 @@
 import org.springframework.web.multipart.MultipartFile;
 
 import java.io.IOException;
-import java.util.List;
-import java.util.Optional;
-import java.util.Random;
+import java.util.*;
 
 @Service
 @RequiredArgsConstructor
@@ -37,9 +36,26 @@ public Pet createPet(String name, MultipartFile imageFile, User user) throws IOE
         return petRepository.save(pet);
     }
 
-    // 모든 펫 조회
-    public List<Pet> getAllPets() {
-        return petRepository.findAll();
+    // 유저가 소유한 펫 조회
+    public UserPetListDto getAllPets(String username) {
+        List<Pet> pets = petRepository.findByUserUsername(username);
+        int[] petsNum = new int[100]; //FIXME: 차후 갯수 수정 필요
+        int max = 0;
+        Map<Integer, Integer> petList = new HashMap<>();
+        for(Pet pet : pets){
+            int type = pet.getType();
+            petsNum[type]++;
+            if(max<type) max = type;
+        }
+
+        for(int i=0;i<=max;i++){
+            petList.put(i, petsNum[i]);
+        }
+
+        return UserPetListDto.builder().
+                username(username).
+                petList(petList).
+                build();
     }
 
     // 특정 펫 조회