Http Servlet请求在读取一次POST body后丢失参数。

110

我想在Java Servlet过滤器中访问两个HTTP请求参数,这里没有什么新鲜事,但惊讶地发现这些参数已经被消耗了!因此,在过滤器链中不再可用。

似乎只有当参数以POST请求主体(例如表单提交)形式到达时才会出现这种情况。

有没有办法读取参数而不消耗它们?

到目前为止,我只找到了这个参考:Servlet Filter using request.getParameter loses Form data

谢谢!


2
也许可以展示一下你是如何实现的代码片段? - Pavel Veller
你使用了 getInputStream() 或者 getReader() 吗?看起来它们会干扰 getParameter() 的执行。 - evanwong
13个回答

132

除此之外,解决这个问题的另一种方法是不使用过滤器链,而是构建自己的拦截器组件,可以使用切面,在解析请求体时操作它。这样做可能更为高效,因为你只需要将请求的 InputStream 转换为自己的模型对象一次。

然而,我仍然认为希望能够多次读取请求体是合理的,特别是当请求通过过滤器链移动时。我通常会使用过滤器链来进行某些操作,以便将它们保留在 HTTP 层,并与服务组件解耦。

Will Hartung所建议的,我通过继承 HttpServletRequestWrapper,消耗请求的InputStream并缓存这些字节来实现这一点。

public class MultiReadHttpServletRequest extends HttpServletRequestWrapper {
  private ByteArrayOutputStream cachedBytes;

  public MultiReadHttpServletRequest(HttpServletRequest request) {
    super(request);
  }

  @Override
  public ServletInputStream getInputStream() throws IOException {
    if (cachedBytes == null)
      cacheInputStream();

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

  private void cacheInputStream() throws IOException {
    /* Cache the inputstream in order to read it multiple times. For
     * convenience, I use apache.commons IOUtils
     */
    cachedBytes = new ByteArrayOutputStream();
    IOUtils.copy(super.getInputStream(), cachedBytes);
  }

 
  /* An input stream which reads the cached request body */
  private static class CachedServletInputStream extends     ServletInputStream {

    private final ByteArrayInputStream buffer;

    public CachedServletInputStream(byte[] contents) {
      this.buffer = new ByteArrayInputStream(contents);
    }

    @Override
    public int read() {
      return buffer.read();
    }

    @Override
    public boolean isFinished() {
      return buffer.available() == 0;
    }

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

    @Override
    public void setReadListener(ReadListener listener) {
  throw new RuntimeException("Not implemented");
    }
  }
}
现在可以通过在传递到过滤器链之前包装原始请求来多次读取请求正文:
public class MyFilter implements Filter {
  @Override
  public void doFilter(ServletRequest request, ServletResponse response,
        FilterChain chain) throws IOException, ServletException {

    /* wrap the request in order to read the inputstream multiple times */
    MultiReadHttpServletRequest multiReadRequest = new MultiReadHttpServletRequest((HttpServletRequest) request);

    /* here I read the inputstream and do my thing with it; when I pass the
     * wrapped request through the filter chain, the rest of the filters, and
     * request handlers may read the cached inputstream
     */
    doMyThing(multiReadRequest.getInputStream());
    //OR
    anotherUsage(multiReadRequest.getReader());
    chain.doFilter(multiReadRequest, response);
  }
}

这个解决方案还可以通过 getParameterXXX 方法多次读取请求主体,因为底层调用是 getInputStream(),它当然会读取缓存的请求 InputStream

编辑

对于较新版本的 ServletInputStream 接口,您需要提供更多方法的实现,如 isReadysetReadListener 等。请参考下面评论中提供的此问题


6
真的吗?底层调用是在原始请求上调用getInputStream(),而您已经读取了这些字节。底层请求不知道您的包装器,所以它怎么会知道要调用包装器的getInputStream()呢? - Frans
2
如果我能把这个加100,我就会这么做。我已经尝试了3-4个小时让它正常工作。感谢您清晰的示例和解释!我很高兴找到了这篇文章! - Doug
24
有什么建议可以让这个工作与Servlet-api 3.0+兼容吗?现在,ServletInputStream已经有了抽象的'isReady()'、'isFinished()'和'setReadListener()'方法来处理必须实现的非阻塞IO。我认为ReadListener可以留空,但不确定该怎么处理'isFinished()'和/或'isReady()'。 - Eric B.
2
我正在为Tomcat 8版本工作。您必须覆盖readPostBody和readChunkedPostBody并使用getStream,以便不设置标志“usingInputStream”和“usingReader”,因为它们会干扰Request#parseParameter()。您也可以覆盖Request#parseParameter(),但如果您不需要使用reader,则会使事情变得过于复杂。当我的代码可用时,我将尝试制定一个完整的答案。 - Silver
13
@EricB. 无论如何还是谢谢你。后来我找到了较新 API 接口的解决方案,这里提供给有兴趣的人。https://dev59.com/WF4b5IYBdhLWcg3wZwu3?lq=1 - dcc
显示剩余12条评论

46

我知道我来晚了,但这个问题对我仍然很重要,而且这篇SO帖子是谷歌搜索中排名前几的结果之一。我要发表我的解决方案,希望其他人能够节省几个小时。

在我的情况下,我需要记录所有请求和响应以及它们的正文。使用Spring Framework,答案实际上很简单,只需使用ContentCachingRequestWrapperContentCachingResponseWrapper即可。

import org.springframework.web.util.ContentCachingRequestWrapper;
import org.springframework.web.util.ContentCachingResponseWrapper;

import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

public class LoggingFilter implements Filter {

