在Java的并发编程中,AbstractQueuedSynchronizer(简称AQS)是一个核心组件,它不仅是实现同步器的基础,也是并发包中多种锁(如ReentrantLock、CountDownLatch等)的底层实现。AQS通过其精巧的设计,为开发者提供了一种高效且灵活的同步机制。
AQS的核心概念
AQS是一个抽象类,全称为AbstractQueuedSynchronizer,它定义了一种基于FIFO(先进先出)队列的同步框架。AQS内部维护了一个volatile的state变量,用于表示同步状态。这个状态变量是AQS的核心,通过它来控制对共享资源的访问。当state为0时,表示没有线程持有锁;当state大于0时,表示有线程持有锁。
AQS支持两种资源共享模式:独占式和共享式。独占式模式下,每次只有一个线程能够持有锁,如ReentrantLock;而共享式模式下,允许多个线程同时访问共享资源,如ReentrantReadWriteLock的读锁部分。
AQS的内部结构
AQS内部使用了一个CLH(Craig, Landin, and Hagersten)队列来管理等待获取锁的线程。这个队列是一个双向链表,通过head和tail两个指针来维护队列的头部和尾部。每个节点(Node)代表一个等待获取锁的线程,节点中包含了线程引用、等待状态等信息。
AQS的工作原理
当一个线程尝试获取锁时,首先会检查state的值。如果state为0,表示当前没有线程持有锁,该线程将成功获取锁,并将state设置为1(或其他值,取决于具体实现)。如果state不为0,表示锁已被其他线程持有,当前线程将被放入等待队列中,并进入阻塞状态。
当持有锁的线程释放锁时,它会将state的值设置为0,并唤醒等待队列中的下一个线程。被唤醒的线程会再次尝试获取锁,如果成功,则继续执行;如果失败,则重新进入等待队列。
示例代码
下面是一个使用AQS实现简单互斥锁的示例代码:
java
import java.util.concurrent.locks.AbstractQueuedSynchronizer;
class Mutex {
private final Sync sync = new Sync();
public void lock() {
sync.acquire(1);
}
public void unlock() {
sync.release(1);
}
private static class Sync extends AbstractQueuedSynchronizer {
@Override
protected boolean tryAcquire(int acquires) {
return compareAndSetState(0, 1);
}
@Override
protected boolean tryRelease(int releases) {
setState(0);
return true;
}
@Override
protected boolean isHeldExclusively() {
return getState() == 1;
}
}
}
// 使用示例
public class Main {
public static void main(String[] args) {
Mutex mutex = new Mutex();
// 线程1尝试获取锁
new Thread(() -> {
mutex.lock();
try {
// 模拟任务执行
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
mutex.unlock();
}
}).start();
// 线程2尝试获取锁(将在线程1释放锁后获取)
new Thread(() -> {
mutex.lock();
try {
// 模拟任务执行
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
} finally {
mutex.unlock();
}
}).start();
}
}
在这个示例中,我们定义了一个名为Mutex的互斥锁类,它内部使用了一个继承自AbstractQueuedSynchronizer的Sync类来实现锁的逻辑。通过重写tryAcquire、tryRelease和isHeldExclusively方法,我们实现了简单的锁获取和释放逻辑。
AQS以其简洁而强大的设计,为Java并发编程提供了坚实的基础。通过理解AQS的工作原理,我们可以更加深入地掌握Java并发编程的精髓。