Skip to content

Commit 71947ae

Browse files
authored
fix: send unit header in gw. (#1758)
1 parent 3756aee commit 71947ae

File tree

3 files changed

+13
-0
lines changed

3 files changed

+13
-0
lines changed

CHANGELOG.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,3 +8,4 @@
88
- [fix:replace with string inside @ConditionalOnClass.](https://github.com/Tencent/spring-cloud-tencent/pull/1750)
99
- [feat: support TagUtils, ContextToHeaderInterceptor in TSF.](https://github.com/Tencent/spring-cloud-tencent/pull/1754)
1010
- [deps:upgrade to spring boot 3.4.12.](https://github.com/Tencent/spring-cloud-tencent/pull/1756)
11+
- [fix: send unit header in tsf gw.](https://github.com/Tencent/spring-cloud-tencent/pull/1758)

spring-cloud-tencent-plugin-starters/spring-cloud-tencent-unit-plugin/src/main/java/com/tencent/cloud/plugin/unit/config/GatewayUnitAutoConfiguration.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
package com.tencent.cloud.plugin.unit.config;
1919

2020
import com.tencent.cloud.common.tsf.ConditionalOnOnlyTsfConsulEnabled;
21+
import com.tencent.cloud.plugin.unit.plugin.UnitScgEnhancedPlugin;
2122
import com.tencent.tsf.unit.core.GatewayUnitArchCallback;
2223
import com.tencent.tsf.unit.core.TencentUnitManager;
2324
import com.tencent.tsf.unit.core.TsfZoneFilterUnitCallback;
@@ -26,6 +27,7 @@
2627

2728
import org.springframework.beans.factory.annotation.Value;
2829
import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
30+
import org.springframework.context.annotation.Bean;
2931
import org.springframework.context.annotation.Configuration;
3032

3133

@@ -44,6 +46,11 @@ static class GatewayUnitEnable {
4446
private String applicationName;
4547

4648

49+
@Bean
50+
public UnitScgEnhancedPlugin unitScgEnhancedPlugin() {
51+
return new UnitScgEnhancedPlugin();
52+
}
53+
4754
@PostConstruct
4855
public void init() {
4956
TencentUnitManager.addArchCallback(new GatewayUnitArchCallback(applicationName));

spring-cloud-tencent-plugin-starters/spring-cloud-tencent-unit-plugin/src/main/java/com/tencent/cloud/plugin/unit/plugin/UnitScgEnhancedPlugin.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@
2727
import com.tencent.cloud.rpc.enhancement.plugin.EnhancedPluginType;
2828
import com.tencent.cloud.rpc.enhancement.plugin.PluginOrderConstant;
2929
import com.tencent.tsf.unit.core.TencentUnitContext;
30+
import com.tencent.tsf.unit.core.TencentUnitManager;
3031
import shade.polaris.com.google.common.collect.ImmutableMap;
3132

3233
import org.springframework.http.server.reactive.ServerHttpRequest;
@@ -50,6 +51,10 @@ public void run(EnhancedPluginContext context) throws Throwable {
5051
return;
5152
}
5253

54+
if (!TencentUnitManager.isEnable()) {
55+
return;
56+
}
57+
5358
// get request builder
5459
ServerHttpRequest.Builder builder = exchange.getRequest().mutate();
5560

0 commit comments

Comments
 (0)