跨平台无锁原语:AtomicFu 底层实现与自定义锁构造
深入 AtomicFu 库在 JVM/JS/Native 上的不同实现策略,并基于它构建一种自定义的 StampedLock。 · 难度:入门 · +10XP
跨平台无锁原语:AtomicFu 底层实现与自定义锁构造
AtomicFu 是 JetBrains 官方的跨平台原子操作库。在 JVM 上它直接映射到 java.util.concurrent.atomic,在 Native 上使用 Kotlin/Native 自带的 atomic 内建函数,在 JS 上则利用 Object 属性原子化。本教程不讲解用法,而是通过分析源码揭示各平台的实现差异,然后利用 AtomicInt 和 AtomicReference 组合出一种支持乐观读的锁结构,用于替代 ReentrantReadWriteLock。
import kotlinx.atomicfu.*
class OptimisticLock {
private val state = atomic(0)
fun tryOptimisticRead(): Boolean {
val stamp = state.value
// 模拟类似于 StampedLock 的乐观读
return stamp and 1 == 0
}
fun writeLock() {
while (!state.compareAndSet(0, 1)) { /* spin */ }
}
fun unlockWrite() {
state.value = 0
}
}