From ca4da3ea8b26455667aa29e7acdef77d66894e86 Mon Sep 17 00:00:00 2001 From: lbw Date: Sat, 18 Mar 2023 15:18:37 +0800 Subject: [PATCH] =?UTF-8?q?:sparkles:=20Introducing=20new=20features.=20to?= =?UTF-8?q?c=20header=20feign=20=E4=BC=A0=E9=80=92?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../PigxClientToCRequestInterceptor.java | 34 +++++++++++++++++++ .../feign/PigxFeignClientConfiguration.java | 5 +++ 2 files changed, 39 insertions(+) create mode 100644 pigx-common/pigx-common-security/src/main/java/com/pig4cloud/pigx/common/security/feign/PigxClientToCRequestInterceptor.java diff --git a/pigx-common/pigx-common-security/src/main/java/com/pig4cloud/pigx/common/security/feign/PigxClientToCRequestInterceptor.java b/pigx-common/pigx-common-security/src/main/java/com/pig4cloud/pigx/common/security/feign/PigxClientToCRequestInterceptor.java new file mode 100644 index 000000000..707bbcd91 --- /dev/null +++ b/pigx-common/pigx-common-security/src/main/java/com/pig4cloud/pigx/common/security/feign/PigxClientToCRequestInterceptor.java @@ -0,0 +1,34 @@ +package com.pig4cloud.pigx.common.security.feign; + +import cn.hutool.core.util.StrUtil; +import com.pig4cloud.pigx.common.core.constant.SecurityConstants; +import com.pig4cloud.pigx.common.core.util.WebUtils; +import feign.RequestInterceptor; +import feign.RequestTemplate; +import lombok.extern.slf4j.Slf4j; + +/** + * TOC 客户标识传递 + * + * @author lengleng + * @date 2023/3/17 + */ +@Slf4j +public class PigxClientToCRequestInterceptor implements RequestInterceptor { + + /** + * Called for every request. Add data using methods on the supplied + * {@link RequestTemplate}. + * @param template + */ + public void apply(RequestTemplate template) { + String reqVersion = WebUtils.getRequest() != null + ? WebUtils.getRequest().getHeader(SecurityConstants.HEADER_TOC) : null; + + if (StrUtil.isNotBlank(reqVersion)) { + log.debug("feign add header toc :{}", reqVersion); + template.header(SecurityConstants.HEADER_TOC, reqVersion); + } + } + +} diff --git a/pigx-common/pigx-common-security/src/main/java/com/pig4cloud/pigx/common/security/feign/PigxFeignClientConfiguration.java b/pigx-common/pigx-common-security/src/main/java/com/pig4cloud/pigx/common/security/feign/PigxFeignClientConfiguration.java index ee95bc346..5295620a1 100644 --- a/pigx-common/pigx-common-security/src/main/java/com/pig4cloud/pigx/common/security/feign/PigxFeignClientConfiguration.java +++ b/pigx-common/pigx-common-security/src/main/java/com/pig4cloud/pigx/common/security/feign/PigxFeignClientConfiguration.java @@ -32,4 +32,9 @@ public class PigxFeignClientConfiguration { return new PigxOAuthRequestInterceptor(tokenResolver); } + @Bean + public RequestInterceptor clientToCRequestInterceptor() { + return new PigxClientToCRequestInterceptor(); + } + }