StampedLock的使用

StampedLock是Java8引入的一种新的锁机制，是对读写锁ReentrantReadWriteLock的增强。该锁提供了三种模式的读写控制，分别为 writeLock 排他锁、readLock 悲观读锁、tryOptimisticRead 乐观读锁。

 

相比ReentrantReadWriteLock，StampedLock通过提供乐观读锁在读线程多写线程少的情况下提供更好的性能，因为乐观读锁不需要进行CAS设置锁的状态而只是简单的测试状态。

 

    StampedLock的内部实现是基于CLH锁的，CLH锁是一种自旋锁，它保证没有饥饿的发生，并且可以保证FIFO(先进先出)的服务顺序。

    CLH锁的基本思想如下:锁维护一个等待线程队列，所有申请锁，但是没有成功的线程都记录在这个队列中，每一个节点代表一个线程，保存一个标记位(locked)，用于判断当前线程是否已经释放锁，locked=true 表示获取到锁，false 表示成功释放了锁。

    当一个线程试图获得锁时，取得等待队列的尾部节点作为其前序节点，并使用类似如下代码判断前序节点是否已经成功释放锁:while (pred.locked) { ... }

    只要前序节点(pred)没有释放锁，则表示当前线程还不能继续执行，因此会自旋等待；反之，如果前序线程已经释放锁，则当前线程可以继续执行。

    释放锁时，也遵循这个逻辑，线程会将自身节点的locked位置标记位false，那么后续等待的线程就能继续执行了。

 

private static void writeLock(int threadCount){
	for(int i=0; i<threadCount; i++){
		service.submit(() -> {
			long stamp = lock.writeLock(); //获取排他写锁
			try{
				TimeUnit.MILLISECONDS.sleep(milli);
				++count;
				System.out.println("\t" + count);
			}catch(Exception ex){
				ex.printStackTrace();
			}finally{
				lock.unlockWrite(stamp); //释放写锁
			}
		});
	}
}

private static void readLock(int threadCount){
	for(int i=0; i<threadCount; i++){
		service.submit(() -> {
			int currentCount = 0;
			
			long stamp = lock.readLock(); //获取排他读锁
			try {
				currentCount = count; //获取变量值
			} finally {
				lock.unlockRead(stamp); //释放读锁
			}
			
			System.out.println(currentCount); //显示最新的变量值
		});
	}
}

private static void optimisticRead(int threadCount){
	for(int i=0; i<threadCount; i++){
		service.submit(() -> {
			long stamp = lock.tryOptimisticRead(); //尝试获取乐观读锁
			int currentCount = count; //获取变量值
			
			if(!lock.validate(stamp)){ //有写锁修改变量值
				stamp = lock.readLock(); //获取排他读锁
				try {
					currentCount = count; //重新获取最新的变量值
				} finally {
					lock.unlockRead(stamp); //释放读锁
				}
			}
			
			System.out.println(currentCount); //显示最新的变量值
		});
	}
}