Remove header AFTER response is received.

This commit is contained in:
Spencer Gibb
2017-10-03 13:28:52 -04:00
parent b3db9d7586
commit 55ffc93a70

View File

@@ -19,6 +19,7 @@ package org.springframework.cloud.gateway.filter.factory;
import org.springframework.tuple.Tuple;
import org.springframework.web.server.WebFilter;
import reactor.core.publisher.Mono;
import java.util.Arrays;
import java.util.List;
@@ -38,10 +39,8 @@ public class SetResponseHeaderWebFilterFactory implements WebFilterFactory {
final String header = args.getString(NAME_KEY);
final String value = args.getString(VALUE_KEY);
return (exchange, chain) -> {
return (exchange, chain) -> chain.filter(exchange).then(Mono.fromRunnable(() -> {
exchange.getResponse().getHeaders().set(header, value);
return chain.filter(exchange);
};
}));
}
}