如何解決HttpServletRequest.getInputStream()多次讀取問(wèn)題

更新時(shí)間：2023年07月31日 09:16:53 作者：yxl_num

這篇文章主要介紹了如何解決HttpServletRequest.getInputStream()多次讀取問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

HttpServletRequest.getInputStream()多次讀取問(wèn)題

使用POST方法發(fā)送數(shù)據(jù)時(shí)，我們習(xí)慣于把數(shù)據(jù)包裝成json格式。

有些情況下，我們會(huì)在Filter中讀取body數(shù)據(jù)進(jìn)行數(shù)據(jù)校驗(yàn)，GET方法獲取參數(shù)比較簡(jiǎn)單。

對(duì)于POST方法，可使用如下方法從request中獲取body參數(shù)：

 private String getBody(HttpServletRequest request) throws IOException {
    InputStream in = request.getInputStream();
    BufferedReader br = new BufferedReader(new InputStreamReader(in, Charset.forName("UTF-8")));
    StringBuffer sb = new StringBuffer("");
    String temp;
    while ((temp = br.readLine()) != null) {
        sb.append(temp);
    }
    if (in != null) {
        in.close();
    }
    if (br != null) {
        br.close();
    }
    return sb.toString();
}

注意，這里有了一次request.getInputStream()調(diào)用。

但是在測(cè)試時(shí)，一直報(bào)JSON格式不正確的錯(cuò)誤。經(jīng)調(diào)查發(fā)現(xiàn)，項(xiàng)目中使用了公司基礎(chǔ)組件中的Filter，而該Filter中也解析了body。

同時(shí)，不出所料，也是通過(guò)調(diào)用getInputStream()方法獲取的。

原來(lái)：

一個(gè)InputStream對(duì)象在被讀取完成后，將無(wú)法被再次讀取，始終返回-1；
InputStream并沒(méi)有實(shí)現(xiàn)reset方法（可以重置首次讀取的位置），無(wú)法實(shí)現(xiàn)重置操作；

因此，當(dāng)自己寫(xiě)的Filter中調(diào)用了一次getInputStream()后，后面再調(diào)用getInputStream()讀取的數(shù)據(jù)都為空，所以才報(bào)JSON格式不正確的錯(cuò)誤。

解決方法

緩存數(shù)據(jù)
使用HttpServletRequestWrapper進(jìn)行包裝

緩存數(shù)據(jù)

所謂緩存數(shù)據(jù)，其實(shí)就是調(diào)用ServletRequest的setAttribute(String s, Object o)來(lái)存儲(chǔ)數(shù)據(jù)。

1.獲取到body后，直接緩存

String body = getBody(request);
request.setAttribute("body", body);

優(yōu)點(diǎn)：

方便

缺點(diǎn)：

不能控制第三方Filter

1.其他地方需要使用body時(shí)，只需調(diào)用getAttribute方法就能獲取數(shù)據(jù)了：

request.getAttribute("body");

HttpServletRequestWrapper包裝

public class RequestWrapper extends HttpServletRequestWrapper {
    private final byte[] body;
    public RequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        body = getBodyStringFromReq(request).getBytes(Charset.forName("UTF-8"));
    }
    public String getBodyString() {
        try {
            return new String(body, "UTF-8");
        } catch (UnsupportedEncodingException ex) {
            return new String(body);
        }
    }
    private String getBodyStringFromReq(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中使用時(shí)，FilterChain.doFilter()傳入Wrapper對(duì)象：

public class TestFilter implements Filter {
    @Override
    public void destroy() {
    }
    @Override
    public void doFilter(ServletRequest request, ServletResponse response,
            FilterChain chain) throws IOException, ServletException {
        RequestWrapper requestWrapper = new RequestWrapper((HttpServletRequest)request);
        String body = requestWrapper.getBodyString();
        chain.doFilter(requestWrapper, response);  //傳入Wrapper對(duì)象
    }
    @Override
    public void init(FilterConfig arg0) throws ServletException {
    }
}

這樣，位于后面的Filter就可以擁有唯一一次調(diào)用HttpServletRequest.getInputStream()的機(jī)會(huì)了。

優(yōu)點(diǎn)：

不影響第三方Filter

缺點(diǎn)：

多寫(xiě)了這么多代碼，麻煩了一些