SPR-6759 - Jetty 7 doesn't like ShallowEtagHeaderFilter
This commit is contained in:
parent
1bbe93e535
commit
0a4a09a09d
|
|
@ -57,6 +57,9 @@ public class ShallowEtagHeaderFilter extends OncePerRequestFilter {
|
|||
filterChain.doFilter(request, responseWrapper);
|
||||
|
||||
byte[] body = responseWrapper.toByteArray();
|
||||
int statusCode = responseWrapper.getStatusCode();
|
||||
|
||||
if (isEligibleForEtag(request, responseWrapper, statusCode, body)) {
|
||||
String responseETag = generateETagHeaderValue(body);
|
||||
response.setHeader(HEADER_ETAG, responseETag);
|
||||
|
||||
|
|
@ -65,6 +68,7 @@ public class ShallowEtagHeaderFilter extends OncePerRequestFilter {
|
|||
if (logger.isTraceEnabled()) {
|
||||
logger.trace("ETag [" + responseETag + "] equal to If-None-Match, sending 304");
|
||||
}
|
||||
response.setContentLength(0);
|
||||
response.setStatus(HttpServletResponse.SC_NOT_MODIFIED);
|
||||
}
|
||||
else {
|
||||
|
|
@ -72,11 +76,42 @@ public class ShallowEtagHeaderFilter extends OncePerRequestFilter {
|
|||
logger.trace("ETag [" + responseETag + "] not equal to If-None-Match [" + requestETag +
|
||||
"], sending normal response");
|
||||
}
|
||||
copyBodyToResponse(body, response);
|
||||
}
|
||||
}
|
||||
else {
|
||||
if (logger.isTraceEnabled()) {
|
||||
logger.trace("Response with status code [" + statusCode + "] not eligible for ETag");
|
||||
}
|
||||
copyBodyToResponse(body, response);
|
||||
}
|
||||
}
|
||||
|
||||
private void copyBodyToResponse(byte[] body, HttpServletResponse response) throws IOException {
|
||||
response.setContentLength(body.length);
|
||||
if (body.length > 0) {
|
||||
FileCopyUtils.copy(body, response.getOutputStream());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Indicates whether the given request and response are eligible for ETag generation.
|
||||
*
|
||||
* <p>Default implementation returns {@code true} for response status codes in the {@code 2xx} series.
|
||||
*
|
||||
* @param request the HTTP request
|
||||
* @param response the HTTP response
|
||||
* @param responseStatusCode the HTTP response status code
|
||||
* @param responseBody the response body
|
||||
* @return {@code true} if eligible for ETag generation; {@code false} otherwise
|
||||
*/
|
||||
protected boolean isEligibleForEtag(HttpServletRequest request,
|
||||
HttpServletResponse response,
|
||||
int responseStatusCode,
|
||||
byte[] responseBody) {
|
||||
return (responseStatusCode >= 200 && responseStatusCode < 300);
|
||||
}
|
||||
|
||||
/**
|
||||
* Generate the ETag header value from the given response body byte array. <p>The default implementation generates an
|
||||
* MD5 hash.
|
||||
|
|
@ -105,10 +140,36 @@ public class ShallowEtagHeaderFilter extends OncePerRequestFilter {
|
|||
|
||||
private PrintWriter writer;
|
||||
|
||||
private int statusCode = -1;
|
||||
|
||||
private ShallowEtagResponseWrapper(HttpServletResponse response) {
|
||||
super(response);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void setStatus(int sc) {
|
||||
super.setStatus(sc);
|
||||
this.statusCode = sc;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void setStatus(int sc, String sm) {
|
||||
super.setStatus(sc, sm);
|
||||
this.statusCode = sc;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void sendError(int sc) throws IOException {
|
||||
super.sendError(sc);
|
||||
this.statusCode = sc;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void sendError(int sc, String msg) throws IOException {
|
||||
super.sendError(sc, msg);
|
||||
this.statusCode = sc;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ServletOutputStream getOutputStream() {
|
||||
return this.outputStream;
|
||||
|
|
@ -135,6 +196,10 @@ public class ShallowEtagHeaderFilter extends OncePerRequestFilter {
|
|||
resetBuffer();
|
||||
}
|
||||
|
||||
private int getStatusCode() {
|
||||
return statusCode;
|
||||
}
|
||||
|
||||
private byte[] toByteArray() {
|
||||
return this.content.toByteArray();
|
||||
}
|
||||
|
|
|
|||
|
|
@ -40,6 +40,15 @@ public class ShallowEtagHeaderFilterTest {
|
|||
filter = new ShallowEtagHeaderFilter();
|
||||
}
|
||||
|
||||
@Test
|
||||
public void isEligibleForEtag() {
|
||||
MockHttpServletRequest request = new MockHttpServletRequest("GET", "/hotels");
|
||||
MockHttpServletResponse response = new MockHttpServletResponse();
|
||||
|
||||
assertTrue(filter.isEligibleForEtag(request, response, 200, new byte[0]));
|
||||
assertFalse(filter.isEligibleForEtag(request, response, 300, new byte[0]));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void filterNoMatch() throws Exception {
|
||||
final MockHttpServletRequest request = new MockHttpServletRequest("GET", "/hotels");
|
||||
|
|
|
|||
Loading…
Reference in New Issue