Replace ThreadLocal with TransmittableThreadLocal in HintManager to s…

…upport cross Thread usage (#31286)
apache · May 19, 2024 · 03b7272 · 03b7272
1 parent b0b9dce
commit 03b7272
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/infra/common/pom.xml b/infra/common/pom.xml
@@ -110,6 +110,10 @@
             <groupId>com.zaxxer</groupId>
             <artifactId>HikariCP</artifactId>
         </dependency>
+        <dependency>
+            <groupId>com.alibaba</groupId>
+            <artifactId>transmittable-thread-local</artifactId>
+        </dependency>
     </dependencies>
 
     <build>

diff --git a/infra/common/src/main/java/org/apache/shardingsphere/infra/hint/HintManager.java b/infra/common/src/main/java/org/apache/shardingsphere/infra/hint/HintManager.java
@@ -17,6 +17,7 @@
 
 package org.apache.shardingsphere.infra.hint;
 
+import com.alibaba.ttl.TransmittableThreadLocal;
 import com.google.common.base.Preconditions;
 import com.google.common.collect.ArrayListMultimap;
 import com.google.common.collect.Multimap;
@@ -34,7 +35,7 @@
 @NoArgsConstructor(access = AccessLevel.PRIVATE)
 public final class HintManager implements AutoCloseable {
 
-    private static final ThreadLocal<HintManager> HINT_MANAGER_HOLDER = new ThreadLocal<>();
+    private static final TransmittableThreadLocal<HintManager> HINT_MANAGER_HOLDER = new TransmittableThreadLocal<>();
 
     private final Multimap<String, Comparable<?>> databaseShardingValues = ArrayListMultimap.create();