    @Override
    public void init(FilterConfig filterConfig) throws ServletException {

    }

    @Override
    public void destroy() {

    }

    @Override
    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
            throws IOException, ServletException {

        ContentCachingRequestWrapper requestWrapper = new ContentCachingRequestWrapper((HttpServletRequest) request);
        ContentCachingResponseWrapper responseWrapper = new ContentCachingResponseWrapper((HttpServletResponse) response);

        try {
            chain.doFilter(requestWrapper, responseWrapper);
        } finally {

            String requestBody = new String(requestWrapper.getContentAsByteArray());
            String responseBody = new String(responseWrapper.getContentAsByteArray());
            // Do not forget this line after reading response content or actual response will be empty!
            responseWrapper.copyBodyToResponse();

            // Write request and response body, headers, timestamps etc. to log files

        }

    }

}

6
这对我没有起作用。requestBodyresponseBody都是空字符串。 - Abhijith Madhav
6
我错了。我本应该使用chain.doFilter(requestWrapper, responseWrapper);而不是chain.doFilter(request, response); - Abhijith Madhav
6
ContentCaching*Wrapper 类的代价昂贵,因为它们消耗输入流,所以 "缓存" 是通过 getContentAsByteArray 方法来完成的。但是,该类不会缓存可能被过滤器链中其他过滤器需要的输入流(这是我使用的情况)。在我看来,这不是一个内容缓存类应有的行为,因此我在 Spring 团队中提出了这个改进 https://jira.spring.io/browse/SPR-16028。 - Federico Piazza
你可以使用Spring的AbstractRequestLoggingFilter,其中大部分工作已经由Spring完成,你只需要覆盖1或2个简单方法即可。 - harsh
1
这对我来说在spring-web-4.3.12.RELEASE版本中不起作用。当我检查源代码时,我发现变量cachedContent被用于存储各种内容,如请求参数和请求输入流。如果仅调用getContentAsByteArray(),它是空的。要获取请求正文,您必须调用getInputStream()。但是,这将使输入流对其他过滤器和处理程序不可用。 - Joseph Tesfaye

10

这基本上是Lathy的答案,但针对更高要求的ServletInputStream进行了更新。

具体来说(对于ServletInputStream),需要实现以下内容:

public abstract boolean isFinished();

public abstract boolean isReady();

public abstract void setReadListener(ReadListener var1);

这是经过编辑的Lathy对象

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

public class RequestWrapper extends HttpServletRequestWrapper {

    private String _body;

