fix :: add user id to login response

dnd-side-project · Sep 26, 2024 · 06599ed · 06599ed
1 parent 837e452
commit 06599ed
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/main/java/com/dnd/spaced/global/security/dto/response/LoginResponse.java b/src/main/java/com/dnd/spaced/global/security/dto/response/LoginResponse.java
@@ -1,4 +1,4 @@
 package com.dnd.spaced.global.security.dto.response;
 
-public record LoginResponse(String accessToken, boolean isSignUp, String role) {
+public record LoginResponse(String accessToken, boolean isSignUp, String role, Long id) {
 }
diff --git a/src/main/java/com/dnd/spaced/global/security/handler/OAuth2SuccessHandler.java b/src/main/java/com/dnd/spaced/global/security/handler/OAuth2SuccessHandler.java
@@ -79,11 +79,11 @@ private void login(HttpServletResponse response, Account account, boolean isSign
         String accessToken = tokenEncoder.encode(LocalDateTime.now(), TokenType.ACCESS, claims);
         String refreshToken = tokenEncoder.encode(LocalDateTime.now(), TokenType.REFRESH, claims);
 
-        writeResponse(response, accessToken, isSignUp, account.getRole().name());
+        writeResponse(response, accessToken, isSignUp, account.getRole().name(), account.getId());
         createRefreshTokenCookie(response, refreshToken);
     }
 
-    private void writeResponse(HttpServletResponse response, String accessToken, boolean isSignUp, String role) {
+    private void writeResponse(HttpServletResponse response, String accessToken, boolean isSignUp, String role, Long id) {
         response.setContentType(MediaType.APPLICATION_JSON_VALUE);
         response.setCharacterEncoding(StandardCharsets.UTF_8.name());
 
@@ -92,12 +92,13 @@ private void writeResponse(HttpServletResponse response, String accessToken, boo
                     .queryParam("accessToken", accessToken)
                     .queryParam("isSignUp", isSignUp)
                     .queryParam("role", role)
+                    .queryParam("id", id)
                     .build().toUriString();
 
             response.sendRedirect(redirectUrl);
             PrintWriter writer = response.getWriter();
 
-            writer.println(objectMapper.writeValueAsString(new LoginResponse(accessToken, isSignUp, role)));
+            writer.println(objectMapper.writeValueAsString(new LoginResponse(accessToken, isSignUp, role, id)));
             writer.flush();
         } catch (IOException e) {
             throw new InvalidResponseWriteException();