Removing extra atomicfu references from LVT for Kotlin 1.5.20

Kotlin · May 28, 2021 · baa8c38 · baa8c38
1 parent 1c38c1d
commit baa8c38
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/atomicfu-transformer/src/main/kotlin/kotlinx/atomicfu/transformer/AtomicFUTransformer.kt b/atomicfu-transformer/src/main/kotlin/kotlinx/atomicfu/transformer/AtomicFUTransformer.kt
@@ -715,11 +715,16 @@ class AtomicFUTransformer(
                     i = i.next
                     hasErrors = true
                 }
+            // make sure all kotlinx/atomicfu references removed
+            removeAtomicReferencesFromLVT()
             // save transformed method if not in analysis phase
             if (!hasErrors && !analyzePhase2)
                 accept(mv)
         }
 
+        private fun removeAtomicReferencesFromLVT() =
+            localVariables?.removeIf { getType(it.desc) in AFU_TYPES }
+
         private fun FieldInsnNode.checkCopyToDelegate(): AbstractInsnNode? {
             val fieldId = FieldId(owner, name, desc)
             if (fieldId in fieldDelegates) {