Spring boot 在拦截器里或者过滤器中获取 Request body里的json请求参数

有的时候 需要在 过滤器中或者拦截器中处理一些 数据问题 需要获取 Request body 里的参数

会发现 HttpServletRequest 这个类 获取不到 application/json;charset=UTF-8格式的数据

只能获取 form/data 的参数

如果使用 form/data 格式传输 会发现 控制器 @RequestBody 这个注解 是获取不到 这种格式的数据

但是又非要用 json格式传输 就需要 重写 HttpServletRequestWrapper 这个类 使用这个类 进行获取

package com.erp.init.wrapper;


import com.erp.init.utils.StreamUtil;

import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStreamReader;

/**
 * @author json
 * @description  json格式的报文请求
 * @date 2018/9/28  13:51.
 */
public class JsonParameterRequestWrapper extends HttpServletRequestWrapper {
    //用于保存读取body中数据
    private  byte[] body;
    private String bodyMessage;
    public JsonParameterRequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        //读取请求的数据保存到本类当中
        body = StreamUtil.readBytes(request.getReader(), "UTF-8");
        bodyMessage =  new String(body,"utf-8");
    }

    //覆盖（重写）父类的方法
    @Override
    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(getInputStream()));
    }
    //覆盖（重写）父类的方法
    @Override
    public ServletInputStream getInputStream() throws IOException {
        final ByteArrayInputStream bais = new ByteArrayInputStream(body);
        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 bais.read();
            }
        };
    }
    /**
     * 获取body中的数据
     * @return
     */
    public byte[] getBody() {
        return body;
    }
    /**
     * 把处理后的参数放到body里面
     * @param body
     */
    public void setBody(byte[] body) {
        this.body = body;
    }

    public String getBodyMessage() {
        return bodyMessage;
    }
}

在过滤器中 使用这个类 来获取 JSON 数据

StreamUtil 工具类

package com.erp.init.utils;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.List;

public class StreamUtil {
	
	public static byte[] readStream(InputStream stream,int length) throws IOException{
		byte[]streamData=null;
		List<Integer> lengths = new ArrayList<Integer>();
		List<byte[]> buffers = new ArrayList<byte[]>();
		int l = 0;  int totalLength = 0;  byte[] buffer = null; // 
		while (totalLength < length && l != -1) { //
			buffer = new byte[length];
			l = stream.read(buffer);
			if (l != -1) {
				lengths.add(new Integer(l));
				buffers.add(buffer);
				totalLength+=l;
			}
		}
		if(totalLength==0) {
			return null;
		}
		l=0;
		streamData = new byte[totalLength];
		length =buffers.size();
		int blength=0;
		byte[] bbuffer=null;
		for (int i = 0; i < length; i++) {
			blength = ((Integer) lengths.get(i)).intValue();
			bbuffer = (byte[]) buffers.get(i);
			System.arraycopy(bbuffer, 0, streamData, l,blength);
			l=l+blength;
		}
		stream=null; lengths=null; buffers=null;	buffer=null;
		return streamData;
	}

	public static byte[] readBytes(BufferedReader bufferedReader,String charset) throws IOException{
		StringBuffer sb = new StringBuffer();
		String s;
		while ((s = bufferedReader.readLine()) != null) {
			sb.append(s);
		}
		if(sb.length() == 0){
			return "".getBytes(charset);
		}
		return sb.toString().getBytes(charset);
	}

}

但是会有一个细节

经常会使用@RequestBody注解，这个注解是非常的好用。在请求参数传到后台的时候做一个参数检验时，使用SpringMVC的拦截器，在拦截器里把request的数据读取出来然后校验。但是在使用了拦截器的时候会出现一个问题，在拦截器读取了request的数据，在Controller里面@RequestBody注解获取Json就会失败就读取不到数据

重写HttpServletRequestWrapper方法

这种方法就是通过重写HttpServletRequestWrapper把request的保存下来，然后通过过滤器保存下来的request在填充进去，这样就可以多次读取request了

这个类已经把这些问题都解决了

粘贴复制 使用即可