完美解决request请求流只能读取一次的问题

更新时间：2020年08月24日 09:31:13 作者：如来神掌十八式

这篇文章主要介绍了完美解决request请求流只能读取一次的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

解决request请求流只能读取一次的问题

实际开发碰到的问题

解决request请求流中的数据二次或多次使用问题

实际开发碰到的问题

springboot项目中，为了防止sql注入，采用Filter拦截器对所有请求流中的json数据进行校验，请求数据没问题则继续向下执行，在后边的代码中应用到请求参数值时，发现request中的json数据为空；

除上边描述的情况，尝试过两次从request中获取json数据，第二次同样是获取不到的。

解决request请求流中的数据二次或多次使用问题

继承HttpServletRequestWrapper，将请求体中的流copy一份，覆写getInputStream()和getReader()方法供外部使用。每次调用覆写后的getInputStream()方法都是从复制出来的二进制数组中进行获取，这个二进制数组在对象存在期间一直存在，这样就实现了流的重复读取。

//增强类
public class BodyReaderHttpServletRequestWrapper extends HttpServletRequestWrapper {
  //保存流
  private byte[] requestBody = null;
 
  public BodyReaderHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
    super(request);
    requestBody = StreamUtils.copyToByteArray(request.getInputStream());
  }
 
  @Override
  public ServletInputStream getInputStream() throws IOException {
 
    final ByteArrayInputStream bais = new ByteArrayInputStream(requestBody);
 
    return new ServletInputStream() {
 
      @Override
      public int read() throws IOException {
        return bais.read();
      }
 
      @Override
      public boolean isFinished() {
        return false;
      }
 
      @Override
      public boolean isReady() {
        return false;
      }
 
      @Override
      public void setReadListener(ReadListener readListener) {
 
      }
    };
  }
 
  @Override
  public BufferedReader getReader() throws IOException{
    return new BufferedReader(new InputStreamReader(getInputStream()));
  }
} 
 
//过滤器
@Component
@WebFilter
public class RequestSqlValidFilter implements Filter {
  private final Logger logger = LoggerFactory.getLogger(this.getClass());
 
  @Override
  public void init(FilterConfig filterConfig) throws ServletException {
 
  }
 
  @Override
  public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
    ServletRequest requestWrapper = new BodyReaderHttpServletRequestWrapper((HttpServletRequest)request);
 
    //请求参数合法，无sql注入
    if((sqlValid(request, response))){
      chain.doFilter(requestWrapper, response);//requestWrapper中保存着供二次使用的请求数据
    }else {
      logger.error("RequestSqlValidFilter sqlValid param error");
    }
  }
 
  @Override
  public void destroy() {
 
  }

补充知识：【java web】解决流读完一次就不能再次获取body数据的问题

问题来自我工作业务上的需求：前端请求时需要将json用RSA算法加密，数据经过后端过滤器进行自动解密，这样做的好处是以后不需要在每一个方法里都手动解密一次，增加代码的简洁性、可维护性。

但这样一来便会面临一个问题：http的request请求的输入流在过滤器中就已经被读取了(因为需要读取并解密request body 里被前端加密了的json数据)，流只能被读取一次，这样一来数据便传不进controller里，导致接下来的业务无法进行。

于是我上网找了一些资料并成功解决了这个问题，基本思路是封装原生的HttpServletRequest请求，将其输入流里的数据保存在字节数组里，最后重写getInputStream方法，使其之后每次读取数据都是从字节数组里读取的。

第一步：写一个Request包装类BodyReaderHttpServletRequestWrapper

public class BodyReaderHttpServletRequestWrapper extends HttpServletRequestWrapper {
 
  private byte[] body;
 
  public BodyReaderHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
    super(request);
    body = HttpHelper.getBodyString(request).getBytes(Charset.forName("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 int read() throws IOException {
        return bais.read();
      }
 
      @Override
      public boolean isFinished() {
        return false;
      }
 
      @Override
      public boolean isReady() {
        return false;
      }
 
      @Override
      public void setReadListener(ReadListener readListener) {
 
      }
 
    };
  }
 
  public void setInputStream(byte[] body){
    this.body = body;
  }

里面涉及一个HttpHelper类，顺便也贴出来

public class HttpHelper {
 
