springboot的Filter过滤参数的空格

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();  
       }  
     }  
}

猜你喜欢

转载自blog.csdn.net/sinat_21184471/article/details/81481088