|
@@ -27,7 +27,6 @@ import javax.servlet.http.HttpServletRequest;
|
|
import javax.servlet.http.HttpServletRequestWrapper;
|
|
import javax.servlet.http.HttpServletRequestWrapper;
|
|
import javax.servlet.http.HttpServletResponse;
|
|
import javax.servlet.http.HttpServletResponse;
|
|
|
|
|
|
-import org.springframework.security.web.header.writers.CompositeHeaderWriter;
|
|
|
|
import org.springframework.security.web.util.OnCommittedResponseWrapper;
|
|
import org.springframework.security.web.util.OnCommittedResponseWrapper;
|
|
import org.springframework.util.Assert;
|
|
import org.springframework.util.Assert;
|
|
import org.springframework.web.filter.OncePerRequestFilter;
|
|
import org.springframework.web.filter.OncePerRequestFilter;
|
|
@@ -49,7 +48,7 @@ public class HeaderWriterFilter extends OncePerRequestFilter {
|
|
* The {@link HeaderWriter} to write headers to the response.
|
|
* The {@link HeaderWriter} to write headers to the response.
|
|
* {@see CompositeHeaderWriter}
|
|
* {@see CompositeHeaderWriter}
|
|
*/
|
|
*/
|
|
- private final HeaderWriter headerWriter;
|
|
|
|
|
|
+ private final List<HeaderWriter> headerWriters;
|
|
|
|
|
|
/**
|
|
/**
|
|
* Indicates whether to write the headers at the beginning of the request.
|
|
* Indicates whether to write the headers at the beginning of the request.
|
|
@@ -64,7 +63,7 @@ public class HeaderWriterFilter extends OncePerRequestFilter {
|
|
*/
|
|
*/
|
|
public HeaderWriterFilter(List<HeaderWriter> headerWriters) {
|
|
public HeaderWriterFilter(List<HeaderWriter> headerWriters) {
|
|
Assert.notEmpty(headerWriters, "headerWriters cannot be null or empty");
|
|
Assert.notEmpty(headerWriters, "headerWriters cannot be null or empty");
|
|
- this.headerWriter = new CompositeHeaderWriter(headerWriters);
|
|
|
|
|
|
+ this.headerWriters = headerWriters;
|
|
}
|
|
}
|
|
|
|
|
|
@Override
|
|
@Override
|
|
@@ -80,13 +79,13 @@ public class HeaderWriterFilter extends OncePerRequestFilter {
|
|
}
|
|
}
|
|
|
|
|
|
private void doHeadersBefore(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws IOException, ServletException {
|
|
private void doHeadersBefore(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws IOException, ServletException {
|
|
- this.headerWriter.writeHeaders(request, response);
|
|
|
|
|
|
+ writeHeaders(request, response);
|
|
filterChain.doFilter(request, response);
|
|
filterChain.doFilter(request, response);
|
|
}
|
|
}
|
|
|
|
|
|
private void doHeadersAfter(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws IOException, ServletException {
|
|
private void doHeadersAfter(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws IOException, ServletException {
|
|
HeaderWriterResponse headerWriterResponse = new HeaderWriterResponse(request,
|
|
HeaderWriterResponse headerWriterResponse = new HeaderWriterResponse(request,
|
|
- response, this.headerWriter);
|
|
|
|
|
|
+ response);
|
|
HeaderWriterRequest headerWriterRequest = new HeaderWriterRequest(request,
|
|
HeaderWriterRequest headerWriterRequest = new HeaderWriterRequest(request,
|
|
headerWriterResponse);
|
|
headerWriterResponse);
|
|
try {
|
|
try {
|
|
@@ -96,6 +95,12 @@ public class HeaderWriterFilter extends OncePerRequestFilter {
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
|
|
+ void writeHeaders(HttpServletRequest request, HttpServletResponse response) {
|
|
|
|
+ for (HeaderWriter writer : this.headerWriters) {
|
|
|
|
+ writer.writeHeaders(request, response);
|
|
|
|
+ }
|
|
|
|
+ }
|
|
|
|
+
|
|
/**
|
|
/**
|
|
* Allow writing headers at the beginning of the request.
|
|
* Allow writing headers at the beginning of the request.
|
|
*
|
|
*
|
|
@@ -107,15 +112,12 @@ public class HeaderWriterFilter extends OncePerRequestFilter {
|
|
this.shouldWriteHeadersEagerly = shouldWriteHeadersEagerly;
|
|
this.shouldWriteHeadersEagerly = shouldWriteHeadersEagerly;
|
|
}
|
|
}
|
|
|
|
|
|
- static class HeaderWriterResponse extends OnCommittedResponseWrapper {
|
|
|
|
|
|
+ class HeaderWriterResponse extends OnCommittedResponseWrapper {
|
|
private final HttpServletRequest request;
|
|
private final HttpServletRequest request;
|
|
- private final HeaderWriter headerWriter;
|
|
|
|
|
|
|
|
- HeaderWriterResponse(HttpServletRequest request, HttpServletResponse response,
|
|
|
|
- HeaderWriter headerWriter) {
|
|
|
|
|
|
+ HeaderWriterResponse(HttpServletRequest request, HttpServletResponse response) {
|
|
super(response);
|
|
super(response);
|
|
this.request = request;
|
|
this.request = request;
|
|
- this.headerWriter = headerWriter;
|
|
|
|
}
|
|
}
|
|
|
|
|
|
/*
|
|
/*
|
|
@@ -134,7 +136,7 @@ public class HeaderWriterFilter extends OncePerRequestFilter {
|
|
if (isDisableOnResponseCommitted()) {
|
|
if (isDisableOnResponseCommitted()) {
|
|
return;
|
|
return;
|
|
}
|
|
}
|
|
- this.headerWriter.writeHeaders(this.request, getHttpResponse());
|
|
|
|
|
|
+ HeaderWriterFilter.this.writeHeaders(this.request, getHttpResponse());
|
|
}
|
|
}
|
|
|
|
|
|
private HttpServletResponse getHttpResponse() {
|
|
private HttpServletResponse getHttpResponse() {
|