springboot的Filter过滤参数的空格

落日映苍穹つ 2022-05-17 05:38 176阅读 0赞

package org.module.organization.permission.filter;
    
    import java.io.IOException;
    
    import javax.servlet.DispatcherType;
    import javax.servlet.Filter;
    import javax.servlet.FilterChain;
    import javax.servlet.FilterConfig;
    import javax.servlet.ServletException;
    import javax.servlet.ServletRequest;
    import javax.servlet.ServletResponse;
    import javax.servlet.annotation.WebFilter;
    import javax.servlet.http.HttpServletRequest;
    
    import org.springframework.stereotype.Component;
    
    
    @Component
    @WebFilter(urlPatterns = "/**", filterName = "ParamsFilter",dispatcherTypes=DispatcherType.REQUEST)
    public class ParamsFilter implements Filter {
    
    
    	@Override
    	public void doFilter(ServletRequest arg0, ServletResponse arg1, FilterChain arg2)
    			throws IOException, ServletException {
    		ParameterRequestWrapper parmsRequest = new ParameterRequestWrapper((HttpServletRequest) arg0);
            arg2.doFilter(parmsRequest, arg1);  
    	}
    
    	@Override
    	public void destroy() {
    		
    	}
    
    	@Override
    	public void init(FilterConfig arg0) throws ServletException {
    		
    	}
    
    
    }

package org.module.organization.permission.filter;
    
    import java.io.ByteArrayInputStream;
    import java.io.IOException;
    import java.util.HashMap;
    import java.util.Iterator;
    import java.util.Map;
    import java.util.Set;
    
    import javax.servlet.ReadListener;
    import javax.servlet.ServletInputStream;
    import javax.servlet.http.HttpServletRequest;
    import javax.servlet.http.HttpServletRequestWrapper;
    
    import org.apache.commons.io.IOUtils;
    import org.apache.commons.lang.StringUtils;
    import org.springframework.http.HttpHeaders;
    import org.springframework.http.MediaType;
    
    import com.alibaba.fastjson.JSON;
    
    public class ParameterRequestWrapper extends HttpServletRequestWrapper {
    
    	 private Map<String , String[]> params = new HashMap<String, String[]>();      
         
         public ParameterRequestWrapper(HttpServletRequest request) {      
             // 将request交给父类，以便于调用对应方法的时候，将其输出，其实父亲类的实现方式和第一种new的方式类似      
             super(request);     
             //将参数表，赋予给当前的Map以便于持有request中的参数      
             Map<String, String[]> requestMap=request.getParameterMap();  
             this.params.putAll(requestMap);  
             //this.modifyParameterValues();  
         }      
         /** 
          * 重写getInputStream方法  post类型的请求参数必须通过流才能获取到值 
          */  
         @Override  
         public ServletInputStream getInputStream() throws IOException {  
               //非json类型，直接返回  
               if(!super.getHeader(HttpHeaders.CONTENT_TYPE).equalsIgnoreCase(MediaType.APPLICATION_JSON_VALUE)){  
                   return super.getInputStream();  
               }  
               //为空，直接返回  
               String json = IOUtils.toString(super.getInputStream(), "utf-8");  
               if (StringUtils.isEmpty(json)) {  
                   return super.getInputStream();  
               }  
               Map<String,Object> map=JSON.parseObject(json);
               Set<String> set =map.keySet();      
               Iterator<String> it=set.iterator();      
               while(it.hasNext()){      
                  String key= (String) it.next();      
                  Object values = map.get(key);
                  if (values instanceof String) {
                	  values= ((String) values).trim();      
                  	}
                  map.put(key, values);      
                }      
               ByteArrayInputStream bis = new ByteArrayInputStream(JSON.toJSONString(map).getBytes("utf-8"));  
               return new MyServletInputStream(bis);  
         }  
         /** 
          * 将parameter的值去除空格后重写回去  
          */  
         public void modifyParameterValues(){  
             Set<String> set =params.keySet();      
             Iterator<String> it=set.iterator();      
             while(it.hasNext()){      
                String key= (String) it.next();      
                String[] values = params.get(key);      
                values[0] = values[0].trim();      
                params.put(key, values);      
              }      
         }      
         /** 
          * 重写getParameter 参数从当前类中的map获取   
          */  
         @Override      
         public String getParameter(String name) {  
             String[]values = params.get(name);      
             if(values == null || values.length == 0) {      
                 return null;      
             }      
             return values[0];      
         }    
         /** 
          * 重写getParameterValues 
          */  
         public String[] getParameterValues(String name) {//同上      
              return params.get(name);      
         }  
           
         class MyServletInputStream extends  ServletInputStream{  
           private ByteArrayInputStream bis;  
           public MyServletInputStream(ByteArrayInputStream bis){  
               this.bis=bis;  
           }  
           @Override  
           public boolean isFinished() {  
               return true;  
           }  
     
           @Override  
           public boolean isReady() {  
               return true;  
           }  
     
           @Override  
           public void setReadListener(ReadListener listener) {  
                 
           }  
           @Override  
           public int read() throws IOException {  
                return bis.read();  
           }  
         }  
    }