ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

WebMvcConfigurer中addArgumentResolvers方法的使用

2021-11-15 16:01:16  阅读:269  来源: 互联网

标签:return Controller addArgumentResolvers token session 参数 WebMvcConfigurer 方法


在Springboot中的WebMvcConfigurer接口在Web开发中经常被使用,例如配置拦截器、配置ViewController、配置Cors跨域等。本文主要讲解另一个方法:addArgumentResolvers()在实例中的应用。

一、方法作用

该方法可以用在对于Controller中方法参数传入之前对该参数进行处理。然后将处理好的参数在传给Controller中的方法。

官方API文档解释:添加解析器以支持自定义控制器方法参数类型。

这不会覆盖对解析处理程序方法参数的内置支持。要自定义对参数解析的内置支持,请RequestMappingHandlerAdapter直接配置。

二、场景描述

在权限场景中,通常会有要求用户登录之后才能访问的场景。对于这些问题可以多种解决方案,如:使用Cookie+Session的会话控制、使用拦截器、使用SpringSecurity或shiro等权限管理框架等。

这里使用Cookie+Session处理。处理的逻辑为:

用户第一次登录之后会得到一个cookie,在以后每次的访问过程中都会携带Cookie进行访问。在后台的Controller中对于需要登录权限的访问接口都要先获取Cookie中的Token,再使用Token从session中获取用户登录信息来判断用户登录情况决定是否放行。

三、存在问题

如果在每个需要登录权限访问的方法中都要写这个逻辑就会使代码重复,出现了冗余。代码基本如下:

@GetMapping("goods")
public Result showGoods1(HttpSession session, @CookieValue("Token") String token){
    // 你每次都要在这里判断token或session
    if (token == null || MyObjectUtil.isEmpty(session.getAttribute(token))){
        return Result.error();
    }
    return Result.ok();
}

所以需要寻找是否有更好的解决方法。

最终找到addArgumentResolvers()这个方法。只需要在这个方法中进行处理即可。

四、问题解决

该方法的处理流程如下:

代码如下:

  • Controller中的方法

    @GetMapping("goods")
    public Result showGoods(User user){
        // 注意:这里的User参数不是由前端传入的,而是由addArgumentResolvers方法处理之后传进来的
        log.info(user.toString());
        // 根据处理之后传入的参数判断是否登录
        if (user == null)
            return Result.error();
        return Result.ok();
    }
    
  • WebMvcController实现了类

    @Configuration
    public class WebConfig implements WebMvcConfigurer {
    
        @Autowired
        private UserArgumentResolver userArgumentResolver;
    
        /**
         * 该方法作用在调用Controller方法的参数传入之前
         * @param resolvers
         */
        @Override
        public void addArgumentResolvers(List<HandlerMethodArgumentResolver> resolvers) {
            // userArgumentResolver该类对Controller传入的参数做了具体处理
            resolvers.add(UserArgumentResolver);
        }
    		
        // 其他配置
    }
    
  • UserArgumentResolver自定义实现类

    /**
     * 该类用于WebConfig配置类的addArgumentResolvers方法传入
     */
    @Component
    public class UserArgumentResolver implements HandlerMethodArgumentResolver {
        /**
         * 该方法用于判断Controller中方法参数中是否有符合条件的参数:
         *                          有则进入下一个方法resolveArgument;
         *                          没有则跳过不做处理
         * @param parameter
         * @return
         */
        @Override
        public boolean supportsParameter(MethodParameter parameter) {
            // 获取传入参数的类型
            Class<?> type = parameter.getParameterType();
            // 如果参数类型有为User类的则符合,进入resolveArgument方法
            if (User.class == type)
                return true;
            return false;
        }
    
        /**
         * 该方法在上一个方法同通过之后调用:
         *      在这里可以进行处理,根据情况返回对象——返回的对象将被赋值到Controller的方法的参数中
         * @param parameter
         * @param mavContainer
         * @param webRequest
         * @param binderFactory
         * @return
         * @throws Exception
         */
        @Override
        public Object resolveArgument(
            MethodParameter parameter, 
            ModelAndViewContainer mavContainer, 
            NativeWebRequest webRequest, 
            WebDataBinderFactory binderFactory
        ) throws Exception {
            HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class);
            HttpSession session = request.getSession();
            // 获取cookie
            Cookie[] cookies = request.getCookies();
            String token = null;
            for (Cookie c : cookies){
                if (c.getName().equals("token")){
                    token = c.getValue();
                    break;
                }
            }
            // 如果token不存在,则返回null
            if (token == null) return null;
            // 获取session中对象
            User user = (User)session.getAttribute(token);
            return user;
        }
    }
    

标签:return,Controller,addArgumentResolvers,token,session,参数,WebMvcConfigurer,方法
来源: https://www.cnblogs.com/wurao/p/15556787.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有