#3 feat : 회원가입 토큰 추가, 회원가입/로그인 userid 넘기기

Zatch-Team · Apr 4, 2023 · ed0c598 · ed0c598
1 parent 08f27cc
commit ed0c598
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/main/java/com/zatch/zatchserver/controller/UserController.java b/src/main/java/com/zatch/zatchserver/controller/UserController.java
@@ -50,10 +50,18 @@ public ResponseEntity postUser(@RequestBody PostUserReqDto postUserReqDto, HttpS
 
             PostUserResDto postUserResDto = new PostUserResDto(newUser.getName(), newUser.getEmail(), adjectives.get(0) + " " + animals.get(0));
 
-            return new ResponseEntity(DefaultRes.res(StatusCode.OK, ResponseMessage.CREATED_USER, postUserReqDto), HttpStatus.OK);
+            // 토큰
+            String email = postUserReqDto.getEmail();
+            String userId = userService.getUserId(email);
+            String accessToken = authService.issueAccessToken(Long.valueOf(userId));
+            response.addHeader("ACCESS_TOKEN", accessToken);
+            String token = userService.token(Long.valueOf(userId), accessToken);
+
+            return new ResponseEntity(DefaultRes.res(StatusCode.OK, ResponseMessage.CREATED_USER, postUserReqDto+" / accessToken : "+token+" / userId : "+userId), HttpStatus.OK);
         }
 
         // 로그인
+        // 토큰
         String email = postUserReqDto.getEmail();
         String userId = userService.getUserId(email);
         String accessToken = authService.issueAccessToken(Long.valueOf(userId));
@@ -62,7 +70,7 @@ public ResponseEntity postUser(@RequestBody PostUserReqDto postUserReqDto, HttpS
 
         System.out.println("token >>>>> "+token);
 
-        return new ResponseEntity(DefaultRes.res(StatusCode.OK, ResponseMessage.LOGIN_SUCCESS, new GetUserReqDto(email)+" / accessToken : "+token), HttpStatus.OK);
+        return new ResponseEntity(DefaultRes.res(StatusCode.OK, ResponseMessage.LOGIN_SUCCESS, new GetUserReqDto(email)+" / accessToken : "+token+" / userId : "+userId), HttpStatus.OK);
     }
 
     @GetMapping("/logout")