feat: support kafka lane & THREAD_LOCAL_CONTEXT support TransmittableThreadLocal.

Author: shedfreez

polaris-common/polaris-metadata/src/main/java/com/tencent/polaris/metadata/core/manager/MetadataContextHolder.java

@@ -20,12 +20,33 @@
 import java.util.function.Consumer;
 import java.util.function.Supplier;
 
+import com.tencent.polaris.logging.LoggerFactory;
+import org.slf4j.Logger;
+
 public class MetadataContextHolder {
 
-    private static final ThreadLocal<MetadataContext> THREAD_LOCAL_CONTEXT = new ThreadLocal<>();
+	private static final Logger LOG = LoggerFactory.getLogger(MetadataContextHolder.class);
+
+    private static final ThreadLocal<MetadataContext> THREAD_LOCAL_CONTEXT;
 
     private static Supplier<MetadataContext> initializer;
 
+    static {
+		ThreadLocal<MetadataContext> tempThreadLocalContext;
+		try {
+			// the class name need to be excluded in maven shade plugin
+			Class<?> clazz = Class.forName("com.alibaba.ttl.TransmittableThreadLocal");
+			tempThreadLocalContext = (ThreadLocal<MetadataContext>) clazz.getDeclaredConstructor(new Class[0]).newInstance();
+			LOG.info("Use TransmittableThreadLocal for thread local context");
+		} catch (Exception e) {
+			LOG.debug("Failed to use TransmittableThreadLocal for thread local context, msg: {}", e.getMessage());
+			LOG.info("Use standard ThreadLocal for thread local context");
+			tempThreadLocalContext = new ThreadLocal<>();
+		}
+		THREAD_LOCAL_CONTEXT = tempThreadLocalContext;
+	}
+
+
     public static MetadataContext getOrCreate() {
         MetadataContext metadataContext = THREAD_LOCAL_CONTEXT.get();
         if (null != metadataContext) {

polaris-distribution/polaris-all/pom.xml

@@ -171,6 +171,9 @@
                                     <pattern>com.alibaba</pattern>
                                     <shadedPattern>shade.polaris.com.alibaba
                                     </shadedPattern>
+                                    <excludes>
+                                        <exclude>com.alibaba.ttl.TransmittableThreadLocal</exclude>
+                                    </excludes>
                                 </relocation>
                                 <relocation>
                                     <pattern>io.perfmark</pattern>

polaris-plugins/polaris-plugins-connector/connector-consul/src/main/java/com/tencent/polaris/plugins/connector/consul/service/lane/LaneService.java

@@ -64,7 +64,9 @@
 import java.util.concurrent.atomic.AtomicReference;
 
 import static com.tencent.polaris.api.config.plugin.DefaultPlugins.SERVER_CONNECTOR_CONSUL;
+import static com.tencent.polaris.metadata.core.constant.TsfMetadataConstants.TSF_APPLICATION_ID;
 import static com.tencent.polaris.metadata.core.constant.TsfMetadataConstants.TSF_GROUP_ID;
+import static com.tencent.polaris.metadata.core.constant.TsfMetadataConstants.TSF_NAMESPACE_ID;
 import static com.tencent.polaris.plugins.connector.consul.service.common.TagConditionUtil.parseMatchStringType;
 
 /**
@@ -216,6 +218,9 @@ private List<LaneProto.LaneGroup> parseResponse(LaneInfoResponse laneInfoRespons
                 if (laneGroup.isEntrance()) {
                     entranceList.add(laneGroup.getGroupId());
                 }
+                if (StringUtils.isNotEmpty(laneGroup.getNamespaceId()) && StringUtils.isNotEmpty(laneGroup.getApplicationId())) {
+                    laneGroupBuilder.putMetadata(laneGroup.getNamespaceId() + "," + laneGroup.getApplicationId(), TSF_NAMESPACE_ID + "," + TSF_APPLICATION_ID);
+                }
             }
         }
         for (String entrance : entranceList) {