로그인 기능 개선

어노테이션 기반으로 개선

• 세션값을 가져오는 부분은 필요할때마다 불러와야한다. 이부분을 어노테이션으로 생성해서 개선해보자

@LoginUser 어노테이션 생성

어노테이션생성

@Target(ElementType.PARAMETER)
@Retention(RetentionPolicy.RUNTIME)
public @interface LoginUser {
}

1. @Target(ElementType.PARAMETER)
   • 이 어노테이션이 생성될 수 있는 위치를 지정
   • PARAMETER로 지정했으니 메소드의 파라미터로 선언된 객체에서만 사용할 수 있음
   • 이 외에도 클래스 선언문에 쓸 수 있는 TYPE 등이 있음
2. @interface
   • 이 파일을 어노테이션 클래스로 지정
   • LoginUser라는 이름을 가진 어노테이션이 생성되었다고 보면 됨

LoginUserArgumentResolver를 생성

• HandlerMethodArgumentResolver 인터페이스를 구현한 클래스
  • 조건에 맞는 경우 메소드가 있다면 HandlerMethodArgumentResolver의 구현체가 지정한 값으로 해당 메소드의 파라미터로 넘길 수 있음

@RequiredArgsConstructor
@Component
public class LoginUserArgumentResolver  implements HandlerMethodArgumentResolver {

    private final HttpSession httpSession;
    
    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        boolean isLoginUserAnnotation = parameter.getParameterAnnotation(LoginUser.class) != null;

        boolean isUserClass = SessionUser.class.equals(parameter.getParameterType());

        return isLoginUserAnnotation && isUserClass;
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
        return httpSession.getAttribute("user");
    }
}

1. supportsParameter()
   • 컨트롤러 메서드의 특정 파라미터를 지원하는지 판단
   • 파라미터에 @Loginuser 어노테이션이 붙어 있고, 파라미터 클래스 타입이 SessionUser.class인 경우 true를 반환
2. resolveArgument()
   • 파라미터에 전달할 객체를 생성
   • 여기서는 세션에서 객체를 가져옴

 

스프링에서 인식될 수 있도록 WebMvcConfigurer에 추가

WebConfig 경로

@RequiredArgsConstructor
@Configuration
public class WebConfig  implements WebMvcConfigurer {
    private final LoginUserArgumentResolver loginUserArgumentResolver;

    @Override
    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) {
        resolvers.add(loginUserArgumentResolver);
    }
}

• WebMvcConfigurer의 addArgumentResolvers()를 통해 추가해야함/ 다른 HandlerMethodArgumentResolver가 필요하다면 같은 방식으로 추가하면 됨!

 

반복되는 부분 개선(@LoginUser)

• 이제 어느 컨트롤러든지 @LoginUser만 사용하면 세션 정보를 가져올 수 있음

• reference

스프링부트와 AWS로 혼자 구현하는 웹서비스