ArgumentResolver 활용

https://rlaehddnd0422.tistory.com/61#article-3-1-1----argumentresolver

Spring MVC 기본 기능 - 응답 매핑 @ResponseBody, MessageConverter, 요청 매핑 핸들러 어댑터 구조

 

ArgumentResolver를 활용해 세션을 가진 회원만 login 홈 화면으로 이동,

                                               세션이 없는 회원은 홈 화면으로 이동하도록 설정해보겠습니다.

 

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

@Login 어노테이션을 사용하기 위해 어노테이션 생성  

 

@GetMapping("/")
public String homeLoginV3ArgumentResolver(@Login Member loginMember, Model model)
{
    // 세션에 회원 데이터가 없으면 home으로 이동
    if(loginMember==null)
    {
        return "home";
    }

    // 세션이 유지되면 로그인으로 이동
    model.addAttribute("member",loginMember);
    return "loginHome";
}

컨트롤러의 @Login Member loginMember 파라미터를 처리하기 위해 ArgumentResolver를 추가했습니다.

@Slf4j
public class LoginMemberArgumentResolver implements HandlerMethodArgumentResolver {
    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        log.info("supportsParameter 실행");

        boolean hasLoginAnnotation = parameter.hasParameterAnnotation(Login.class);
        boolean hasMemberType = Member.class.isAssignableFrom(parameter.getParameterType());

        // @Login 어노테이션이고 && 타입이 Member여야 resolveArguemnt를 사용 가능
        return hasLoginAnnotation && hasMemberType;
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer,
                                  NativeWebRequest webRequest, WebDataBinderFactory binderFactory)
            throws Exception {
        log.info("resolverArgument 실행");

        HttpServletRequest request = (HttpServletRequest) webRequest.getNativeRequest();
        HttpSession session = request.getSession();

        if(session==null || session.getAttribute(SessionConst.LOGIN_MEMBER)==null)
        {
            return null;
        }

        Object member = session.getAttribute(SessionConst.LOGIN_MEMBER);

        return member;
    }
}

• ArgumentResolver를 추가하기 위해서 HandlerMethodArgumentResolver 인터페이스를 구현해야 합니다!
• boolean supportsParameter(MethodParameter parameter) : parameter에 어노테이션 @Login이 붙어있고, Member 타입인 경우 이 ArgumentResolver를 사용합니다. 설정한 조건이 true인 경우에만 resolverArgument 메소드 사용 가능.
• resolveArgument(...) : 어떤 조건의 Object를 리턴할 지 작성. 이 코드에서는 로그인한 회원에 대한 세션이 있으면 해당 회원을 리턴합니다.

@Configuration
public class WebConfig implements WebMvcConfigurer {
    
    @Override
    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) {
        resolvers.add(new LoginMemberArgumentResolver());
    }
}

마지막으로 만들었던 ArgumentResolver를 사용하기 위해서는 WebMvcConfigurer를 상속받아 ArgumentResolver를 addArgumentResolvers() 메소드를 통해 반드시 등록해 주어야합니다.