http请求过滤器

初始化过滤器filter

import com.alibaba.fastjson.JSONObject;
import com.example.domain.User;
import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.Map;

public class UserFilte  implements Filter {
    @Override
    public void init(FilterConfig filterConfig) throws ServletException {

    }

    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
        HttpServletRequest req= (HttpServletRequest) servletRequest;

        //获取请求路径url
        String url=req.getRequestURI();
        //获取请求方法
        if("GET".equals(req.getMethod())){
            //get方式下获取请求body下的参数
            String accountId=req.getParameter("accountId");
            //get方式下获取请求body下的所有参数
            Map<String, String[]> reqMap= req.getParameterMap();
            //get方式下获取请求header下的参数
            String headerStr=req.getHeader("accountId");
            filterChain.doFilter(servletRequest,servletResponse);
        }
        MyRequestWrapper waper=new MyRequestWrapper(req);
       String body= waper.getBody();
        User user = JSONObject.parseObject(body, User.class);
        System.out.println(user.toString());
        //向响应信息中写入
        String result="{'dfdf':'45'}";
        PrintWriter write=servletResponse.getWriter();
        write.print(result);
        return;
    }

    @Override
    public void destroy() {

    }
}

post请求的时候，请求参数只能从输入流中获取，但是流读取一次之后就没了，真正的方法将获取不到请求参数，所以此处包装一下。

import com.alibaba.fastjson.JSONObject;
import org.apache.commons.lang.StringUtils;
import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.*;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;

public class MyRequestWrapper   extends HttpServletRequestWrapper {
    private String body;

    public MyRequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        StringBuilder stringBuilder = new StringBuilder();
        BufferedReader bufferedReader = null;
        try {
            InputStream inputStream = request.getInputStream();
            if (inputStream != null) {
                bufferedReader = new BufferedReader(new InputStreamReader(inputStream,"UTF-8"));
                char[] charBuffer = new char[128];
                int bytesRead = -1;
                while ((bytesRead = bufferedReader.read(charBuffer)) > 0) {
                    stringBuilder.append(charBuffer, 0, bytesRead);
                }
            } else {
                stringBuilder.append("");
            }
        } catch (IOException ex) {
            throw ex;
        } finally {
            if (bufferedReader != null) {
                try {
                    bufferedReader.close();
                } catch (IOException ex) {
                    throw ex;
                }
            }
        }
        body = stringBuilder.toString();
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {
        final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes());
        ServletInputStream servletInputStream = new ServletInputStream() {
            @Override
            public boolean isFinished() {
                return false;
            }

            @Override
            public boolean isReady() {
                return false;
            }

            @Override
            public void setReadListener(ReadListener readListener) {

            }

            public int read() throws IOException {
                return byteArrayInputStream.read();
            }
        };
        return servletInputStream;
    }

    @Override
    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(this.getInputStream()));
    }

    public String getBody() {
        return this.body;
    }
    @Override
    public String getParameter(String name) {
        return super.getParameter(name);
    }

    @Override
    public Map<String, String[]> getParameterMap() {
        return super.getParameterMap();
    }

    @Override
    public Enumeration<String> getParameterNames() {
        return super.getParameterNames();
    }

    @Override
    public String[] getParameterValues(String name) {
        return super.getParameterValues(name);
    }
    /**
     * 设置自定义post参数 //
     *
     * @param paramMaps
     * @return
     */
    public void setParamsMaps(Map paramMaps) {
        Map paramBodyMap = new HashMap();
        if (!StringUtils.isEmpty(body)) {
            paramBodyMap = JSONObject.parseObject(body, Map.class);
        }
        paramBodyMap.putAll(paramMaps);
        body = JSONObject.toJSONString(paramBodyMap);
    }
}

import org.springframework.boot.web.servlet.FilterRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class InitFilterService {

    @Bean
    public FilterRegistrationBean UserFilerBean(){
        FilterRegistrationBean registBean=new FilterRegistrationBean();
        UserFilte UserFilter =new UserFilte();
        registBean.setFilter(UserFilter);
        registBean.setName("UserFilter");
        return registBean;
    }
}