Java 可重入自旋锁简单实现

最新推荐文章于 2025-06-18 00:15:00 发布

置顶 a1282379904

最新推荐文章于 2025-06-18 00:15:00 发布

阅读量2.3k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： Java 文章标签：锁自旋锁

本文链接：https://blog.youkuaiyun.com/a1282379904/article/details/80324078

Java 专栏收录该内容

42 篇文章

订阅专栏

本文详细介绍了自旋锁的特点及其实现方式，并提供了一个Java自旋锁的具体实现案例。通过对自旋锁的深入探讨，可以帮助读者理解其工作原理及适用场景。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

自旋锁特点：

1. 轻量级操作，无需挂起线程

2. 特别吃CPU，如果线程在临界区的操作比较耗时或者线程对临界区的竞争很激烈，那还是老老实实用普通的锁

以下是Java的简单实现

public class SpinLock implements Lock {

    /**
     * 锁持有线程, null表示锁未被任何线程持有
     */
    private final AtomicReference<Thread> owner = new AtomicReference<Thread>();
    
    /**
     * owner持有锁次数
     */
    private int holdCount;
    
    @Override
    public void lock() {
        final AtomicReference<Thread> owner = this.owner;

        final Thread current = Thread.currentThread();
        if (owner.get() == current) { // 当前线程已持有锁, 增加持有计数即可
            ++holdCount;
            return;
        }
        
        while (!owner.compareAndSet(null, current)) {
        }

        holdCount = 1;
    }

    @Override
    public void lockInterruptibly() throws InterruptedException {
        final AtomicReference<Thread> owner = this.owner;
        
        final Thread current = Thread.currentThread();
        if (owner.get() == current) {
            ++holdCount;
            return;
        }
        
        while (!owner.compareAndSet(null, current)) {
            // 响应中断
            if (current.isInterrupted()) {
                current.interrupt(); // 重设中断标志
                throw new InterruptedException();
            }
        }

        holdCount = 1;
    }

    @Override
    public boolean tryLock() {
        boolean locked =  owner.compareAndSet(null, Thread.currentThread());
        if (locked) {
            holdCount = 1;
        }
        return locked;
    }

    @Override
    public boolean tryLock(long time, TimeUnit unit) throws InterruptedException {
        final AtomicReference<Thread> owner = this.owner;
        final Thread current = Thread.currentThread();
        if (owner.get() == current) {
            ++holdCount;
            return true;
        }
        
        final long start = System.nanoTime();
        final long timeoutNanos = unit.toNanos(time);
        while (!owner.compareAndSet(null, current)) {
            // 响应中断
            if (current.isInterrupted()) {
                current.interrupt();
                throw new InterruptedException();
            }
            // 判断是否超时
            long elapsed = System.nanoTime() - start;
            if (elapsed >= timeoutNanos) {
                return false;
            }
        }

        holdCount = 1;
        return true;
    }

    @Override
    public void unlock() {
        final AtomicReference<Thread> owner = this.owner;
        final Thread current = Thread.currentThread();
        
        if (owner.get() != current) {
            throw new IllegalMonitorStateException();
        }
        // 持有多少次, 就必须释放多少次
        if (--holdCount == 0) {
            owner.set(null);
        }
    }

    @Override
    public Condition newCondition() {
        throw new UnsupportedOperationException();
    }
}