    public RequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        _body = "";
        BufferedReader bufferedReader = request.getReader();
        String line;
        while ((line = bufferedReader.readLine()) != null){
            _body += line;
        }
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {

        CustomServletInputStream kid = new CustomServletInputStream(_body.getBytes());
        return kid;
    }

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

我在某个地方找到了这个(它是一个一流的类,用于处理“额外”方法。

import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import java.io.IOException;
import java.io.UnsupportedEncodingException;

public class CustomServletInputStream extends ServletInputStream {

    private byte[] myBytes;

    private int lastIndexRetrieved = -1;
    private ReadListener readListener = null;

    public CustomServletInputStream(String s) {
        try {
            this.myBytes = s.getBytes("UTF-8");
        } catch (UnsupportedEncodingException ex) {
            throw new IllegalStateException("JVM did not support UTF-8", ex);
        }
    }

    public CustomServletInputStream(byte[] inputBytes) {
        this.myBytes = inputBytes;
    }

    @Override
    public boolean isFinished() {
        return (lastIndexRetrieved == myBytes.length - 1);
    }

    @Override
    public boolean isReady() {
        // This implementation will never block
        // We also never need to call the readListener from this method, as this method will never return false
        return isFinished();
    }

    @Override
    public void setReadListener(ReadListener readListener) {
        this.readListener = readListener;
        if (!isFinished()) {
            try {
                readListener.onDataAvailable();
            } catch (IOException e) {
                readListener.onError(e);
            }
        } else {
            try {
                readListener.onAllDataRead();
            } catch (IOException e) {
                readListener.onError(e);
            }
        }
    }

    @Override
    public int read() throws IOException {
        int i;
        if (!isFinished()) {
            i = myBytes[lastIndexRetrieved + 1];
            lastIndexRetrieved++;
            if (isFinished() && (readListener != null)) {
                try {
                    readListener.onAllDataRead();
                } catch (IOException ex) {
                    readListener.onError(ex);
                    throw ex;
                }
            }
            return i;
        } else {
            return -1;
        }
    }
};

最终,我只是想记录请求日志。 上述拼凑在一起的部件帮助我创建了以下内容。

import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.security.Principal;
import java.util.Enumeration;
import java.util.LinkedHashMap;
import java.util.Map;

import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.io.IOUtils;

//one or the other based on spring version
//import org.springframework.boot.autoconfigure.web.ErrorAttributes;
import org.springframework.boot.web.servlet.error.ErrorAttributes;

import org.springframework.core.Ordered;
import org.springframework.http.HttpStatus;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.ServletRequestAttributes;
import org.springframework.web.context.request.WebRequest;
import org.springframework.web.filter.OncePerRequestFilter;


/**
 * A filter which logs web requests that lead to an error in the system.
 */
@Component
public class LogRequestFilter extends OncePerRequestFilter implements Ordered {

    // I tried apache.commons and slf4g loggers.  (one or the other in these next 2 lines of declaration */
    //private final static org.apache.commons.logging.Log logger = org.apache.commons.logging.LogFactory.getLog(LogRequestFilter.class);
    private static final org.slf4j.Logger logger = org.slf4j.LoggerFactory.getLogger(LogRequestFilter.class);

    // put filter at the end of all other filters to make sure we are processing after all others
    private int order = Ordered.LOWEST_PRECEDENCE - 8;
    private ErrorAttributes errorAttributes;

    @Override
    public int getOrder() {
        return order;
    }

    @Override
    protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain)
            throws ServletException, IOException {

        String temp = ""; /* for a breakpoint, remove for production/real code */

        /* change to true for easy way to comment out this code, remove this if-check for production/real code */
        if (false) {
            filterChain.doFilter(request, response);
            return;
        }

        /* make a "copy" to avoid issues with body-can-only-read-once issues */
        RequestWrapper reqWrapper = new RequestWrapper(request);

        int status = HttpStatus.INTERNAL_SERVER_ERROR.value();
        // pass through filter chain to do the actual request handling
        filterChain.doFilter(reqWrapper, response);
        status = response.getStatus();

        try {
            Map<String, Object> traceMap = getTrace(reqWrapper, status);
            // body can only be read after the actual request handling was done!
            this.getBodyFromTheRequestCopy(reqWrapper, traceMap);
            
            /* now do something with all the pieces of information gatherered */
            this.logTrace(reqWrapper, traceMap);
        } catch (Exception ex) {
            logger.error("LogRequestFilter FAILED: " + ex.getMessage(), ex);
        }
    }

    private void getBodyFromTheRequestCopy(RequestWrapper rw, Map<String, Object> trace) {
        try {
            if (rw != null) {
                byte[] buf = IOUtils.toByteArray(rw.getInputStream());
                //byte[] buf = rw.getInputStream();
                if (buf.length > 0) {
                    String payloadSlimmed;
                    try {
                        String payload = new String(buf, 0, buf.length, rw.getCharacterEncoding());
                        payloadSlimmed = payload.trim().replaceAll(" +", " ");
                    } catch (UnsupportedEncodingException ex) {
                        payloadSlimmed = "[unknown]";
                    }

                    trace.put("body", payloadSlimmed);
                }
            }
        } catch (IOException ioex) {
            trace.put("body", "EXCEPTION: " + ioex.getMessage());
        }
    }

    private void logTrace(HttpServletRequest request, Map<String, Object> trace) {
        Object method = trace.get("method");
        Object path = trace.get("path");
        Object statusCode = trace.get("statusCode");

        logger.info(String.format("%s %s produced an status code '%s'. Trace: '%s'", method, path, statusCode,
                trace));
    }

    protected Map<String, Object> getTrace(HttpServletRequest request, int status) {
        Throwable exception = (Throwable) request.getAttribute("javax.servlet.error.exception");

        Principal principal = request.getUserPrincipal();

        Map<String, Object> trace = new LinkedHashMap<String, Object>();
        trace.put("method", request.getMethod());
        trace.put("path", request.getRequestURI());
        if (null != principal) {
            trace.put("principal", principal.getName());
        }
        trace.put("query", request.getQueryString());
        trace.put("statusCode", status);

        Enumeration headerNames = request.getHeaderNames();
        while (headerNames.hasMoreElements()) {
            String key = (String) headerNames.nextElement();
            String value = request.getHeader(key);
            trace.put("header:" + key, value);
        }

        if (exception != null && this.errorAttributes != null) {
            trace.put("error", this.errorAttributes
                    .getErrorAttributes((WebRequest) new ServletRequestAttributes(request), true));
        }

        return trace;
    }
}

请抱着谨慎的态度看待这段代码。
最重要的“测试”是使用有效载荷进行POST请求。这将暴露“双重读取”问题。
伪代码示例。
import org.springframework.web.bind.annotation.*;

@RestController
@RequestMapping("myroute")
public class MyController {
    @RequestMapping(method = RequestMethod.POST, produces = "application/json")
    @ResponseBody
    public String getSomethingExample(@RequestBody MyCustomObject input) {

        String returnValue = "";

        return returnValue;
    }
}

如果只是想测试,可以将“MyCustomObject”替换为普通的“Object”。

本答案汇集了多个SOF帖子和示例,但花费了一些时间将它们整合在一起,希望能帮助未来的读者。

请在我的答案之前给Lathy的答案点赞,在没有它的帮助下我可能无法做到这一步。

以下是我在解决问题时遇到的部分异常。

getReader()已经被调用了

看起来我“借用”的一些地方在这里:

http://slackspace.de/articles/log-request-body-with-spring-boot/

https://github.com/c0nscience/spring-web-logging/blob/master/src/main/java/org/zalando/springframework/web/logging/LoggingFilter.java

https://howtodoinjava.com/servlets/httpservletrequestwrapper-example-read-request-body/

https://www.oodlestechnologies.com/blogs/How-to-create-duplicate-object-of-httpServletRequest-object

https://github.com/c0nscience/spring-web-logging/blob/master/src/main/java/org/zalando/springframework/web/logging/LoggingFilter.java

2021年1月更新。

我已经深刻地认识到,上述代码不适用于“x-www-form-urlencoded”。

x-www-form-urlencoded

请考虑下面的示例:

