为了多次读取ServletInputStream引发的一系列问题

目录
  • 多次读取ServletInputStream引发的问题
    • 因为服务器和app之间传输方式是JSON
    • 于是写了下面的filter来让ServletInputStream来多次读取
    • 解决思路:自己解决
  • ServletInputStream重复读取问题
    • 直接贴代码了,亲测能用

多次读取ServletInputStream引发的问题

因为服务器和app之间传输方式是JSON

格式如下

{
 head:null
body:null
token:xxxxxxxxxxxxxxxxxxxxx
}

所以想在服务端写个拦截器或者filter先读取token,来验证身份.但是如果前置拦截,那么在springMVC里的参数控制器就会读取不到,导致@RequestBody 这个注解的参数不起作用.

原因在于ServletInputStream已经在前置拦截器里读取过一次,在接下来的ArgumentHandler里读取的时候就不能再读了,导致取不到数据而无法赋值.

于是写了下面的filter来让ServletInputStream来多次读取

先写个request

public class MyHttpRequest extends HttpServletRequestWrapper {
 private static Logger log=Logger.getLogger(MyHttpRequest.class);
    private  byte[] bytes;
    /**
     * @param request {@link javax.servlet.http.HttpServletRequest} object.
     * @throws IOException
     */
    public MyHttpRequest(HttpServletRequest request) throws IOException {
        super(request);
        bytes= IOUtils.toByteArray( request.getInputStream());
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {
        final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(bytes);
        return new DelegatingServletInputStream(byteArrayInputStream);
    }
    @Override
    public BufferedReader getReader() throws IOException {
     return new BufferedReader(new InputStreamReader(getInputStream()));
    }

以上内容是先读取出来,然后放到一个byte[]里,以后每次拿流的时候就可以再新建一个关于Byte[]的流.

然后再写一个filter

public class MyFilter implements Filter{
 @Override
 public void destroy() {
 }

 @Override
 public void doFilter(ServletRequest req, ServletResponse res, FilterChain chain)
   throws IOException, ServletException {
  HttpServletRequest request=(HttpServletRequest) req;
  MyHttpRequest myrequest=new MyHttpRequest(request);
  chain.doFilter(myrequest, res);
 }

 @Override
 public void init(FilterConfig arg0) throws ServletException {
 }
}

在web.xml里配置一下,试试效果,非常不错.可以多次读取了,

但是,我去

在post请求下,request.getParamter这种东西又不起作用了.

原因在于我们重写的那个request是默认调用父类的super.getparamter方法,而super.getparamter又依赖于super.getInputStream,但是在我们重写的request里的构造方法里,第一句话就老request注入到super里,然后下一句话就把老的request的流给读了,导致我们调用super.getparameter的时候遇到了一个空的流。

解决思路:自己解决

package com.hrhs.jyj.filter;
import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import org.apache.commons.io.IOUtils;
import org.apache.log4j.Logger;
import org.springframework.mock.web.DelegatingServletInputStream;

public class MyHttpRequest extends HttpServletRequestWrapper {
 private static Logger log = Logger.getLogger(MyHttpRequest.class);
 private byte[] bytes;
 private String body;
 private Map<String, List<String>> map;
 private int readMap=0;
 private String queryString;
 /**
  * @param request
  *            {@link javax.servlet.http.HttpServletRequest} object.
  * @throws IOException
  */
 public MyHttpRequest(HttpServletRequest request) throws IOException {
  super(request);
  bytes = IOUtils.toByteArray(request.getInputStream());
  queryString = request.getQueryString();

 }

 @Override
 public ServletInputStream getInputStream() throws IOException {
  final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(bytes);
  return new DelegatingServletInputStream(byteArrayInputStream);
 }

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

 @Override
 public String getParameter(String name) {
  log.info("走了getParameter");
  return super.getParameter(name);
 }

 @Override
 public Map<String, String[]> getParameterMap() {
  log.info("走了getParameterMap");
  return super.getParameterMap();
 }

 @Override
 public Enumeration<String> getParameterNames() {
  log.info("走了getParameterNames");
  return super.getParameterNames();
 }
 //暂时先改写这一个,其他的也可以修改
 @Override
 public String[] getParameterValues(String name) {
  log.info("走了getParameterValues");
  try {
   Map<String, List<String>> nameVals = doParameter();
   List<String> list = nameVals.get(name);
   if(list!=null&&list.size()>0){
    return list.toArray(new String[]{});
   }
  } catch (UnsupportedEncodingException e) {
   e.printStackTrace();
  }
  return new String[]{};
 }
 //这里获取所有参数值的map
 public Map<String, List<String>> doParameter() throws UnsupportedEncodingException {
  if(readMap==0){
   //这里把post里的参数和地址栏参数结合到一起,然后解析
   body = new String(bytes, getCharacterEncoding())+"&"+queryString;
   String[] nameVals = body.split("&");
   map = new HashMap<String, List<String>>();
   for (String nameVal : nameVals) {
    String name = nameVal.split("=")[0];
    String val = nameVal.split("=")[1];
    if (map.containsKey(name)) {
     List<String> vals = map.get(name);
     vals.add(val);
     map.put(name, vals);
    } else {
     List<String> vals = new ArrayList<String>();
     vals.add(val);
     map.put(name, vals);
    }
   }
   readMap=1;
  }
  return map;
 }
}

ServletInputStream重复读取问题

本来是想实现tomcat的acess_log打印post请求参数。吐槽一下,tomcat功能和nginx差了好几条街。网上找了个方法,用tomcat的filter来实现。

但是,写filter的时候,发现了ServletInputStream重复读取问题。

网上找个几个方法,都不能直接用。综合网上的资料,根据自己的理解,终于能完美运行了。

直接贴代码了,亲测能用

首先要写个BufferHttpServletRequestWrapper类,用来复制HttpServletRequest request。

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

public class BufferHttpServletRequestWrapper extends HttpServletRequestWrapper {
 private final byte[] body;
 public BufferHttpServletRequestWrapper(HttpServletRequest request) throws IOException {
  super(request);
  InputStream is = request.getInputStream();
        ByteArrayOutputStream baos = new ByteArrayOutputStream();
        byte buff[] = new byte[ 1024 ];
        int read;
        while( ( read = is.read( buff ) ) > 0 ) {
            baos.write( buff, 0, read );
        }
        body = baos.toByteArray();
 }

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

然后filter实现如下:

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
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.http.HttpServletRequest;

/**
 * Servlet Filter implementation class PostDataDumperFilter
 */

public class PostDataDumperFilter implements Filter {

 private FilterConfig filterConfig = null;

    /**
     * Default constructor.
     */
    public PostDataDumperFilter() {
        // TODO Auto-generated constructor stub
    }

 /**
  * @see Filter#destroy()
  */
 public void destroy() {
  this.filterConfig = null;
 }

 /**
  * @see Filter#doFilter(ServletRequest, ServletResponse, FilterChain)
  */
 public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
  if (filterConfig == null)
   return;
  //备份HttpServletRequest
  ServletRequest requestWrapper = null;
        if(request instanceof HttpServletRequest) {
            requestWrapper = new BufferHttpServletRequestWrapper((HttpServletRequest) request);
        }
  //使用流
  InputStream reader = requestWrapper.getInputStream();
  ByteArrayOutputStream byteOutput = new ByteArrayOutputStream(100);
  int i =0;
  byte [] b = new byte[100];
  while((i = reader.read(b))!= -1){
   byteOutput.write(b, 0, i);
  }
  request.setAttribute("post", new String(byteOutput.toByteArray()));
  // pass the request along the filter chain
  if(null == requestWrapper){
   chain.doFilter(request, response);
  } else {
            chain.doFilter(requestWrapper, response);
        }
 }

 /**
  * @see Filter#init(FilterConfig)
  */
 public void init(FilterConfig fConfig) throws ServletException {
  this.filterConfig = fConfig;
 }
}

好了,就写到这里。以上为个人经验,希望能给大家一个参考,也希望大家多多支持我们。

(0)

相关推荐

  • spring应用中多次读取http post方法中的流遇到的问题

    一.问题简述 先说下为啥有这个需求,在基于spring的web应用中,一般会在controller层获取http方法body中的数据. 方式1: 比如http请求的content-type为application/json的情况下,直接用@RequestBody接收. 方式2: 也有像目前我们在做的这个项目,比较原始,是直接手动读取流.(不要问我为啥这么原始,第一版也不是我写的.) @RequestMapping("/XXX.do") public void XXX(HttpServl

  • 解决spring 处理request.getInputStream()输入流只能读取一次问题

    一般我们会在InterceptorAdapter拦截器中对请求进行验证 正常普通接口请求,request.getParameter()可以获取,能多次读取 如果我们的接口是用@RequestBody来接受数据,那么我们在拦截器中 需要读取request的输入流 ,因为 ServletRequest中getReader()和getInputStream()只能调用一次 这样就会导致controller 无法拿到数据. 解决方法 : 1.自定义一个类 BodyReaderHttpServletReq

  • Java INPUTSTREAM如何实现重复使用

    引语: 之前做项目的时候遇到一个问题,就是从网络中读取的图片要上传到oss,而且要对图片进行裁剪和压缩,其中上传和裁剪都要使用到图片的inputStream, 又因为inputstream不能重复读,导致裁剪是成功的,而上传是失败的.我们今天就提供两种方法来解决,inputStream不能重复读的问题. 问题分析: inputStream的内部有个pos指针,当读取的时候指针会不断的移动,当移动到末尾的时候,就无法再次读取了. 我们写个简单的例子来看下: String text = "测试inp

  • 为了多次读取ServletInputStream引发的一系列问题

    目录 多次读取ServletInputStream引发的问题 因为服务器和app之间传输方式是JSON 于是写了下面的filter来让ServletInputStream来多次读取 解决思路:自己解决 ServletInputStream重复读取问题 直接贴代码了,亲测能用 多次读取ServletInputStream引发的问题 因为服务器和app之间传输方式是JSON 格式如下 { head:null body:null token:xxxxxxxxxxxxxxxxxxxxx } 所以想在服务

  • PYTHON 中使用 GLOBAL引发的一系列问题

    哪里出问题了 python 中,使用 global 会将全局变量设为本函数可用.同时,在函数内部访问变量会先本地再全局. 在嵌套函数中,使用 global 会产生不合常理的行为. 上代码: In [96]: def x(): b = 12 def y(): global a,b a = 1 b = 2 y() print "b =",b ....: In [97]: a = 111 In [98]: del b In [99]: x() b = 12 In [100]: a Out[1

  • 针对distinct疑问引发的一系列思考

    有人提出了这样一个问题,整理出来给大家也参考一下 假设有如下这样一张表格: 这里的数据,具有如下的特征:在一个DepartmentId中,可能会有多个Name,反之也是一样.就是说Name和DepartmentId是多对多的关系. 现在想实现这样一个查询:按照DepartmentID排完序之后(第一步),再获取Name列的不重复值(第二步),而且要保留在第一步后的相对顺序.以本例而言,应该返回三个值依次是:ACB 我们首先会想到下面这样一个写法 select distinct name from

  • java log is判断引发的一系列事件解析

    目录 序 log 自动加上if判断 调研 思考 这个topic干什么的啊? 思考 小结 说回来 lombok 到底有没有自动加if的插件啊 错误案例 错误分析 原因分析 log isDebug 判断的必要性 总结 后续 序 首先来源是 老大code Review的时候 说到日志,让我从 log.debug 加上一个if 判断,log.isDebug ,这一听就想到是 为了防止debug 日志不打印,导致的String拼接等 没必要的损耗 嗯嗯 老大说的确实有道理,改吧 我靠,这一查几百个 log

  • 关于Poi读取Excel引发内存溢出问题的解决方法

    前言 最近生产环境有个老项目一直内存报警,不时的还出现内存泄漏,导致需要重启服务器,已经严重影响正常服务了. 分析 1.dump内存文件 liunx使用如下命令: ./jmap -dump:format=b,file=heap.hprof pid 2.使用Eclipse Memory Analysis进行分析 异常如下: at org.apache.poi.xssf.usermodel.XSSFRow.<init>(Lorg/openxmlformats/schemas/spreadsheet

  • mysql乱码问题分析与解决方法

    开发过程中总避免不了遇到恶心的乱码,或者由乱码引发的一系列问题.出现乱码是字符集的原因一般而言和逻辑没有太大关系,也就是说整个系统大的方向没有问题,小的地方出现了漏洞,进而导致程序不能正常运行,所以说出现乱码是一件令人非常很恶心的事情.这里简要介绍一下自己遇到的乱码问题和解决问题的过程中的想法以及大致的操作,我们要学会的是如何分析问题进而解决问题,而不仅仅是照着网上的操作去一次次的解决眼前的困难,"鱼"与"渔"的区别就在于此. 交代背景: 要实现的功能很简单,用do

  • mysql服务性能优化—my.cnf_my.ini配置说明详解(16G内存)

    此配置是老男孩生产线上使用的配置,在培训的时候,他给的,我在这里,对各参数添加了中文说明 这配置已经优化的不错了,如果你的mysql没有什么特殊情况的话,可以直接使用该配置参数 MYSQL服务器my.cnf配置文档详解 硬件:内存16G [client] port = 3306 socket = /data/3306/mysql.sock [mysql] no-auto-rehash [mysqld] user = mysql port = 3306 socket = /data/3306/my

  • MySQL配置文件my.cnf优化详解(mysql5.5)

    MySQL 5.5.13 参数说明: [client] character-set-server = utf8 port = 3306 socket = /data/mysql/3306/mysql.sock [mysqld] character-set-server = utf8 user = mysql port = 3306 socket = /data/mysql/3306/mysql.sock basedir = /usr/local/webserver/mysql datadir =

  • 浅谈javascript基础之客户端事件驱动

    我们知道,面向对象发展起来后,"一夜之间",几乎所有的语言都能基于对象了,JavaScript也是基于对象的语言.用户在浏览器上的行为称作"事件",之后引发的一系列动作,比如弹窗啦,改变浏览器大小啦,验证啦,balabala,都叫做"事件驱动".当然,这次我主要介绍几个常常发生的事件. ps:对于js脚本的支持以浏览器而定!!!有的低版本的浏览器可能不支持!!! 1.单击事件(onClick) 啥叫单击事件呢?当用户单击鼠标按钮是,就会产生单击事

  • 详解Linux进程间通信——使用信号量

    一.什么是信号量 为了防止出现因多个程序同时访问一个共享资源而引发的一系列问题,我们需要一种方法,它可以通过生成并使用令牌来授权,在任一时刻只能有一个执行线程访问代码的临界区域.临界区域是指执行数据更新的代码需要独占式地执行.而信号量就可以提供这样的一种访问机制,让一个临界区同一时间只有一个线程在访问它,也就是说信号量是用来调协进程对共享资源的访问的. 信号量是一个特殊的变量,程序对其访问都是原子操作,且只允许对它进行等待(即P(信号变量))和发送(即V(信号变量))信息操作.最简单的信号量是只

随机推荐