经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » Java相关 » Java » 查看文章
面试官:素有Java锁王称号的‘StampedLock’你知道吗?我:这什么鬼?
来源:cnblogs  作者:JavaBuild  时间:2024/4/29 9:26:21  对本文有异议

一、写在开头


我们在上一篇写ReentrantReadWriteLock读写锁的末尾留了一个小坑,那就是读写锁因为写锁的悲观性,会导致 “写饥饿”,这样一来会大大的降低读写效率,而今天我们就来将此坑填之!填坑工具为:StampedLock,一个素有Java锁王称号的同步类,也是在 java.util.concurrent.locks 包中。

需要声明的是,这个类在Java的面试过程中极少被问及,如果仅仅是为了准备面试的话,这部分内容可以忽略,但这个类的实现逻辑还是值得一学的。


二、StampedLock 是什么?


StampedLock是由Java8时引入的一个性能更好的读写锁,作者:Doug Lea,支持读锁、写锁,这与ReentrantReadWriteLock类似,但同时多了一个乐观读锁的实现,这一点直接提升了它的性能。


三、StampedLock的原理


虽然StampedLock性能更好,但是!不可重入且不支持条件变量 Condition,且并没有直接实现Lock或者ReadWriteLock接口,而是与AQS类似的采用CLH(Craig, Landin, and Hagersten locks)作为底层实现。

在Java的官方docs中对于它进行了如下的描述:

image

并且官方还提供了一个示例,我们来看一下:

  1. class Point {
  2. //共享变量
  3. private double x, y;
  4. private final StampedLock sl = new StampedLock();
  5. // 写锁的使用
  6. void move(double deltaX, double deltaY) {
  7. long stamp = sl.writeLock(); //涉及对共享资源的修改,使用写锁-独占操作
  8. try {
  9. x += deltaX;
  10. y += deltaY;
  11. } finally {
  12. sl.unlockWrite(stamp); // 释放写锁
  13. }
  14. }
  15. /**
  16. * 使用乐观读锁访问共享资源
  17. * 注意:乐观读锁在保证数据一致性上需要拷贝一份要操作的变量到方法栈,并且在操作数据时候 可能其他写线程已经修改了数据,
  18. * 而我们操作的是方法栈里面的数据,也就是一个快照,所以最多返回的不是最新的数据,但是一致性还是得到保障的。
  19. *
  20. * @return
  21. */
  22. double distanceFromOrigin() {
  23. long stamp = sl.tryOptimisticRead(); // 获取乐观读锁
  24. double currentX = x, currentY = y; // 拷贝共享资源到本地方法栈中
  25. if (!sl.validate(stamp)) { // //检查乐观读锁后是否有其他写锁发生,有则返回false
  26. stamp = sl.readLock(); // 获取一个悲观读锁
  27. try {
  28. currentX = x;
  29. currentY = y;
  30. } finally {
  31. sl.unlockRead(stamp); // 释放悲观读锁
  32. }
  33. }
  34. return Math.sqrt(currentX * currentX + currentY * currentY);
  35. }
  36. // 悲观读锁以及读锁升级写锁的使用
  37. void moveIfAtOrigin(double newX, double newY) {
  38. long stamp = sl.readLock(); // 悲观读锁
  39. try {
  40. while (x == 0.0 && y == 0.0) {
  41. // 读锁尝试转换为写锁:转换成功后相当于获取了写锁,转换失败相当于有写锁被占用
  42. long ws = sl.tryConvertToWriteLock(stamp);
  43. if (ws != 0L) { // 如果转换成功
  44. stamp = ws; // 读锁的票据更新为写锁的
  45. x = newX;
  46. y = newY;
  47. break;
  48. }
  49. else { // 如果转换失败
  50. sl.unlockRead(stamp); // 释放读锁
  51. stamp = sl.writeLock(); // 强制获取写锁
  52. }
  53. }
  54. } finally {
  55. sl.unlock(stamp); // 释放所有锁
  56. }
  57. }
  58. }

在StampedLock 的底层提供了三种锁

  1. 写锁: 独占锁,一把锁只能被一个线程获得。当一个线程获取写锁后,其他请求读锁和写锁的线程必须等待。类似于 ReentrantReadWriteLock 的写锁,不过这里的写锁是不可重入的。
  2. 读锁 (悲观读):共享锁,没有线程获取写锁的情况下,多个线程可以同时持有读锁。如果己经有线程持有写锁,则其他线程请求获取该读锁会被阻塞。类似于 ReentrantReadWriteLock 的读锁,不过这里的读锁是不可重入的。
  3. 乐观读 :允许多个线程获取乐观读以及读锁。同时允许一个写线程获取写锁。

