对get/post请求的参数，进行处理，将字符串前后空格置换

秒速五厘米 2023-02-23 12:08 14阅读 0赞

import com.cdls.carp.business.filter.ParamsFilter;
    import org.springframework.boot.web.servlet.FilterRegistrationBean;
    import org.springframework.context.annotation.Bean;
    import org.springframework.context.annotation.Configuration;
    
    import javax.servlet.DispatcherType;
    
    /**
     * @description: 过滤器配置类
     * @author xrj
     * @date 2020/7/1 10:31
     */
    @Configuration
    public class FilterConfig {
    
        @Bean
        public FilterRegistrationBean parmFilterRegistration() {
            FilterRegistrationBean registration = new FilterRegistrationBean();
            registration.setDispatcherTypes(DispatcherType.REQUEST);
            registration.setFilter(new ParamsFilter());
            registration.addUrlPatterns("/*");
            registration.setName("ParamsFilter");
            registration.setOrder(Integer.MAX_VALUE - 1);
            return registration;
        }
    
    }

import javax.servlet.*;
    import javax.servlet.http.HttpServletRequest;
    import java.io.IOException;
    
    /**
     * @author xrj
     * @description: 请求参数过滤器
     * @date 2020/7/1 10:33
     */
    public class ParamsFilter implements Filter {
    
        @Override
        public void init(FilterConfig filterConfig) throws ServletException {
    
        }
    
        @Override
        public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
    
            HttpServletRequest httpRequest = (HttpServletRequest) servletRequest;
    
            ParamsRequestWrapper requestWrapper = new ParamsRequestWrapper(httpRequest);
            filterChain.doFilter(requestWrapper, servletResponse);
    
        }
    
        @Override
        public void destroy() {
    
        }
    
    }

import com.alibaba.fastjson.JSON;
    import lombok.extern.slf4j.Slf4j;
    import org.apache.commons.io.IOUtils;
    import org.apache.commons.lang3.StringUtils;
    import org.springframework.http.HttpHeaders;
    import org.springframework.http.MediaType;
    
    import javax.servlet.ServletInputStream;
    import javax.servlet.http.HttpServletRequest;
    import javax.servlet.http.HttpServletRequestWrapper;
    import java.io.ByteArrayInputStream;
    import java.io.IOException;
    import java.util.HashMap;
    import java.util.Iterator;
    import java.util.Map;
    import java.util.Set;
    
    /**
     * @author xrj
     * @description: 对请求参数进行过滤
     * @date 2020/7/1 11:02
     */
    @Slf4j
    public class ParamsRequestWrapper extends HttpServletRequestWrapper {
    
        private Map<String, String[]> params = new HashMap<>();
        private static final String ENCODING = "UTF-8";
        private static final String CLASS_TYPE = "java.lang.String";
    
        /**
         * get请求参数的string去空格
         * @param request
         */
        public ParamsRequestWrapper(HttpServletRequest request) {
            super(request);
            // 将参数表，赋予给当前的Map以便于持有request中的参数
            Map<String, String[]> requestMap = request.getParameterMap();
            log.info("ParamsRequestWrapper转化前参数：" + JSON.toJSONString(requestMap));
            this.params.putAll(requestMap);
            this.modifyParameters();
            log.info("ParamsRequestWrapper转化后参数：" + JSON.toJSONString(params));
        }
    
    
        /**
         * get请求，将parameter的值去除空格后重写回去
         */
        private void modifyParameters() {
            Set<String> set = params.keySet();
            Iterator<String> it = set.iterator();
            while (it.hasNext()) {
                String key = it.next();
                String[] values = params.get(key);
                values[0] = values[0].trim();
                params.put(key, values);
            }
        }
    
    
        /**
         * 重写getInputStream方法  post请求参数必须通过流才能获取到值，string去空格
         */
        @Override
        public ServletInputStream getInputStream() throws IOException {
    
            ServletInputStream stream = super.getInputStream();
    
            //非json类型，直接返回
            if (!super.getHeader(HttpHeaders.CONTENT_TYPE).equalsIgnoreCase(MediaType.APPLICATION_JSON_VALUE)) {
                return stream;
            }
            String json = IOUtils.toString(stream, ENCODING);
    
            if (StringUtils.isBlank(json)) {
                return stream;
            }
    
            log.info("getInputStream转化前参数：" + json);
            Map<String, Object> map = modifyParams(json);
            log.info("getInputStream转化后参数：" + JSON.toJSONString(map));
    
            ByteArrayInputStream bis = new ByteArrayInputStream(JSON.toJSONString(map).getBytes(ENCODING));
    
            return new ParamsServletInputStream(bis);
        }
    
        /**
         * 转换
         * @param json
         * @return
         */
        private static Map<String, Object> modifyParams(String json) {
    
            Map<String, Object> params = JSON.parseObject(json);
            Map<String, Object> maps = new HashMap<>(params.size());
            for (String key : params.keySet()) {
                Object value = getValue(params.get(key));
                maps.put(key, value);
            }
            return maps;
        }
    
        /**
         * 字符串处理
         * @param obj
         * @return
         */
        private static Object getValue(Object obj) {
    
            if (obj == null) {
                return null;
            }
            String type = obj.getClass().getName();
            // 对字符串的处理
            if (CLASS_TYPE.equals(type)) {
                obj = obj.toString().trim();
            }
            return obj;
        }
    
    
    
        /**
         * 重写getParameter 参数从当前类中的map获取
         */
        @Override
        public String getParameter(String name) {
            String[] values = params.get(name);
            if (values == null || values.length == 0) {
                return null;
            }
            return values[0];
        }
    
    }

import javax.servlet.ReadListener;
    import javax.servlet.ServletInputStream;
    import java.io.ByteArrayInputStream;
    import java.io.IOException;
    
    /**
     * @author xrj
     * @description: 请求参数输入流
     * @date 2020/7/1 10:46
     */
    public class ParamsServletInputStream extends ServletInputStream {
    
        private ByteArrayInputStream bis;
    
        public ParamsServletInputStream(ByteArrayInputStream bis) {
            this.bis = bis;
        }
    
        @Override
        public boolean isFinished() {
            return true;
        }
    
        @Override
        public boolean isReady() {
            return true;
        }
    
        @Override
        public void setReadListener(ReadListener readListener) {
    
        }
    
        @Override
        public int read() throws IOException {
            return bis.read();
        }
    
    }

博主强烈推荐：https://blog.csdn.net/persistencegoing/article/details/84376427

希望大家关注我一波,防止以后迷路,有需要的可以加群讨论互相学习java ,学习路线探讨,经验分享与java求职

群号:721 515 304