From 6e359c5e6b2b4c3dedba7ab4ccd3d14b4c443d9c Mon Sep 17 00:00:00 2001 From: Francis Dong Date: Tue, 14 Feb 2023 10:55:52 +0800 Subject: [PATCH] Set X-Forwarded-For header --- .../src/main/java/com/fizzgate/filter/AggregateFilter.java | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/fizz-core/src/main/java/com/fizzgate/filter/AggregateFilter.java b/fizz-core/src/main/java/com/fizzgate/filter/AggregateFilter.java index 2eee103..96a73fc 100644 --- a/fizz-core/src/main/java/com/fizzgate/filter/AggregateFilter.java +++ b/fizz-core/src/main/java/com/fizzgate/filter/AggregateFilter.java @@ -134,11 +134,10 @@ public class AggregateFilter implements WebFilter { Input input = aggregateResource.getInput(); HttpHeaders hds = request.getHeaders(); - if (CollectionUtils.isEmpty(hds.get(X_FORWARDED_FOR)) && systemConfig.isFizzWebClientXForwardedForEnable()) { - hds.add(X_FORWARDED_FOR, WebUtils.getOriginIp(exchange)); - } - Map headers = MapUtil.headerToHashMap(hds); + if (CollectionUtils.isEmpty(hds.get(X_FORWARDED_FOR)) && systemConfig.isFizzWebClientXForwardedForEnable()) { + headers.put(X_FORWARDED_FOR, WebUtils.getOriginIp(exchange)); + } Map fizzHeaders = (Map) exchange.getAttributes().get(WebUtils.APPEND_HEADERS); if (fizzHeaders != null && !fizzHeaders.isEmpty()) { Set> entrys = fizzHeaders.entrySet();