   @CrossOrigin
    @ResponseBody
    @PostMapping(path = "/mypath", consumes = {MediaType.APPLICATION_FORM_URLENCODED_VALUE})
    public ResponseEntity myMethodName(@RequestParam Map<String, String> parameters
    ) {
        /* DO YOU GET ANY PARAMETERS HERE?  Or are they empty because of logging/auditing filter ?*/
        return new ResponseEntity(HttpStatus.OK);

    }

我不得不浏览了这里的其他几个例子。

我想出了一个“包装器”,专门针对APPLICATION_FORM_URLENCODED_VALUE。

import org.apache.commons.io.IOUtils;
import org.springframework.http.MediaType;
import org.springframework.web.util.ContentCachingRequestWrapper;

import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import java.io.IOException;

/**
 * Makes a "copy" of the HttpRequest so the body can be accessed more than 1 time.
 * WORKS WITH APPLICATION_FORM_URLENCODED_VALUE
 * See : https://dev59.com/aFcP5IYBdhLWcg3wkK3U#44187955
 */
public final class AppFormUrlEncodedSpecificContentCachingRequestWrapper extends ContentCachingRequestWrapper {

    public static final String ERROR_MSG_CONTENT_TYPE_NOT_SUPPORTED = "ContentType not supported. (Input ContentType(s)=\"%1$s\", Supported ContentType(s)=\"%2$s\")";

    public static final String ERROR_MSG_PERSISTED_CONTENT_CACHING_REQUEST_WRAPPER_CONSTRUCTOR_FAILED = "AppFormUrlEncodedSpecificContentCachingRequestWrapper constructor failed";

    private static final org.slf4j.Logger LOGGER = org.slf4j.LoggerFactory.getLogger(AppFormUrlEncodedSpecificContentCachingRequestWrapper.class);

    private byte[] body;

    private ServletInputStream inputStream;

    public AppFormUrlEncodedSpecificContentCachingRequestWrapper(HttpServletRequest request) {
        super(request);
        super.getParameterMap(); // init cache in ContentCachingRequestWrapper.  THIS IS THE VITAL CALL so that "@RequestParam Map<String, String> parameters" are populated on the REST Controller.  See https://dev59.com/KWkv5IYBdhLWcg3w_Vun#64924380

        String contentType = request.getContentType();
        /* EXPLICTLY check for APPLICATION_FORM_URLENCODED_VALUE and allow nothing else */
        if (null == contentType || !contentType.equalsIgnoreCase(MediaType.APPLICATION_FORM_URLENCODED_VALUE)) {
            IllegalArgumentException ioex = new IllegalArgumentException(String.format(ERROR_MSG_CONTENT_TYPE_NOT_SUPPORTED, contentType, MediaType.APPLICATION_FORM_URLENCODED_VALUE));
            LOGGER.error(ERROR_MSG_PERSISTED_CONTENT_CACHING_REQUEST_WRAPPER_CONSTRUCTOR_FAILED, ioex);
            throw ioex;
        }

        try {
            loadBody(request);
        } catch (IOException ioex) {
            throw new RuntimeException(ioex);
        }
    }

    private void loadBody(HttpServletRequest request) throws IOException {
        body = IOUtils.toByteArray(request.getInputStream());
        inputStream = new CustomServletInputStream(this.getBody());
    }

    private byte[] getBody() {
        return body;
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {
        if (inputStream != null) {
            return inputStream;
        }
        return super.getInputStream();
    }
}

请参考这个页面上Andrew Sneck的回答。大致意思是:https://programmersought.com/article/23981013626/ 我还没有时间来协调以上两种实现(指我的两种实现)。
因此,我创建了一个工厂来“选择”这两个实现中的一个:
import org.springframework.http.MediaType;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.IOException;

/**
 * Factory to return different concretes of HttpServletRequestWrapper. APPLICATION_FORM_URLENCODED_VALUE needs a different concrete.
 */
public class HttpServletRequestWrapperFactory {

    public static final String ERROR_MSG_HTTP_SERVLET_REQUEST_WRAPPER_FACTORY_CREATE_HTTP_SERVLET_REQUEST_WRAPPER_FAILED = "HttpServletRequestWrapperFactory createHttpServletRequestWrapper FAILED";

