add reflection of favorite works

oven-2023 · Nov 8, 2023 · 39ae8a1 · 39ae8a1
1 parent 30bf9f5
commit 39ae8a1
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/src/main/java/com/oven/server/api/user/dto/request/JoinRequest.java b/src/main/java/com/oven/server/api/user/dto/request/JoinRequest.java
@@ -3,6 +3,8 @@
 import io.swagger.v3.oas.annotations.media.Schema;
 import lombok.Getter;
 
+import java.util.List;
+
 @Getter
 public class JoinRequest {
 
@@ -14,5 +16,7 @@ public class JoinRequest {
 
     @Schema(description = "패스워드", example = "password123@")
     private String password;
+
+    private List<Long> workIdList;
 
 }
diff --git a/src/main/java/com/oven/server/api/user/service/AuthService.java b/src/main/java/com/oven/server/api/user/service/AuthService.java
@@ -1,5 +1,6 @@
 package com.oven.server.api.user.service;
 
+import com.oven.server.api.user.domain.RatingWork;
 import com.oven.server.api.user.domain.RefreshToken;
 import com.oven.server.api.user.domain.User;
 import com.oven.server.api.user.dto.request.IdCheckRequest;
@@ -9,8 +10,11 @@
 import com.oven.server.api.user.dto.response.AccessTokenResponse;
 import com.oven.server.api.user.dto.response.IdCheckResponse;
 import com.oven.server.api.user.dto.response.LoginSuccessResponse;
+import com.oven.server.api.user.repository.RatingWorkRepository;
 import com.oven.server.api.user.repository.RefreshTokenRepository;
 import com.oven.server.api.user.repository.UserRepository;
+import com.oven.server.api.work.domain.Work;
+import com.oven.server.api.work.repository.WorkRepository;
 import com.oven.server.common.exception.BaseException;
 import com.oven.server.common.response.ResponseCode;
 import com.oven.server.config.jwt.JwtTokenProvider;
@@ -22,6 +26,7 @@
 import org.springframework.transaction.annotation.Transactional;
 
 import java.util.Date;
+import java.util.List;
 
 @Service
 @Transactional
@@ -30,9 +35,11 @@
 public class AuthService {
 
     private final UserRepository userRepository;
+    private final WorkRepository workRepository;
     private final PasswordEncoder passwordEncoder;
     private final JwtTokenProvider jwtTokenProvider;
     private final RefreshTokenRepository refreshTokenRepository;
+    private final RatingWorkRepository ratingWorkRepository;
 
     public IdCheckResponse idDuplicateCheck(IdCheckRequest idCheckRequest) throws BaseException {
 
@@ -52,6 +59,13 @@ public void join(JoinRequest joinRequest) throws BaseException {
 
         userRepository.save(newUser);
 
+        List<Long> workIdList = joinRequest.getWorkIdList();
+        for (int i = 0; i < workIdList.size(); i++) {
+            Work work = workRepository.findById(workIdList.get(i)).get();
+            RatingWork ratingWork = new RatingWork(newUser, work, 5F);
+            ratingWorkRepository.save(ratingWork);
+        }
+
     }
 
     public LoginSuccessResponse login(LoginRequest loginRequest) {