java中如何实现可重入的自旋锁

admin 百科 14
可重入自旋锁通过volatile变量记录持有线程和重入次数,利用CAS实现原子更新;1. 用owner字段记录当前持有锁的线程;2. count记录重入次数;3. 加锁时先CAS设置owner为当前线程,成功则增加count,否则判断是否已持有并递增count;4. 解锁时先减count,仅当count为0时CAS清空owner;5. 需防止非持有线程解锁,且避免在synchronized中嵌套使用以防死锁。

java中如何实现可重入的自旋锁-第1张图片-佛山资讯网

Java 中实现可重入的自旋锁,核心在于:用一个 volatile int 记录持有线程的重入次数,并用 Thread.currentThread() 判断是否为当前线程重入;同时借助 Unsafe.compareAndSetInt(或 JDK9+ 的 VarHandle)实现无锁的原子状态更新。它不依赖 synchronized 或 AQS,纯靠 CPU 自旋等待,适合临界区极短、竞争不激烈的场景。

关键字段设计

一个可重入自旋锁需要三个基本状态信息:

  • owner:记录当前持有锁的线程(用 AtomicReference 或 volatile Thread 字段)
  • count:记录该线程的重入次数(用 AtomicInteger 或 volatile int,但需保证与 owner 更新的原子性)
  • (可选)spinLimit:限制最大自旋次数,避免过度消耗 CPU

保证 owner 和 count 原子更新

单独用两个 volatile 变量无法保证“先设 owner 再增 count”的原子性。推荐两种方式:

  • 单变量编码法:把 owner 的 identityHashCode 和 count 合并成一个 long(低位存 count,高位存 hash),用 Unsafe.compareAndSwapLong 一次更新。缺点是可能哈希冲突(极低概率),且无法直接比对 Thread 实例
  • 双字段 + 循环 CAS 法(更实用):先 CAS 设置 owner 为当前线程;成功后,再 CAS 增加 count;若期间 owner 被改,就重试。代码结构类似:

while (true) {
  Thread t = owner.get();
  if (t == null) {
    if (owner.compareAndSet(null, Thread.currentThread())) {
      count.incrementAndGet();
      break;
    }
  } else if (t == Thread.currentThread()) {
    count.incrementAndGet();
    break;
  } else {
    // 自旋等待或 yield
    Thread.onSpinWait();
  }
}

立即学习“Java免费学习笔记(深入)”;

标签: java 编码 ai 无锁 有锁

发布评论 0条评论)

还木有评论哦,快来抢沙发吧~