    public static HttpServletRequestWrapper createHttpServletRequestWrapper(final HttpServletRequest request) {
        HttpServletRequestWrapper returnItem = null;

        if (null != request) {
            String contentType = request.getContentType();
            if (null != contentType && contentType.equalsIgnoreCase(MediaType.APPLICATION_FORM_URLENCODED_VALUE)) {
                returnItem = new AppFormUrlEncodedSpecificContentCachingRequestWrapper(request);
            } else {
                try {
                    returnItem = new PersistedBodyRequestWrapper(request);
                } catch (IOException ioex) {
                    throw new RuntimeException(ERROR_MSG_HTTP_SERVLET_REQUEST_WRAPPER_FACTORY_CREATE_HTTP_SERVLET_REQUEST_WRAPPER_FAILED, ioex);
                }
            }
        }

        return returnItem;
    }

}

下面这个是可以与JSON等配合使用的“其他”类型,它是工厂能够输出的另一种具体类型。我把它放在这里是为了让我的2021年1月追加部分保持一致。我不知道下面的代码是否完全符合我原来的答案:

import org.springframework.http.MediaType;

import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.nio.charset.StandardCharsets;
import java.util.Map;

/**
 * Makes a "copy" of the HttpRequest so the body can be accessed more than 1 time.
 * See : https://dev59.com/aFcP5IYBdhLWcg3wkK3U#44187955
 * DOES NOT WORK WITH APPLICATION_FORM_URLENCODED_VALUE
 */
public final class PersistedBodyRequestWrapper extends HttpServletRequestWrapper {

    public static final String ERROR_MSG_CONTENT_TYPE_NOT_SUPPORTED = "ContentType not supported. (ContentType=\"%1$s\")";

    public static final String ERROR_MSG_PERSISTED_BODY_REQUEST_WRAPPER_CONSTRUCTOR_FAILED = "PersistedBodyRequestWrapper constructor FAILED";

    private static final org.slf4j.Logger LOGGER = org.slf4j.LoggerFactory.getLogger(PersistedBodyRequestWrapper.class);

    private String persistedBody;

    private final Map<String, String[]> parameterMap;