【源码示例】

  1. // 写锁
  2. public long writeLock() {
  3. long s, next; // bypass acquireWrite in fully unlocked case only
  4. return ((((s = state) & ABITS) == 0L &&
  5. U.compareAndSwapLong(this, STATE, s, next = s + WBIT)) ?
  6. next : acquireWrite(false, 0L));
  7. }
  8. // 读锁
  9. public long readLock() {
  10. long s = state, next; // bypass acquireRead on common uncontended case
  11. return ((whead == wtail && (s & ABITS) < RFULL &&
  12. U.compareAndSwapLong(this, STATE, s, next = s + RUNIT)) ?
  13. next : acquireRead(false, 0L));
  14. }
  15. // 乐观读
  16. public long tryOptimisticRead() {
  17. long s;
  18. return (((s = state) & WBIT) == 0L) ? (s & SBITS) : 0L;
  19. }

StampedLock 在获取锁的时候会返回一个 long 型的数据戳,该数据戳用于稍后的锁释放参数,如果返回的数据戳为 0 则表示锁获取失败。当前线程持有了锁再次获取锁还是会返回一个新的数据戳,这也是StampedLock不可重入的原因。此外,在官网给的示例中我们也看到了,StampedLock 还支持这3种锁的转换:

  1. long tryConvertToWriteLock(long stamp){}
  2. long tryConvertToReadLock(long stamp){}
  3. long tryConvertToOptimisticRead(long stamp){}

内部常量说明

在源码中我们看到,无论哪种锁,在获取的时候都会返回一个long类型的时间戳,这其实就是StampedLock命名的由来,而这个时间戳的第8位用来标识写锁,前 7 位(LG_READERS)来表示读锁,每获取一个悲观读锁,就加 1(RUNIT),每释放一个悲观读锁,就减 1。而悲观读锁最多只能装 128 个(7 位限制),很容易溢出,所以用一个 int 类型的变量来存储溢出的悲观读锁。

image


四、StampedLock的使用


结果上面的StampedLock特性和官方的示例,我们写一个小demo来感受一下它的使用,需要注意的是在获取乐观锁时,如果有写锁改变数据时,为保证数据一致性,要切换为普通的读锁模式。

【测试示例】

  1. public class Test {
  2. private final StampedLock sl = new StampedLock();
  3. private int data = 0;
  4. public void write(int value) {
  5. long stamp = sl.writeLock();
  6. try {
  7. data = value;
  8. } finally {
  9. sl.unlockWrite(stamp);
  10. }
  11. }
  12. public int read() {
  13. long stamp = sl.tryOptimisticRead();
  14. int currentData = data;
  15. // 如果有写锁被占用,可能造成数据不一致,所以要切换到普通读锁模式
  16. if (!sl.validate(stamp)) {
  17. stamp = sl.readLock();
  18. try {
  19. currentData = data;
  20. } finally {
  21. sl.unlockRead(stamp);
  22. }
  23. }
  24. return currentData;
  25. }
  26. public static void main(String[] args) {
  27. Test test = new Test();
  28. Thread writer = new Thread(() -> {
  29. for (int i = 0; i < 5; i++) {
  30. test.write(i);
  31. System.out.println("当前线程" + Thread.currentThread().getName() + ":Write: " + i);
  32. }
  33. });
  34. Thread reader = new Thread(() -> {
  35. for (int i = 0; i < 5; i++) {
  36. int value = test.read();
  37. System.out.println("当前线程" + Thread.currentThread().getName() + ":Read: " + value);
  38. }
  39. });
  40. writer.start();
  41. reader.start();
  42. }
  43. }

输出:

  1. 当前线程Thread-0:Write: 0
  2. 当前线程Thread-0:Write: 1
  3. 当前线程Thread-1:Read: 0
  4. 当前线程Thread-0:Write: 2
  5. 当前线程Thread-1:Read: 2
  6. 当前线程Thread-0:Write: 3
  7. 当前线程Thread-1:Read: 3
  8. 当前线程Thread-0:Write: 4
  9. 当前线程Thread-1:Read: 4
  10. 当前线程Thread-1:Read: 4

五、总结

相比于传统读写锁多出来的乐观读是StampedLock比 ReadWriteLock 性能更好的关键原因。StampedLock 的乐观读允许一个写线程获取写锁,所以不会导致所有写线程阻塞,也就是当读多写少的时候,写线程有机会获取写锁,减少了线程饥饿的问题,吞吐量大大提高。

不过,需要注意的是StampedLock不可重入,不支持条件变量 Condition,对中断操作支持也不友好(使用不当容易导致 CPU 飙升)。如果你需要用到 ReentrantLock 的一些高级性能,就不太建议使用 StampedLock 了。

六、结尾彩蛋

如果本篇博客对您有一定的帮助,大家记得留言+点赞+收藏呀。原创不易,转载请联系Build哥!

image

如果您想与Build哥的关系更近一步,还可以关注“JavaBuild888”,在这里除了看到《Java成长计划》系列博文,还有提升工作效率的小笔记、读书心得、大厂面经、人生感悟等等,欢迎您的加入!
image

原文链接:https://www.cnblogs.com/JavaBuild/p/18164898

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号