服务端动态修改springmvc请求参数，处理GET,POST @RequestBody参数的请求

偏执的太偏执、 2022-12-28 08:22 209阅读 0赞

背景:每一次请求都用记录请求的用户和IP等业务信息，在rest方法中写代码及其冗余，特使用动态处理的方式。

楼主赋值的是：ip和user

处理：

使用Filter和HttpServletRequestWrapper的方法进行动态赋值

1.创建参数过滤器

package com.bxm.sentinel.web.resolver;
     
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.core.annotation.Order;
    import org.springframework.web.filter.OncePerRequestFilter;
     
    import javax.servlet.FilterChain;
    import javax.servlet.ServletException;
    import javax.servlet.annotation.WebFilter;
    import javax.servlet.http.HttpServletRequest;
    import javax.servlet.http.HttpServletResponse;
    import java.io.IOException;
     
    /**
     * 过滤器拦截所有请求
     * 括号中的数字越大，在多个过滤器的执行顺序越靠前
     *
     * @Author kk.xie
     * @Date 2019/9/19 19:35
     * @Version 1.0
     **/
    @WebFilter(filterName = "paramFilter", urlPatterns = "/*")
    @Order(100)
    public class ParamFilter extends OncePerRequestFilter {
     
        @Autowired
        private ParameterRequestWrapperFactory factory;
     
        @Override
        protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
            //定义一个新的request（名称是wrapper）
            HttpServletRequest wrapper = factory.build(request);
            //将修改过的request（wrapper）返回
            filterChain.doFilter(wrapper, response);
        }
     
     
    }

2.创建请求参数工厂类，有些参数需要调用service获取

package com.bxm.sentinel.web.resolver;
     
    import com.alibaba.fastjson.JSONObject;
    import com.bxm.acl.facade.AclUserManagerService;
    import com.bxm.acl.facade.Utils;
    import com.bxm.acl.facade.model.ResultModel;
    import com.bxm.acl.facade.model.UserVo;
    import com.bxm.warcar.utils.IpHelper;
    import org.apache.commons.lang3.StringUtils;
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.stereotype.Component;
     
    import javax.servlet.ReadListener;
    import javax.servlet.ServletInputStream;
    import javax.servlet.http.HttpServletRequest;
    import javax.servlet.http.HttpServletRequestWrapper;
    import java.io.*;
    import java.util.ArrayList;
    import java.util.Collections;
    import java.util.Enumeration;
     
    /**
     * 请求参数动态赋值工厂类
     *
     * @Author kk.xie
     * @Date 2019/9/20 10:50
     * @Version 1.0
     **/
    @Component
    public class ParameterRequestWrapperFactory {
     
        private Logger logger = LoggerFactory.getLogger(this.getClass());
     
        private final String IP = "ip";
        private final String USER = "user";
     
        @Autowired
        private AclUserManagerService aclUserManagerService;
     
        public ParameterRequestWrapper build(HttpServletRequest request) throws IOException {
            return new ParameterRequestWrapper(request);
        }
     
        class ParameterRequestWrapper extends HttpServletRequestWrapper {
     
            private Logger logger = LoggerFactory.getLogger(this.getClass());
     
            private byte[] bytes;
     
            private HttpServletRequest request;
     
            /**
             * POST请求的RequestBody添加 ip和user
             *
             * @param request
             * @return
             * @throws
             * @author kk.xie
             * @date 2019/9/20 11:04
             */
            public ParameterRequestWrapper(HttpServletRequest request) throws IOException {
                super(request);
                this.request = request;
     
                try (BufferedInputStream bis = new BufferedInputStream(request.getInputStream());
                     ByteArrayOutputStream baos = new ByteArrayOutputStream()) {
                    byte[] buffer = new byte[1024];
                    int len;
                    while ((len = bis.read(buffer)) > 0) {
                        baos.write(buffer, 0, len);
                    }
                    bytes = baos.toByteArray();
                    String body = new String(bytes);
     
                    if(StringUtils.isNotBlank(body)){
                        JSONObject json = JSONObject.parseObject(body);
                        // 获取IP
                        String ip = IpHelper.getIpFromHeader(request);
                        json.put(IP, ip);
                        //获取user
                        String user = getUser(request);
                        json.put(USER, user);
                        body = json.toJSONString();
                        bytes = body.getBytes();
                    }
     
                    logger.info("body: {}", body);
                } catch (IOException ex) {
                    throw ex;
                }
     
            }
     
            /**
             * POST请求的RequestBody添加 ip和user
             *
             * @param
             * @return javax.servlet.ServletInputStream
             * @throws
             * @author kk.xie
             * @date 2019/9/20 11:04
             */
            @Override
            public ServletInputStream getInputStream() throws IOException {
                final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(bytes);
                return new ServletInputStream() {
                    @Override
                    public boolean isFinished() {
                        return false;
                    }
     
                    @Override
                    public boolean isReady() {
                        return false;
                    }
     
                    @Override
                    public void setReadListener(ReadListener readListener) {
     
                    }
     
                    @Override
                    public int read() throws IOException {
                        return byteArrayInputStream.read();
                    }
                };
     
            }
     
            /**
             * POST请求的RequestBody添加 ip和user
             *
             * @param
             * @return java.io.BufferedReader
             * @throws
             * @author kk.xie
             * @date 2019/9/20 11:04
             */
            @Override
            public BufferedReader getReader() throws IOException {
                return new BufferedReader(new InputStreamReader(this.getInputStream()));
            }
     
            /**
             * GET 请求添加 ip和user
             *
             * @param
             * @return java.util.Enumeration<java.lang.String>
             * @throws
             * @author kk.xie
             * @date 2019/9/20 11:07
             */
            @Override
            public Enumeration<String> getParameterNames() {
                Enumeration<String> enumeration = super.getParameterNames();
                ArrayList<String> list = Collections.list(enumeration);
                list.add(IP);
                list.add(USER);
                return Collections.enumeration(list);
            }
     
            /**
             * GET 请求添加 ip和user
             *
             * @param name
             * @return java.lang.String
             * @throws
             * @author kk.xie
             * @date 2019/9/20 11:08
             */
            @Override
            public String getParameter(String name) {
                if(IP.equals(name)){
                    return IpHelper.getIpFromHeader(request);
                }else if(USER.equals(name)){
                    return getUser(request);
                }
                return super.getParameter(name);
            }
     
            /**
             * GET 请求添加 ip和user 
             *
             * @param name
             * @return java.lang.String[]
             * @throws
             * @author kk.xie
             * @date 2019/9/20 11:39
             */
            @Override
            public String[] getParameterValues(String name) {
                if(IP.equals(name)){
                    return new String[]{IpHelper.getIpFromHeader(request)};
                }else if(USER.equals(name)){
                    return new String[]{getUser(request)};
                }
                return super.getParameterValues(name);
            }
        }
     
        private String getUser(HttpServletRequest request) {
            String username = null;
            try {
                // 获取token
                String token = Utils.getToken(request);
     
                ResultModel<UserVo> result = aclUserManagerService.auth(token, 7, request.getRequestURI());
                UserVo userVo = result.getReturnValue();
                username = userVo.getUsername();
            } catch (Exception e) {
                logger.error("get user error!", e);
            }
     
            return username;
        }
    }