  /**
   * 获取请求Body
   * @param request
   * @return
   */
  public static String getBodyString(ServletRequest request) {
    StringBuilder sb = new StringBuilder();
    InputStream inputStream = null;
    BufferedReader reader = null;
    try {
      inputStream = request.getInputStream();
      reader = new BufferedReader(new InputStreamReader(inputStream, Charset.forName("UTF-8")));
      String line = "";
      while ((line = reader.readLine()) != null) {
        sb.append(line);
      }
    } catch (IOException e) {
      e.printStackTrace();
    } finally {
      if (inputStream != null) {
        try {
          inputStream.close();
        } catch (IOException e) {
          e.printStackTrace();
        }
      }
      if (reader != null) {
        try {
          reader.close();
        } catch (IOException e) {
          e.printStackTrace();
        }
      }
    }
    return sb.toString();
  }  
}

第二步:编写包装Filter

这个filter类可以将经过的原生Request请求自动包装成BodyReaderHttpServletRequestWrapper

/**
 * desc : 用于包装原生request, 解决流读完一次就不能再次获取body数据的问题
 * Created by Lon on 2018/3/9.
 */
public class RequestWrapperFilter implements Filter{
 
  private static final Logger LOGGER = LoggerFactory.getLogger(RequestWrapperFilter.class);
 
  @Override
  public void init(FilterConfig filterConfig) throws ServletException {
 
  }
 
  @Override
  public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
    ServletRequest requestWrapper = null;
    if(request instanceof HttpServletRequest) {
      requestWrapper = new BodyReaderHttpServletRequestWrapper((HttpServletRequest) request);
    }
    if(null == requestWrapper) {
      LOGGER.error("包装request失败!将返回原来的request");
      chain.doFilter(request, response);
    } else {
      LOGGER.info("包装request成功");
      chain.doFilter(requestWrapper, response);
    } 
  }
 
  @Override
  public void destroy() {
 
  }
}

第三步:在web.xml上配置过滤器

 <filter>
  <filter-name>RequestWrapperFilter</filter-name>
  <filter-class>com.kx.security.filter.RequestWrapperFilter</filter-class>
 </filter>
 <filter-mapping>
  <filter-name>RequestWrapperFilter</filter-name>
  <url-pattern>/*</url-pattern>
 </filter-mapping>

这里要注意的是这个包装过滤器可能要写在web.xml配置文件里某些过滤器的前面，比如解密过滤器，否则被解密过滤器先读取流的话，包装过滤器就读取不了流了。

至此一个简单的解决方法就完成啦！

以上这篇完美解决request请求流只能读取一次的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Java工厂模式的深入了解
这篇文章主要为大家介绍了Java工厂模式，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助
2022-01-01
Java根据开始时间和结束时间及周几计算日期的示例代码
在Java 7中,java.time包不存在,所以我们需要使用java.util.Calendar和java.util.Date类来实现类似的功能,这篇文章主要介绍了Java根据开始时间和结束时间及周几计算出日期的示例代码,需要的朋友可以参考下
2024-06-06
Spring MVC过滤器-登录过滤的代码实现
本篇文章主要介绍了Spring MVC过滤器-登录过滤，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧。
2017-01-01
Java最简洁数据结构之冒泡排序快速理解
冒泡排序是编程中数据结构绕不过的一个基础点，有关于冒泡排序的文章也有很多，但可能会比较缭乱未能理解，本章将一子u为简洁明了的例图带你通关冒泡排序
2021-11-11
使用idea解决maven依赖冲突的问题
这篇文章主要介绍了使用idea解决maven依赖冲突,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2020-12-12
java统计字符串中指定元素出现次数方法
这篇文章主要介绍了java统计字符串中指定元素出现次数方法,需要的朋友可以参考下
2015-12-12
Java实现字节数B转化为KB、MB、GB的方法示例【测试可用】
这篇文章主要介绍了Java实现字节数B转化为KB、MB、GB的方法,结合实例形式分析了java字节数的转换运算相关操作技巧,需要的朋友可以参考下
2017-08-08
java中@ModelAttribute注解的作用
本文主要介绍了java中@ModelAttribute注解的作用。具有很好的参考价值，下面跟着小编一起来看下吧
2017-02-02
Spring Security自定义异常 AccessDeniedHandler不生效解决方法
本文主要介绍了Spring Security自定义异常 AccessDeniedHandler不生效解决方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2023-07-07
Java使用正则表达式判断字符串是否以字符开始
这篇文章主要介绍了Java使用正则表达式判断字符串是否以字符开始的相关资料,需要的朋友可以参考下
2017-06-06

完美解决request请求流只能读取一次的问题

相关文章

最新评论

大家感兴趣的内容

最近更新的内容

常用在线小工具