#28 [update]

폴더 리랙토링

src/main/java/com/mpnp/baechelin/bookmark/controller/FolderController.java

@@ -5,6 +5,8 @@
 import com.mpnp.baechelin.bookmark.dto.FolderResponseDto;
 import com.mpnp.baechelin.bookmark.service.FolderService;
 import lombok.RequiredArgsConstructor;
+import org.springframework.security.core.annotation.AuthenticationPrincipal;
+import org.springframework.security.core.userdetails.User;
 import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
@@ -16,33 +18,43 @@ public class FolderController {
     private final FolderService folderService;
 
 
+
     /** 폴더 신규 생성 */
     @PostMapping("/folder")
-    public void folder (@RequestBody FolderRequestDto folderRequestDto){
+    public void folder (@RequestBody FolderRequestDto folderRequestDto,
+                        @AuthenticationPrincipal User user){
 
-        folderService.folder(folderRequestDto);
+        if(user==null){ throw new IllegalArgumentException("해당하는 회원 정보가 없습니다."); }
+        folderService.folder(folderRequestDto, user.getUsername());
     }
 
-    /** 폴더 삭제 -> 삭제 시 안에 담긴 모든 북마크가 삭제됨 */
 
+
+    /** 폴더 삭제 -> 삭제 시 안에 담긴 모든 북마크가 삭제됨 */
     @DeleteMapping("/folder/{folderId}")
-    public void folderDelete (@PathVariable int folderId){
+    public void folderDelete (@PathVariable int folderId) {
+
 
         folderService.folderDelete(folderId);
     }
 
+
+
     /** 폴더 명 변경 */
     @PutMapping("/folderUpdate/{folderId}")
-    public void folderUpdate (@PathVariable int folderId, @RequestParam String newFolderName){
-        System.out.println(newFolderName);
+    public void folderUpdate (@PathVariable int folderId,
+                              @RequestParam String newFolderName){
+
 
         folderService.folderUpdate(folderId, newFolderName);
     }
 
+
+
     /** 폴더 리스트 */
     @PostMapping("/folderList")
-    public List<FolderResponseDto> folderList (int userId){
+    public List<FolderResponseDto> folderList (@AuthenticationPrincipal User user){
 
-        return folderService.folderList(userId);
+        return folderService.folderList(user.getUsername());
     }
 }

src/main/java/com/mpnp/baechelin/bookmark/service/FolderService.java

@@ -22,42 +22,50 @@ public class FolderService {
     private final FolderRepository folderRepository;
     private final UserRepository userRepository;
 
+    /** 폴더 생성 */
     @Transactional
-    public void folder(FolderRequestDto folderRequestDto) {
-
-        Optional<User> user = userRepository.findById(1);
+    public void folder(FolderRequestDto folderRequestDto, String socialId) {
 
+        User user = userRepository.findBySocialId(socialId);
         Folder folder = Folder.builder()
                 .folderName(folderRequestDto.getFolderName())
-                .userId(user.get())
+                .userId(user)
                 .build();
         folderRepository.save(folder);
 
     }
 
+    /** 폴더 삭제 */
     public void folderDelete(int folderId) {
         folderRepository.deleteById(folderId);
     }
 
-
+    /** 폴더 수정 */
     public void folderUpdate(int folderId, String newFolderName) {
-        Optional<Folder> folder = folderRepository.findById(folderId);
 
+
+        Optional<Folder> folder = folderRepository.findById(folderId);
         folder.get().setFolderName(newFolderName);
 
+
         folderRepository.save(folder.get());
     }
 
+
+    /** 폴더 조회 */
     @Transactional(readOnly = true)
-    public List<FolderResponseDto> folderList(int userId) {
-        Optional<User> user = userRepository.findById(userId);
+    public List<FolderResponseDto> folderList(String socialId) {
+
 
+        User user = userRepository.findBySocialId(socialId);
         List<FolderResponseDto> folderResponseDtoList = new ArrayList<>();
 
-        for(Folder obj : user.get().getFolderList()){
+
+        for(Folder obj : user.getFolderList()){
             folderResponseDtoList.add(FolderResponseDto.FolderDtoRes(obj));
         }
 
+
         return folderResponseDtoList;
     }
 }