[Spring] 싱글톤 패턴의 주의점

스프링 시큐리티로 코드 작성하던 중에, 코드에 문제점이 있었다는 것을 발견했다. (정확히는 발견해주신 것)

문제가 되는 코드를 보자.

@Service
@RequiredArgsConstructor
public class LoadUserService {

    private final RestTemplate restTemplate = new RestTemplate();

    private SocialLoadStrategy socialLoadStrategy;//추상 클래스, 로그인을 진행하는 사이트레 따라 달라짐


    public OAuth2UserDetails getOAuth2UserDetails(AccessTokenSocialTypeToken authentication)  {

        SocialType socialType = authentication.getSocialType();

        setSocialLoadStrategy(socialType);//SocialLoadStrategy 설정

        String socialPk = socialLoadStrategy.getSocialPk(authentication.getAccessToken());//PK 가져오기

        return OAuth2UserDetails.builder() //PK와 SocialType을 통해 회원 생성
                .socialId(socialPk)
                .socialType(socialType)
                .build();
    }

    private void setSocialLoadStrategy(SocialType socialType) {
        this.socialLoadStrategy = switch (socialType){

            case KAKAO -> new KakaoLoadStrategy();
            case GOOGLE ->  new GoogleLoadStrategy();
            case NAVER ->  new NaverLoadStrategy();
            default -> throw new IllegalArgumentException("지원하지 않는 로그인 형식입니다");
        };
    }


}

위 코드에서 무엇이 문제가 되는지 한번 맞춰보자.

 

 

 

---

SocialLoadStrategy 가 문제였다!

 

무엇이 문제인지 살펴보자

 

LoadUserService는 싱글톤으로 관리된다. 따라서 객체의 인스턴스가 하나만 생성되고 공유된다.

 

현재 코드를 보면 SocialLoadStrategy는 final도 아니고 메서드 블록 안에서 선언된 지역변수도 아니다. 

LoadUserService는 멤버변수이며, setSocialLoadStrategy에 따라 상태가 바뀐다.

즉 멀티 스레드 환경에서 여러 요청이 LoadUserService의 상태를 바꿔버릴 수 있는 것이다.

따라서 위의 코드는 다음과 같이 고쳐야 한다.

 

@Service
@RequiredArgsConstructor
public class LoadUserService {

    private final RestTemplate restTemplate = new RestTemplate();


    public OAuth2UserDetails getOAuth2UserDetails(AccessTokenSocialTypeToken authentication)  {

        SocialType socialType = authentication.getSocialType();

        SocialLoadStrategy socialLoadStrategy = getSocialLoadStrategy(socialType);//SocialLoadStrategy 설정

        String socialPk = socialLoadStrategy.getSocialPk(authentication.getAccessToken());//PK 가져오기

        return OAuth2UserDetails.builder() //PK와 SocialType을 통해 회원 생성
                .socialId(socialPk)
                .socialType(socialType)
                .build();
    }

    private SocialLoadStrategy getSocialLoadStrategy(SocialType socialType) {
        return switch (socialType){

            case KAKAO -> new KakaoLoadStrategy();
            case GOOGLE ->  new GoogleLoadStrategy();
            case NAVER ->  new NaverLoadStrategy();
            default -> throw new IllegalArgumentException("지원하지 않는 로그인 형식입니다");
        };
    }


}