Springmvc controller custom parameter injection

Declaration annotation

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

 

Declare the generation of custom parameter interceptors

@ Slf4j
@Component
public class ArgumentResolverInterceptor implements HandlerInterceptor {

    @Autowired
    private AppUserService appUserService;

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object o) throws Exception {
        log.info("ArgumentResolverInterceptor.uri={}", request.getRequestURI());

        String token = request.getHeader("app_session");
        log.info("app_session={}", token);
       
        // Real environment, token parsing
        if (StringUtil.isNotNull(token)){
            UserProfile userProfile = UamBizApi.getUserProfile(request);
            if (null == userProfile) {
                AppCommonUtil.forbidAccess(response, AjaxResponse.INVALID_TOKEN, "Access is forbidden, token has expired");
                return false;
            }
            log.info("ArgumentResolverInterceptor.userProfile={}, {}, {}", userProfile.getUid(), userProfile.getUsername(), userProfile.getPhone());
            String userId = userProfile.getUid();
            // Put it in the request for easy access by @CurrentUser
            CurrentUserInfo currentUser = appUserService.getOne(userId);
            log.info("ArgumentResolverInterceptor.currentUser={}", currentUser);
            request.setAttribute("currentUser", currentUser);
            log.info("ArgumentResolverInterceptor process complete"); return true;
        }
        return false;
    }

    @Override
    public void postHandle(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, ModelAndView modelAndView) throws Exception {}

    @Override
    public void afterCompletion(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, Object o, Exception e) throws Exception {}
}

 

Declare custom parameter parser

@ Slf4j
public class CurrentUserArgumentResolver implements HandlerMethodArgumentResolver {

    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        return parameter.hasParameterAnnotation(CurrentUser.class);
    }

    @Override
    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer modelAndView, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
        Object currentUserInfo = webRequest.getAttribute("currentUser", 0);
        if(null == currentUserInfo) log.warn("---------------User is not logged in---------------");
        return currentUserInfo;
    }

}

 

Let SpringMVC load our definition parameter parser

@Configuration
public class AppWebMvcConfigurationSupport extends WebMvcConfigurationSupport {

    /**
     * Controller method parameter injection
     * @param argumentResolvers
     */
    @Override
    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
        argumentResolvers.add(new CurrentUserArgumentResolver());
    }

    /**
     * Register interceptor
     * @param registry
     */
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry
                .addInterceptor(new ArgumentResolverInterceptor())
                .addPathPatterns("/**");
    }

}

 

use in controller

@PostMapping("/car/save")
public AjaxResponse save(@CurrentUser CurrentUserInfo currentUser, @RequestBody CarSaveReq req) {

}