3.Application修改，添加@ServletComponentScan 注解

@ServletComponentScan
    @SpringBootApplication
    public class SentinelApiApplication {
     
        public static void main(String[] args) {
            SpringApplication application = new SpringApplication(SentinelApiApplication.class);
            application.run(args);
        }
     
    }

示例rest接口

import com.bxm.sentinel.model.base.PrimaryKeyDto;
    import com.bxm.sentinel.model.dto.MonitorStrategyDto;
    import com.bxm.sentinel.model.dto.MonitorStrategyPageDto;
    import com.bxm.sentinel.model.vo.MonitorStrategyVo;
    import com.bxm.sentinel.service.monitor.MonitorStrategyService;
    import com.bxm.warcar.utils.response.ResultModel;
    import com.bxm.warcar.utils.response.ResultModelFactory;
    import com.github.pagehelper.PageInfo;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.web.bind.annotation.RequestBody;
    import org.springframework.web.bind.annotation.RequestMapping;
    import org.springframework.web.bind.annotation.RequestMethod;
    import org.springframework.web.bind.annotation.RestController;
     
    import javax.servlet.http.HttpServletRequest;
     
    /**
     * 监控策略控制器
     *
     * @Author kk.xie
     * @Date 2019/9/18 16:18
     * @Version 1.0
     **/
    @RestController
    @RequestMapping("/monitorStrategy")
    public class MonitorStrategyController {
     
        @Autowired
        private MonitorStrategyService monitorStrategyService;
        /**
         * 新增
         *
         * @param monitorStrategyDto
         * @return com.bxm.warcar.utils.response.ResultModel<java.lang.Boolean>
         * @throws
         * @author kk.xie
         * @date 2019/9/18 16:49
         */
        @RequestMapping(value = "/add", method = RequestMethod.POST)
        public ResultModel<Boolean> add(@RequestBody MonitorStrategyDto monitorStrategyDto){
            return ResultModelFactory.SUCCESS(monitorStrategyService.add(monitorStrategyDto));
        }
     
        /**
         * 查询
         *
         * @param monitorStrategyPageDto
         * @return com.bxm.warcar.utils.response.ResultModel<com.github.pagehelper.PageInfo<com.bxm.sentinel.model.vo.MonitorStrategyVo>>
         * @throws
         * @author kk.xie
         * @date 2019/9/18 16:50
         */
        @RequestMapping(value = "/selectAll", method = RequestMethod.GET)
        public ResultModel<PageInfo<MonitorStrategyVo>> selectAll(MonitorStrategyPageDto monitorStrategyPageDto){
            return ResultModelFactory.SUCCESS(monitorStrategyService.selectAll(monitorStrategyPageDto));
        }
    }

MonitorStrategyDto和MonitorStrategyPageDto均继承CommonBaseDto

package com.bxm.sentinel.model.base;
     
    import lombok.Data;
     
    /**
     * 通用请求对象
     *
     * @Author kk.xie
     * @Date 2019/9/19 15:17
     * @Version 1.0
     **/
    @Data
    public class CommonBaseDto {
        /**
         * 请求IP
         */
        private String ip;
     
        /**
         * 请求的用户账号
         */
        private String user;
     
    }

参考：

[https://blog.csdn.net/gaoleijie/article/details/79446361][https_blog.csdn.net_gaoleijie_article_details_79446361]

[https://www.cnblogs.com/tianwyam/p/post\_param.html][https_www.cnblogs.com_tianwyam_p_post_param.html]

[https_blog.csdn.net_gaoleijie_article_details_79446361]: https://blog.csdn.net/gaoleijie/article/details/79446361
[https_www.cnblogs.com_tianwyam_p_post_param.html]: https://www.cnblogs.com/tianwyam/p/post_param.html