FizzWebClient support x-forwarded-for header

This commit is contained in:
lancer.hong
2022-11-25 22:10:31 +08:00
parent 2d45c5d61e
commit 011e25a280
3 changed files with 14 additions and 0 deletions

View File

@@ -125,6 +125,9 @@ fizz:
fast-fail-when-registry-center-down: false fast-fail-when-registry-center-down: false
web-client:
x-forwarded-for: false
# dedicated-line: # dedicated-line:
# server: # server:
# enable: true # enable: true

View File

@@ -132,6 +132,13 @@ public class SystemConfig {
@Value("${fizz.fast-fail-when-registry-center-down:false}") @Value("${fizz.fast-fail-when-registry-center-down:false}")
private boolean fastFailWhenRegistryCenterDown; private boolean fastFailWhenRegistryCenterDown;
@Value("${fizz.web-client.x-forwarded-for:false}")
private boolean fizzWebClientXForwardedFor;
public boolean isFizzWebClientXForwardedFor() {
return fizzWebClientXForwardedFor;
}
public boolean isFastFailWhenRegistryCenterDown() { public boolean isFastFailWhenRegistryCenterDown() {
return fastFailWhenRegistryCenterDown; return fastFailWhenRegistryCenterDown;
} }

View File

@@ -39,6 +39,7 @@ import we.exception.ExternalService4xxException;
import we.fizz.exception.FizzRuntimeException; import we.fizz.exception.FizzRuntimeException;
import we.service_registry.RegistryCenterService; import we.service_registry.RegistryCenterService;
import we.util.Consts; import we.util.Consts;
import we.util.NetworkUtils;
import we.util.ThreadContext; import we.util.ThreadContext;
import we.util.WebUtils; import we.util.WebUtils;
@@ -236,6 +237,9 @@ public class FizzWebClient {
); );
} }
setHostHeader(uri, hdrs); setHostHeader(uri, hdrs);
if (systemConfig.isFizzWebClientXForwardedFor()) {
hdrs.add(com.google.common.net.HttpHeaders.X_FORWARDED_FOR, NetworkUtils.getServerIp());
}
} }
); );