    public PersistedBodyRequestWrapper(final HttpServletRequest request) throws IOException {
        super(request);

        String contentType = request.getContentType();
        /* Allow everything EXCEPT APPLICATION_FORM_URLENCODED_VALUE */
        if (null != contentType && contentType.equalsIgnoreCase(MediaType.APPLICATION_FORM_URLENCODED_VALUE)) {
            IllegalArgumentException ioex = new IllegalArgumentException(String.format(ERROR_MSG_CONTENT_TYPE_NOT_SUPPORTED, MediaType.APPLICATION_FORM_URLENCODED_VALUE));
            LOGGER.error(ERROR_MSG_PERSISTED_BODY_REQUEST_WRAPPER_CONSTRUCTOR_FAILED, ioex);
            throw ioex;
        }

        parameterMap = request.getParameterMap();
        this.persistedBody = "";
        BufferedReader bufferedReader = request.getReader();
        String line;
        while ((line = bufferedReader.readLine()) != null) {
            this.persistedBody += line;
        }
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {
        CustomServletInputStream csis = new CustomServletInputStream(this.persistedBody.getBytes(StandardCharsets.UTF_8));
        return csis;
    }

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

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

@KnockingHeads。太多巫术了。:( - granadaCoder
非常感谢您,让我理解了一个重要的概念。我只是好奇当我们编写请求包装器时,是否也需要包装响应? - agaonsindhe
我不是“超级了解”,但我强烈感觉“响应不行”。响应似乎是更多“完整有效载荷,非FHIR软管阅读器”类型的对象。 - granadaCoder

9
上述答案非常有帮助,但在我的经验中仍存在一些问题。在Tomcat 7 Servlet 3.0上,getParamter和getParamterValues也必须被覆盖。此解决方案包括get查询参数和post-body。它允许轻松获取原始字符串。
与其他解决方案一样,它使用Apache commons-io和Google的Guava。
在这个解决方案中,getParameter*方法不会抛出IOException,而是使用super.getInputStream()(获取主体),这可能会抛出IOException。我捕获它并抛出RuntimeException。这不太好。
import com.google.common.collect.Iterables;
import com.google.common.collect.ObjectArrays;

import org.apache.commons.io.IOUtils;
import org.apache.http.NameValuePair;
import org.apache.http.client.utils.URLEncodedUtils;
import org.apache.http.entity.ContentType;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.nio.charset.Charset;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;

/**
 * Purpose of this class is to make getParameter() return post data AND also be able to get entire
 * body-string. In native implementation any of those two works, but not both together.
 */
public class MultiReadHttpServletRequest extends HttpServletRequestWrapper {
    public static final String UTF8 = "UTF-8";
    public static final Charset UTF8_CHARSET = Charset.forName(UTF8);
    private ByteArrayOutputStream cachedBytes;
    private Map<String, String[]> parameterMap;

    public MultiReadHttpServletRequest(HttpServletRequest request) {
        super(request);
    }

    public static void toMap(Iterable<NameValuePair> inputParams, Map<String, String[]> toMap) {
        for (NameValuePair e : inputParams) {
            String key = e.getName();
            String value = e.getValue();
            if (toMap.containsKey(key)) {
                String[] newValue = ObjectArrays.concat(toMap.get(key), value);
                toMap.remove(key);
                toMap.put(key, newValue);
            } else {
                toMap.put(key, new String[]{value});
            }
        }
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {
        if (cachedBytes == null) cacheInputStream();
        return new CachedServletInputStream();
    }

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

    private void cacheInputStream() throws IOException {
    /* Cache the inputStream in order to read it multiple times. For
     * convenience, I use apache.commons IOUtils
     */
        cachedBytes = new ByteArrayOutputStream();
        IOUtils.copy(super.getInputStream(), cachedBytes);
    }

    @Override
    public String getParameter(String key) {
        Map<String, String[]> parameterMap = getParameterMap();
        String[] values = parameterMap.get(key);
        return values != null && values.length > 0 ? values[0] : null;
    }

    @Override
    public String[] getParameterValues(String key) {
        Map<String, String[]> parameterMap = getParameterMap();
        return parameterMap.get(key);
    }

    @Override
    public Map<String, String[]> getParameterMap() {
        if (parameterMap == null) {
            Map<String, String[]> result = new LinkedHashMap<String, String[]>();
            decode(getQueryString(), result);
            decode(getPostBodyAsString(), result);
            parameterMap = Collections.unmodifiableMap(result);
        }
        return parameterMap;
    }

    private void decode(String queryString, Map<String, String[]> result) {
        if (queryString != null) toMap(decodeParams(queryString), result);
    }

    private Iterable<NameValuePair> decodeParams(String body) {
        Iterable<NameValuePair> params = URLEncodedUtils.parse(body, UTF8_CHARSET);
        try {
            String cts = getContentType();
            if (cts != null) {
                ContentType ct = ContentType.parse(cts);
                if (ct.getMimeType().equals(ContentType.APPLICATION_FORM_URLENCODED.getMimeType())) {
                    List<NameValuePair> postParams = URLEncodedUtils.parse(IOUtils.toString(getReader()), UTF8_CHARSET);
                    params = Iterables.concat(params, postParams);
                }
            }
        } catch (IOException e) {
            throw new IllegalStateException(e);
        }
        return params;
    }

    public String getPostBodyAsString() {
        try {
            if (cachedBytes == null) cacheInputStream();
            return cachedBytes.toString(UTF8);
        } catch (UnsupportedEncodingException e) {
            throw new RuntimeException(e);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    /* An inputStream which reads the cached request body */
    public class CachedServletInputStream extends ServletInputStream {
        private ByteArrayInputStream input;

        public CachedServletInputStream() {
            /* create a new input stream from the cached request body */
            input = new ByteArrayInputStream(cachedBytes.toByteArray());
        }

        @Override
        public int read() throws IOException {
            return input.read();
        }
    }

    @Override
    public String toString() {
        String query = dk.bnr.util.StringUtil.nullToEmpty(getQueryString());
        StringBuilder sb = new StringBuilder();
        sb.append("URL='").append(getRequestURI()).append(query.isEmpty() ? "" : "?" + query).append("', body='");
        sb.append(getPostBodyAsString());
        sb.append("'");
        return sb.toString();
    }
}

太好了!我已经试图解决这个问题好几天了,而且这个方法适用于servlet 3.1。有一个问题:为什么在getParameterMap()中要执行decode(getPostBodyAsString(), result);?这会创建一个键为请求正文和值为null的参数,这相当奇怪。 - orlade
为什么不在您的getParameterMap中调用super.getParameterMap()呢?这样可以获得一个<String,String[]>类型的映射,而无需进行所有字符串解析。 - Ean V
我在paramMap方面也遇到了一些问题。请看我的回答以及2021年1月的“APPEND”。那里有一行神奇的代码(来自另一个问题的答案)...这可能是不必手动保留parameterMaps的“触发器”。代码的神奇部分是super.getParameterMap(); //在ContentCachingRequestWrapper中初始化缓存。 这个关键的调用会使REST Controller上的“@RequestParam Map<String, String> parameters”被填充。 - granadaCoder
在multipart/form-data的情况下,当调用getParameterMap进行缓存时,会出现“org.springframework.web.multipart.support.MissingServletRequestPartException: Required request part 'file' is not present”的错误。而如果不进行缓存,则可以正常工作,但请求体为空字符串。 - Gourav Kumar

7
我发现了一个适用于任何请求体格式的好解决方案。我已经测试了application/x-www-form-urlencodedapplication/json,两者都非常有效。问题在于ContentCachingRequestWrapper只设计用于x-www-form-urlencoded请求体,而不能与例如json一起使用。我找到了json的解决方案link。它的问题是不支持x-www-form-urlencoded。我将两者结合在我的代码中:
import org.apache.commons.io.IOUtils;
import org.springframework.web.util.ContentCachingRequestWrapper;

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

public class MyContentCachingRequestWrapper extends ContentCachingRequestWrapper {

    private byte[] body;

    public MyContentCachingRequestWrapper(HttpServletRequest request) throws IOException {
        super(request);
        super.getParameterMap(); // init cache in ContentCachingRequestWrapper
        body = super.getContentAsByteArray(); // first option for application/x-www-form-urlencoded
        if (body.length == 0) {
          try {
            body = IOUtils.toByteArray(super.getInputStream()); // second option for other body formats
          } catch (IOException ex) {
            body = new byte[0];
          }
        }
    }

    public byte[] getBody() {
        return body;
    }

    @Override
    public ServletInputStream getInputStream() {
        return new RequestCachingInputStream(body);
    }

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

    private static class RequestCachingInputStream extends ServletInputStream {

        private final ByteArrayInputStream inputStream;

        public RequestCachingInputStream(byte[] bytes) {
            inputStream = new ByteArrayInputStream(bytes);
        }

        @Override
        public int read() throws IOException {
            return inputStream.read();
        }

        @Override
        public boolean isFinished() {
            return inputStream.available() == 0;
        }

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

        @Override
        public void setReadListener(ReadListener readlistener) {
        }

    }

}

你调用 "super.getParameterMap();" 对我来说真是太神奇了。 - granadaCoder
我更新了我的回答...但是你的回答(我点赞了)有一个神奇的代码行...它帮助了我的代码。并且你的代码旁边有一个恰当的注释。谢谢。super.getParameterMap(); // 在ContentCachingRequestWrapper中初始化缓存 - granadaCoder
1
@granadaCoder,你应该点击超级方法并查看它正在做什么。 - Andrew Sneck
@AndrewSneck 非常感谢!它对application/x-www-form-urlencoded和application/json都有效。 - RAZAFINARIVO Hanania
不支持multipart/form-data格式。 - Gourav Kumar

6
唯一的方法是在过滤器中自己消耗整个输入流,从中提取所需内容,然后为您读取的内容创建一个新的InputStream,并将其放入ServletRequestWrapper(或HttpServletRequestWrapper)中。
缺点是你必须自己解析数据,标准并没有向你提供这种能力。
附录-
正如我所说,您需要查看HttpServletRequestWrapper。
在过滤器中,您继续调用FilterChain.doFilter(request,response)。
对于简单的过滤器,请求和响应与传递给过滤器的请求和响应相同。但不必如此。您可以用自己的请求和/或响应替换它们。
HttpServletRequestWrapper专门设计用于促进此操作。您将原始请求传递给它,然后可以拦截所有调用。您创建自己的此类子类,并将getInputStream方法替换为自己的方法。您无法更改原始请求的输入流,因此您有此包装器并返回自己的输入流。
最简单的情况是将原始请求的输入流消耗到字节数组缓冲区中,在其中进行所需的操作,然后从该缓冲区创建一个新的ByteArrayInputStream。这就是在您的包装器中返回的内容,该内容传递给FilterChain.doFilter方法。
您需要子类化ServletInputStream并为ByteArrayInputStream制作另一个包装器,但这也不是什么大问题。

我无法成功读取InputStream并在之后恢复它,因为没有get/set方法可以直接访问该流。您的建议听起来不错,但我不知道如何实现它。 - amuniz

6

我也遇到了同样的问题,我相信下面的代码更简单,而且对我来说它是有效的:

public class MultiReadHttpServletRequest extends  HttpServletRequestWrapper {

 private String _body;

public MultiReadHttpServletRequest(HttpServletRequest request) throws IOException {
   super(request);
   _body = "";
   BufferedReader bufferedReader = request.getReader();           
   String line;
   while ((line = bufferedReader.readLine()) != null){
       _body += line;
   }
}

@Override
public ServletInputStream getInputStream() throws IOException {
   final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(_body.getBytes());
   return new ServletInputStream() {
       public int read() throws IOException {
           return byteArrayInputStream.read();
       }
   };
}

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

在 Java 类的过滤器中,
HttpServletRequest properRequest = ((HttpServletRequest) req);
MultiReadHttpServletRequest wrappedRequest = new MultiReadHttpServletRequest(properRequest);
req = wrappedRequest;
inputJson = IOUtils.toString(req.getReader());
System.out.println("body"+inputJson);

如果您有任何疑问,请告知我。


首先,感谢您!其次,这种方法有什么缺点吗?会增加内存使用或其他什么吗?我问这个问题是因为我不明白为什么在Spring中这么难实现,他们为什么不把这部分作为框架本身的一部分呢?我正在使用Spring构建一个简单的REST API,所以如果担心二进制文件被保存在内存中的话,应该没有风险。 - BoomShaka

3

Spring框架内置了AbstractRequestLoggingFilter来支持此功能:

@Bean
public Filter loggingFilter(){
    final AbstractRequestLoggingFilter filter = new AbstractRequestLoggingFilter() {
        @Override
        protected void beforeRequest(final HttpServletRequest request, final String message) {

        }

        @Override
        protected void afterRequest(final HttpServletRequest request, final String message) {

        }
    };

    filter.setIncludePayload(true);
    filter.setIncludeQueryString(false);
    filter.setMaxPayloadLength(1000000);

    return filter;
}

不幸的是,您仍然无法直接从请求中读取有效负载,但是字符串消息参数将包含有效负载,因此您可以像下面这样从中获取:

String body = message.substring(message.indexOf("{"), message.lastIndexOf("]"));


我希望使用您的解决方案来生成审计日志,但我需要记录请求是否成功,我能否钩入HTTP响应并获取此类中的代码。 - jonesy

1

仅仅覆盖 getInputStream() 方法在我的情况下不起作用。我的服务器实现似乎会在不调用此方法的情况下解析参数。我没有找到其他方法,只能重新实现所有四个 getParameter* 方法。这里是 getParameterMap 的代码(使用 Apache Http Client 和 Google Guava 库):

@Override
public Map<String, String[]> getParameterMap() {
    Iterable<NameValuePair> params = URLEncodedUtils.parse(getQueryString(), NullUtils.UTF8);

    try {
        String cts = getContentType();
        if (cts != null) {
            ContentType ct = ContentType.parse(cts);
            if (ct.getMimeType().equals(ContentType.APPLICATION_FORM_URLENCODED.getMimeType())) {
                List<NameValuePair> postParams = URLEncodedUtils.parse(IOUtils.toString(getReader()), NullUtils.UTF8);
                params = Iterables.concat(params, postParams);
            }
        }
    } catch (IOException e) {
        throw new IllegalStateException(e);
    }
    Map<String, String[]> result = toMap(params);
    return result;
}

public static Map<String, String[]> toMap(Iterable<NameValuePair> body) {
    Map<String, String[]> result = new LinkedHashMap<>();
    for (NameValuePair e : body) {
        String key = e.getName();
        String value = e.getValue();
        if (result.containsKey(key)) {
            String[] newValue = ObjectArrays.concat(result.get(key), value);
            result.remove(key);
            result.put(key, newValue);
        } else {
            result.put(key, new String[] {value});
        }
    }
    return result;
}

1
Jetty很不幸地存在这个问题,http://grepcode.com/file/repo1.maven.org/maven2/org.eclipse.jetty/jetty-server/9.3.0.RC1/org/eclipse/jetty/server/Request.java - mikeapr4
您可能正在使用带有Servlet 3.0的Tomcat 7或更高版本?您还有其他3种方法的代码吗? - Silver
其他三种方法只需调用getParameterMap()并获取所需的值。 - 30thh
我也遇到了paramMap的一些问题。请看我的答案以及2021年1月的“APPEND”。那里有一行神奇的代码(来自这个问题的另一个答案)...可能是不必手动保留参数映射的“触发器”。代码的神奇部分是super.getParameterMap(); //在ContentCachingRequestWrapper中初始化缓存。这是重要的调用,使得REST控制器上的“@RequestParam Map<String, String> parameters”被填充。 - granadaCoder

1
Spring类ContentCachingRequestWrapper的方法getContentAsByteArray()会读取请求体多次,但是该类的getInputStream()和getReader()方法不会多次读取请求体:
“这个类通过消耗InputStream来缓存请求体。如果我们在过滤器中读取了InputStream,那么过滤器链中的其他后续过滤器就不能再读取它了。由于这种限制,这个类并不适用于所有情况。”
在我的情况下,解决这个问题的更普遍的方法是将以下三个类添加到我的Spring Boot项目中(以及pom文件中所需的依赖项):

CachedBodyHttpServletRequest.java:

public class CachedBodyHttpServletRequest extends HttpServletRequestWrapper {

    private byte[] cachedBody;

    public CachedBodyHttpServletRequest(HttpServletRequest request) throws IOException {
        super(request);
        InputStream requestInputStream = request.getInputStream();
        this.cachedBody = StreamUtils.copyToByteArray(requestInputStream);
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {
        return new CachedBodyServletInputStream(this.cachedBody);
    }

    @Override
    public BufferedReader getReader() throws IOException {
        // Create a reader from cachedContent
        // and return it
        ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(this.cachedBody);
        return new BufferedReader(new InputStreamReader(byteArrayInputStream));
    }
}

CachedBodyServletInputStream.java:

public class CachedBodyServletInputStream extends ServletInputStream {

    private InputStream cachedBodyInputStream;

    public CachedBodyServletInputStream(byte[] cachedBody) {
        this.cachedBodyInputStream = new ByteArrayInputStream(cachedBody);
    }

    @Override
    public boolean isFinished() {
        try {
            return cachedBodyInputStream.available() == 0;
        } catch (IOException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
        return false;
    }

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

    @Override
    public void setReadListener(ReadListener readListener) {
        throw new UnsupportedOperationException();
    }

    @Override
    public int read() throws IOException {
        return cachedBodyInputStream.read();
    }
}

ContentCachingFilter.java:

@Order(value = Ordered.HIGHEST_PRECEDENCE)
@Component
@WebFilter(filterName = "ContentCachingFilter", urlPatterns = "/*")
public class ContentCachingFilter extends OncePerRequestFilter {

    @Override
    protected void doFilterInternal(HttpServletRequest httpServletRequest, HttpServletResponse httpServletResponse, FilterChain filterChain) throws ServletException, IOException {
        System.out.println("IN  ContentCachingFilter ");
        CachedBodyHttpServletRequest cachedBodyHttpServletRequest = new CachedBodyHttpServletRequest(httpServletRequest);
        filterChain.doFilter(cachedBodyHttpServletRequest, httpServletResponse);
    }
}

我还在pom文件中添加了以下依赖项:

<dependency>
    <groupId>org.springframework</groupId>
    <artifactId>spring-webmvc</artifactId>
    <version>5.2.0.RELEASE</version>
</dependency>
<dependency>
    <groupId>javax.servlet</groupId>
    <artifactId>javax.servlet-api</artifactId>
    <version>4.0.1</version>
</dependency>
<dependency>
    <groupId>com.fasterxml.jackson.core</groupId>
    <artifactId>jackson-databind</artifactId>
    <version>2.10.0</version>
</dependency>

这里提供了教程和完整源代码: https://www.baeldung.com/spring-reading-httpservletrequest-multiple-times


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接