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

背景:每一次请求都用记录请求的用户和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://www.cnblogs.com/tianwyam/p/post_param.html