Java 读写锁源码分析

  • Z时代
  • 2024-01-10
  • 分类：综合

前言

在实际项目中，比如我们有一个共享资源文件，我们程序会会同时并发的去读、写这个共享资源文件，那怎么能保证在高并发场景下安全、高效读写呢？OK，看了下文便知

提示：以下是本篇文章正文内容，案例仅供参考

一、技术介绍

1.ReentranReadWriteLock是什么？

ReadWriteLock提供了readLock和writeLock两种锁的操作机制，一个是读锁，一个是写锁，而它的实现类就是ReentranReadWriteLock

读锁可以在没有写锁的时候被多个线程同时持有，写锁是独占的(排他的)。这意味着每次只能有一个线程写，但可以多个线程并发地读。

理论上，读写锁比互斥锁允许对于共享数据更大程度的并发。与互斥锁相比，读写锁是否能够提高性能取决于读写数据的频率、读取和写入操作的持续时间、以及读线程和写线程之间的竞争。

二、源码分析

1.ReadLock

直接看ReadLock源码

可以看到ReadLock的lock操作实际实现对应Sync的 tryAcquireShared 方法，我们点进去看看

  1. 如果另一个线程持有写锁，则失败。

  2. readerShouldBlock():读锁是否需要等待（公平锁原则）； r < MAX_COUNT：持有线程小于最大数 ；compareAndSetState(c, c + SHARED_UNIT)：设置读取锁状态；整体意思就是读线程是否应该被阻塞、并且小于最大值、并且比较设置成功

  3. else判断里读锁数量不为0并且不为当前线程，获取计数器并且+1

  4. 最后的这行方法执行是由于线程显然不符合条件，或者CAS失败或计数饱和，使用完整版本的获取，这里面的方法可处理CAS miss 和tryAcquireShared中未处理的可重入读

2.WriteLock

再接着看WriteLock源码

exclusiveCount方法的作用是将同步变量与0xFFFF做&运算，计算结果就是写锁的数量,因此w的值的含义就是写锁的数量 如果c不为0就表示锁被占用了，占用的是写锁还是读锁呢？这个时候就需要根据w的值来判断了。 如果c等于0就表示此时锁还没有被任何线程占用，那就让线程直接去尝试获取锁.

如果w为0，说明写锁数量为0，而此时又因为c不等于0，说明锁被占用，但是不是写锁，那么此时锁的状态一定是读锁，既然是读锁状态，那么写锁此时来获取锁时，就肯定失败，因此当w等于0时，tryAcquire方法返回false。

如果w不为0，说明此时锁的状态时写锁，接着进行current != getExclusiveOwnerThread()判断，判断持有锁的线程是否是当前线程

如果不是当前线程，那么tryAcquire()返回false；如果是当前线程，那么就进行后面的逻辑。为什么是当前线程持有锁，就还能执行后面的逻辑呢？那是因为读写锁是可重入的。

三、单元测试

OK，不废话，接下来上简单直接的单元测试

package com.hyh.core.test;
import org.junit.Test;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantReadWriteLock;
/**
 * 可重入读写锁 TEST
 *
 * @Author: heyuhua
 * @Date: 2021/1/28 14:14
 */
public class ReentranReadWriteLockTest {
  ReentrantReadWriteLock reentrantReadWriteLock = new ReentrantReadWriteLock();
  Lock readLock = reentrantReadWriteLock.readLock();
  Lock writeLock = reentrantReadWriteLock.writeLock();
  @Test
  public void test() throws InterruptedException {
    ExecutorService threadPool = Executors.newFixedThreadPool(10);
    for (int i = 0; i < 5; i++) {
      threadPool.execute(() -> {
        readLock();
      });
      threadPool.execute(() -> {
        writeLock();
      });
    }
    threadPool.shutdown();
    TimeUnit.SECONDS.sleep(10);
  }
  private void readLock() {
    readLock.lock();
    try {
      System.out.println(Thread.currentThread().getName() + "-获取读锁...");
      TimeUnit.SECONDS.sleep(1);
    } catch (InterruptedException e) {
      e.printStackTrace();
    } finally {
      System.out.println(Thread.currentThread().getName() + "-释放读锁...");
      readLock.unlock();
    }
  }
  private void writeLock() {
    writeLock.lock();
    try {
      System.out.println(Thread.currentThread().getName() + "-获取写锁...");
      TimeUnit.SECONDS.sleep(1);
    } catch (InterruptedException e) {
      e.printStackTrace();
    } finally {
      System.out.println(Thread.currentThread().getName() + "-释放写锁...");
      writeLock.unlock();
    }
  }
}

这里模拟5个读和5个写的并发情况，马上看下执行结果

可以看到多个线程都可以同时获得读锁，而写锁必须等读锁释放了后才能获取，同样的读锁必须等写锁释放了后才能获取

总结

读写锁总结一句话：读读共享、写写互斥、读写互斥。

读写锁的适用场合：相比于 ReentrantLock 适用于一般场合，ReadWriteLock 适用读多写少的场景，合理使用可以进一步提高并发

作者寄语

是不是感觉很简单？更多用法请点击下方查看源码,关注我带你揭秘更多高级用法

源码地址：点此查看源码

以上是 Java 读写锁源码分析 的全部内容， 来源链接： utcz.com/z/355144.html

回到顶部