Servlet过滤器使用Servlet请求包装器修改标题值无效。

Question

我试图在请求中更改Content-Type头信息，将其更改为"application/json"，以便在到达我的Spring Rest控制器之前进行更改。我创建了一个ServletRequestWrapper来更改值，但当请求到达控制器时，它仍然是"text/plain"。日志记录显示，在经过doFilter()之前已更改头信息的值。

以下是扩展HttpServletRequestWrapper类的代码：

class HttpServletRequestWritableWrapper extends HttpServletRequestWrapper {
private final Logger logger = org.slf4j.LoggerFactory.getLogger(HttpServletRequestWritableWrapper.class);
private final ByteArrayInputStream decryptedBody;

HttpServletRequestWritableWrapper(HttpServletRequest request, byte[] decryptedData) {
    super(request);
    decryptedBody = new ByteArrayInputStream(decryptedData);
}



@Override
public String getHeader(String name) {
    String headerValue = super.getHeader(name);
    if("Accept".equalsIgnoreCase(name))
    {
        logger.debug("Accept header changing :");
        return headerValue.replaceAll(
                MediaType.TEXT_PLAIN_VALUE, MediaType.APPLICATION_JSON_VALUE
        );
    }
    else if ("Content-Type".equalsIgnoreCase(name))
    {
        logger.debug("Content type change: ");
        return headerValue.replaceAll(MediaType.TEXT_PLAIN_VALUE, MediaType.APPLICATION_JSON_VALUE);
    }
    return headerValue;
}

@Override
public Enumeration<String> getHeaderNames() {
    return super.getHeaderNames();
}

@Override
public String getContentType() {
    String contentTypeValue = super.getContentType();
    if (MediaType.TEXT_PLAIN_VALUE.equalsIgnoreCase(contentTypeValue)) {
        logger.debug("Changing on getContentType():");
        return MediaType.APPLICATION_JSON_VALUE;
    }
    return contentTypeValue;
}


@Override
public BufferedReader getReader() throws UnsupportedEncodingException {
    return new BufferedReader(new InputStreamReader(decryptedBody, UTF_8));
}

@Override
public ServletInputStream getInputStream() throws IOException {
    return new ServletInputStream() {
        @Override
        public int read() {
            return decryptedBody.read();
        }
    };
}

以下是我的筛选器：

@WebFilter(displayName = "EncryptionFilter", urlPatterns = "/*")
public class EncryptionFilter implements Filter {
private final Logger logger = org.slf4j.LoggerFactory.getLogger(EncryptionFilter.class);

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

}

@Override
public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
    byte[] data = "{\"currentClientVersion\":{\"majorElement\":\"1\",\"minorElement\":\"2\"}}".getBytes();

    logger.debug("data string " + data.toString());

    logger.debug("Content-type before: " + servletRequest.getContentType());

    HttpServletRequest request = (HttpServletRequest) servletRequest;
    HttpServletRequestWritableWrapper requestWrapper = new HttpServletRequestWritableWrapper(request, data);
    //logger.debug("Accept Header: " + requestWrapper.getHeader("Accept"));
    //logger.debug("Content-Type: " + requestWrapper.getHeader("Content-Type"));
    //logger.debug("Contenttype" + requestWrapper.getContentType());

    filterChain.doFilter(requestWrapper, servletResponse);


}

@Override
public void destroy() {

}


}

- ajaxon

Answer 1

看起来在我的过滤器之后某处调用了getHeaders方法，并未返回使用我更新的值。

我在我的HttpServletRequestWrapper中添加了这个覆盖，现在它可以正常工作：

@Override
public Enumeration<String> getHeaders(String name) {
    List<String> headerVals = Collections.list(super.getHeaders(name));
    int index = 0;
    for (String value : headerVals) {
        if ("Content-Type".equalsIgnoreCase(name)) {
            logger.debug("Content type change: ");
            headerVals.set(index, MediaType.APPLICATION_JSON_VALUE);
        }

        index++;
    }

    return Collections.enumeration(headerVals);
}

- ajaxon