SpringCloud/spring-cloud-gateway
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Only add headers if response is not committed.

Browse Source
This commit is contained in:
Spencer Gibb
2019-06-13 16:36:14 -04:00
parent 563cfb7479
commit f11d6a5088
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
spring-cloud-gateway-core/src/test/java/org/springframework/cloud/gateway/test/BaseWebClientTests.java
View File

@@ -97,11 +97,15 @@ public class BaseWebClientTests {
log.info("modifyResponseFilter start");
String value = exchange.getAttributeOrDefault(GATEWAY_HANDLER_MAPPER_ATTR,
"N/A");
exchange.getResponse().getHeaders().add(HANDLER_MAPPER_HEADER, value);
if (!exchange.getResponse().isCommitted()) {
exchange.getResponse().getHeaders().add(HANDLER_MAPPER_HEADER, value);
}
Route route = exchange.getAttributeOrDefault(GATEWAY_ROUTE_ATTR, null);
if (route != null) {
exchange.getResponse().getHeaders().add(ROUTE_ID_HEADER,
route.getId());
if (!exchange.getResponse().isCommitted()) {
exchange.getResponse().getHeaders().add(ROUTE_ID_HEADER,
route.getId());
}
}
return chain.filter(exchange);
};