upon clearing the ks object cache, clear retained thread locals

google · Oct 12, 2024 · d2cd414 · d2cd414
1 parent 5b6d886
commit d2cd414
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/common-util/src/main/kotlin/com/google/devtools/ksp/common/KSObjectCacheManager.kt b/common-util/src/main/kotlin/com/google/devtools/ksp/common/KSObjectCacheManager.kt
@@ -28,7 +28,10 @@ class KSObjectCacheManager {
             get() = caches_prop.get()
 
         fun register(cache: KSObjectCache<*, *>) = caches.add(cache)
-        fun clear() = caches.forEach { it.clear() }
+        fun clear() {
+            caches.forEach { it.clear() }
+            caches_prop.remove()
+        }
     }
 }
 
@@ -44,5 +47,8 @@ abstract class KSObjectCache<K, V> {
             return cache_prop.get()
         }
 
-    open fun clear() = cache.clear()
+    open fun clear() {
+        cache.clear()
+        cache_prop.remove()
+    }
 }