fix : user file null

JunKyung-Jo · Jan 8, 2024 · c5040e8 · c5040e8
1 parent 1e8df15
commit c5040e8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/src/main/java/com/investment/findfriend/domain/user/domain/User.java b/src/main/java/com/investment/findfriend/domain/user/domain/User.java
@@ -55,6 +55,7 @@ public class User {
     private List<Feed> feed;
 
     @OneToOne
+    @Setter
     private File file;
 
     public void update(UpdateUserInfoRequest request) {

diff --git a/src/main/java/com/investment/findfriend/domain/user/service/UpdateUserInfoService.java b/src/main/java/com/investment/findfriend/domain/user/service/UpdateUserInfoService.java
@@ -27,13 +27,23 @@ public ResponseEntity<String> execute(UpdateUserInfoRequest request, MultipartFi
         User user = userRepository.findByEmail(jwtUtil.extractEmail(httpServletRequest)).orElseThrow(
                 () -> UserNotFoundException.EXCEPTION
         );
+
         user.update(request);
+
         if (!file.isEmpty()) {
             File userFile = user.getFile();
-            userFile.setPath(fileSaveUtil.save(file));
-            fileRepository.save(userFile);
+
+            if (userFile == null) {
+                user.setFile(File.builder().build());
+            }
+
+            user.getFile().setPath(fileSaveUtil.save(file));
+            fileRepository.save(user.getFile());
         }
+
         userRepository.save(user);
         return ResponseEntity.ok("success");
     }
+
+
 }