UserSesseion 재로드하기

최초 소셜 로그인 시, 닉네임을 입력받는 기능을 구현하려 했다.

만약 닉네임이 있다면, 닉네임을 입력받는 폼을 띄우지 않는다.

@PutMapping("/users/update")
public RedirectView updateUserInfo(@LoginUser SessionUser sessionUser, SetNicknameForm nicknameForm, BindingResult result) {

    String nickname = nicknameForm.getNickname();

    if (result.hasErrors()) {
        System.out.println("form error");
        return new RedirectView("/home");
    }

    userService.updateUserInfo(sessionUser.getId(), nickname);
    sessionUser.updateNickname(nickname);

    userService.reloadSessionUserInfo("user", sessionUser);

    return new RedirectView("/home");
}

위와 같이 닉네임을 입력받아 userRepository에 닉네임을 저장하고 sessionUser에도 닉네임을 설정했다.DB에도 저장된 것을 확인했다.

그런데 홈화면에 접속하면 계속 닉네임을 입력받는 창이 떴다..

 

문제는 HttpSession에 있었다. 사용자의 정보를 @LoginUser SessionUser sessionUser 파라미터로 받아오는데 SessionUser는 HttpSession에 저장되어 있다. 위 코드에선 변경된 sessionUser 정보를 HttpSession에 반영해주지 않아, 기존의 닉네임이 없는 사용자의 세션 정보가 조회된 것이 문제였다. 

 

따라서 HttpSession에서 해당 유저를 삭제(remove)한 뒤, 다시 유저를 설정(set)하는 로직을 추가했고, 기능이 정상적으로 작동하는 것을 확인할 수 있었다.

public class UserController {
    @PutMapping("/users/update")
    public RedirectView updateUserInfo(@LoginUser SessionUser sessionUser, SetNicknameForm nicknameForm, BindingResult result) {

        String nickname = nicknameForm.getNickname();

        if (result.hasErrors()) {
            System.out.println("form error");
            return new RedirectView("/home");
        }

        userService.updateUserInfo(sessionUser.getId(), nickname);
        sessionUser.updateNickname(nickname);

        userService.reloadSessionUserInfo("user", sessionUser);

        return new RedirectView("/home");
    }
}

public class UserService {
    public void reloadSessionUserInfo(String user, SessionUser sessionUser) {
    	// HttpSession의 유저 정보 재로딩
        httpSession.removeAttribute("user");
        httpSession.setAttribute("user", sessionUser);
    }
}