流量标签透传插件：优化servicecomb 服务端拦截场景

sermant-plugins/sermant-tag-transmission/tag-transmission-plugin/src/main/java/com/huaweicloud/sermant/tag/transmission/interceptors/http/server/HttpServletInterceptor.java

@@ -17,9 +17,12 @@
 package com.huaweicloud.sermant.tag.transmission.interceptors.http.server;
 
 import com.huaweicloud.sermant.core.plugin.agent.entity.ExecuteContext;
+import com.huaweicloud.sermant.core.plugin.service.PluginServiceManager;
+import com.huaweicloud.sermant.core.utils.StringUtils;
 import com.huaweicloud.sermant.core.utils.tag.TrafficUtils;
 import com.huaweicloud.sermant.tag.transmission.config.strategy.TagKeyMatcher;
 import com.huaweicloud.sermant.tag.transmission.interceptors.AbstractServerInterceptor;
+import com.huaweicloud.sermant.tag.transmission.service.HeaderParseService;
 
 import java.util.Collections;
 import java.util.Enumeration;
@@ -41,6 +44,11 @@ public class HttpServletInterceptor extends AbstractServerInterceptor<HttpServle
      */
     protected static final ThreadLocal<Boolean> LOCK_MARK = new ThreadLocal<>();
 
+    private static final String SERVICECOMB_HEADER_KEY_NAME = "x-cse-context";
+
+    private static HeaderParseService parseService =
+            PluginServiceManager.getPluginService(HeaderParseService.class);
+
     @Override
     public ExecuteContext doBefore(ExecuteContext context) {
         if (LOCK_MARK.get() != null) {
@@ -81,6 +89,27 @@ public ExecuteContext onThrow(ExecuteContext context) {
     @Override
     protected Map<String, List<String>> extractTrafficTagFromCarrier(HttpServletRequest httpServletRequest) {
         Map<String, List<String>> tagMap = new HashMap<>();
+
+        // servicecomb场景，consumer端为servicecomb rpc，需从http的request中获取servicecomb的header字符串并解析
+        String serviceCombHeaderValue = httpServletRequest.getHeader(SERVICECOMB_HEADER_KEY_NAME);
+        if (!StringUtils.isBlank(serviceCombHeaderValue)) {
+            Map<String, String> headers = parseService.parseHeaderFromJson(serviceCombHeaderValue);
+            for (String key : headers.keySet()) {
+                if (!TagKeyMatcher.isMatch(key)) {
+                    continue;
+                }
+                String value = headers.get(key);
+                if (value != null) {
+                    tagMap.put(key, Collections.singletonList(value));
+                    continue;
+                }
+
+                // 流量标签的value为null时，也需存入本地变量，覆盖原来的value，以防误用旧流量标签
+                tagMap.put(key, null);
+            }
+        }
+
+        // 常规http访问场景
         Enumeration<String> keyEnumeration = httpServletRequest.getHeaderNames();
         while (keyEnumeration.hasMoreElements()) {
             String key = keyEnumeration.nextElement();

sermant-plugins/sermant-tag-transmission/tag-transmission-plugin/src/main/java/com/huaweicloud/sermant/tag/transmission/service/HeaderParseService.java

@@ -0,0 +1,37 @@
+/*
+ *   Copyright (C) 2023-2023 Huawei Technologies Co., Ltd. All rights reserved.
+ *
+ *   Licensed under the Apache License, Version 2.0 (the "License");
+ *   you may not use this file except in compliance with the License.
+ *   You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *   Unless required by applicable law or agreed to in writing, software
+ *   distributed under the License is distributed on an "AS IS" BASIS,
+ *   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *   See the License for the specific language governing permissions and
+ *   limitations under the License.
+ */
+
+package com.huaweicloud.sermant.tag.transmission.service;
+
+import com.huaweicloud.sermant.core.plugin.service.PluginService;
+
+import java.util.Map;
+
+/**
+ * 用于从http header的value中解析json字符串为rpc服务的header
+ *
+ * @author daizhenyu
+ * @since 2023-09-14
+ **/
+public interface HeaderParseService extends PluginService {
+    /**
+     * 将json字符串解析为rpc服务 的header
+     *
+     * @param header Json字符串
+     * @return rpc服务 的header
+     */
+    Map<String, String> parseHeaderFromJson(String header);
+}

sermant-plugins/sermant-tag-transmission/tag-transmission-plugin/src/main/resources/META-INF/services/com.huaweicloud.sermant.core.plugin.agent.declarer.PluginDeclarer

@@ -28,7 +28,6 @@ com.huaweicloud.sermant.tag.transmission.declarers.rpc.sofarpc.SofaRpcServerDecl
 com.huaweicloud.sermant.tag.transmission.declarers.rpc.grpc.GrpcServerDeclarer
 com.huaweicloud.sermant.tag.transmission.declarers.rpc.grpc.ClientCallImplDeclarer
 com.huaweicloud.sermant.tag.transmission.declarers.rpc.servicecomb.ServiceCombRpcConsumerDeclarer
-com.huaweicloud.sermant.tag.transmission.declarers.rpc.servicecomb.ServiceCombRpcProviderDeclarer
 com.huaweicloud.sermant.tag.transmission.declarers.mq.kafka.KafkaProducerDeclarer
 com.huaweicloud.sermant.tag.transmission.declarers.http.client.httpclient.HttpClient3xDeclarer
 com.huaweicloud.sermant.tag.transmission.declarers.http.client.okhttp.OkHttp2xDeclarer

sermant-plugins/sermant-tag-transmission/tag-transmission-service/pom.xml

@@ -23,6 +23,10 @@
             <groupId>org.yaml</groupId>
             <artifactId>snakeyaml</artifactId>
         </dependency>
+        <dependency>
+            <groupId>com.alibaba</groupId>
+            <artifactId>fastjson</artifactId>
+        </dependency>
         <dependency>
             <groupId>com.huaweicloud.sermant</groupId>
             <artifactId>sermant-agentcore-core</artifactId>