Kotlin Native弱引用GC测试失败原因及解决方法问询

问题描述 #

以下Kotlin Native测试代码试图通过弱引用+手动触发GC验证对象回收，但最后一个断言始终失败——即使清空强引用并调用GC，弱引用仍能获取到原对象：

import kotlin.test.Test
import kotlin.test.assertNotNull
import kotlin.test.assertNull
import kotlin.test.assertSame

class WeakReferenceTest {
    @Test fun weakReference() {
        var strong: Any? = Any()
        val weak = kotlin.native.ref.WeakReference(strong!!)

        kotlin.native.internal.GC.collect()
        assertSame(strong, assertNotNull(weak.get()))

        strong = null
        kotlin.native.internal.GC.collect()
        assertNull(weak.get()) // 此处断言失败
    }
}

而替换为JVM弱引用的相同逻辑代码可正常通过测试：

import kotlin.test.Test
import kotlin.test.assertNotNull
import kotlin.test.assertNull
import kotlin.test.assertSame

class WeakReferenceTest {
    @Test fun weakReference() {
        var strong: Any? = Any()
        val weak = java.lang.ref.WeakReference(strong!!)

        System.gc()
        assertSame(strong, assertNotNull(weak.get()))

        strong = null
        System.gc()
        assertNull(weak.get())
    }
}

失败原因 #

1. 栈分配优化：Kotlin Native的LLVM后端会对局部对象做逃逸分析，若判断对象不会逃出当前函数栈帧，会直接将其分配在栈上而非堆上。栈上对象由栈帧管理，不受GC控制，弱引用自然不会被回收。
2. GC执行特性：Kotlin Native的GC.collect()是启发式触发，仅标记待回收对象，不会立即完成回收操作；且小对象可能被延迟处理，单次调用无法保证对象被回收。
3. 编译器残留引用：即使对象分配在堆上，编译器可能因测试函数作用域较小，将对象地址保留在寄存器或栈帧中，GC会误判存在强引用，跳过回收。

修复方法 #

方法1：强制GC完成回收 #

调用GC.waitForCompletion()确保GC完成标记-清除流程，配合多次触发GC提高可靠性：

import kotlin.test.Test
import kotlin.test.assertNotNull
import kotlin.test.assertNull
import kotlin.test.assertSame
import kotlin.native.ref.WeakReference
import kotlin.native.internal.GC

class WeakReferenceTest {
    @Test fun weakReference() {
        var strong: Any? = Any()
        val weak = WeakReference(strong!!)

        GC.collect()
        assertSame(strong, assertNotNull(weak.get()))

        strong = null
        // 触发GC并等待完成，重复一次确保回收
        GC.collect()
        GC.waitForCompletion()
        GC.collect()
        GC.waitForCompletion()
        
        assertNull(weak.get())
    }
}

方法2：让对象逃逸出栈帧 #

通过辅助函数让对象逃逸，避免编译器将其分配在栈上：

import kotlin.test.Test
import kotlin.test.assertNotNull
import kotlin.test.assertNull
import kotlin.test.assertSame
import kotlin.native.ref.WeakReference
import kotlin.native.internal.GC

// 辅助函数，让对象逃逸出当前栈帧
private fun escape(obj: Any): Any = obj

class WeakReferenceTest {
    @Test fun weakReference() {
        var strong: Any? = escape(Any())
        val weak = WeakReference(strong!!)

        GC.collect()
        assertSame(strong, assertNotNull(weak.get()))

        strong = null
        GC.collect()
        GC.waitForCompletion()
        
        assertNull(weak.get())
    }
}

排查思路 #

• 验证GC是否执行：
  • 使用kotlin.native.internal.GC.usedMemory打印GC前后的内存占用，对比变化判断GC是否生效；
  • 编译时添加-Xgc-log=info参数，运行测试时查看控制台的GC日志，确认回收流程是否触发。
• 检查编译器生成的代码：
  • 用kotlinc-native -Xdump-ir test.kt生成LLVM IR代码，搜索对象分配相关指令：若出现alloca则为栈分配，kotlin_alloc则为堆分配；
  • 用objdump -d <生成的二进制文件>反编译，查看局部变量的生命周期，确认是否存在未被清除的残留引用。
• 排除编译器优化干扰：
  • 尝试给对象添加@OptIn(ExperimentalStdlibApi::class)并调用freeze()（仅适用于不可变对象），强制堆分配；
  • 将对象存入全局集合再清空，模拟真实场景中的引用关系，避免局部变量优化。

内容的提问来源于stack exchange，提问作者Tom