expt: fix #293

alibaba · Sep 27, 2021 · 81087ce · 81087ce
1 parent 84107ff
commit 81087ce
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/main/java/com/alibaba/ttl/TransmittableThreadLocal.java b/src/main/java/com/alibaba/ttl/TransmittableThreadLocal.java
@@ -227,7 +227,8 @@ private void removeThisFromHolder() {
     }
 
     private static void doExecuteCallback(boolean isBefore) {
-        for (TransmittableThreadLocal<Object> threadLocal : holder.get().keySet()) {
+        WeakHashMap<TransmittableThreadLocal<Object>, ?> ttlInstances = new WeakHashMap<TransmittableThreadLocal<Object>, Object>(holder.get());
+        for (TransmittableThreadLocal<Object> threadLocal : ttlInstances.keySet()) {
             try {
                 if (isBefore) threadLocal.beforeExecute();
                 else threadLocal.afterExecute();