首页 > 编程语言 >Java多线程（6）：锁与AQS（下）

Java多线程（6）：锁与AQS（下）

时间：2022-10-31 06:55:05浏览次数：38  
标签：return AQS int void Override Java 多线程 public

您好，我是湘王，这是我的博客园，欢迎您来，欢迎您再来～

 

之前说过，AQS（抽象队列同步器）是Java锁机制的底层实现。既然它这么优秀，是骡子是马，就拉出来溜溜吧。

首先用重入锁来实现简单的累加，就像这样：

/**
 * 用重入锁实现累加
 *
 * @author 湘王
 */
public class MyLockTest {
    private final Lock lock = new ReentrantLock();
    private int value;
    public int getNext() {
        lock.lock();
        try {
            value++;
        } finally {
            lock.unlock();
        }
        return value;
    }
    public static void main(String[] args) {
        MyLockTest myLock = new MyLockTest();
        for (int i = 0; i < 5; i++) {
            new Thread(new Runnable() {
                @Override
                public void run() {
                    for (int i = 0; i < 5; i++) {
                        System.out.println(myLock.getNext());
                    }
                }
            }).start();
        }
    }
}

 

运行结果显示数据有重复：

 

 

 

这么简单的计算都能出现重复，这肯定是无法接受的。

再用独占锁来试试看：

/**
 * 利用AQS实现自定义独占锁
 *
 * @author 湘王
 */
public class MyExclusiveLock implements Lock {
    @Override
    public void lock() {

    }

    @Override
    public void lockInterruptibly() throws InterruptedException {

    }

    @Override
    public boolean tryLock() {
        return false;
    }

    @Override
    public boolean tryLock(long time, TimeUnit unit) throws InterruptedException {
        return false;
    }

    @Override
    public void unlock() {

    }

    @Override
    public Condition newCondition() {
        return null;
    }
}

 

 

可以看到，实现lock接口，就需要实现若干自定义的接口。然后以内部类继承AQS的方式，实现排他锁，昨天也说过，AQS中tryAcquire()和tryRelease()是一一对应的，也就是也管获取，一个管释放，所以代码是：

/**
 * 内部类继承AQS的方式，实现排他锁
 */
private static class SyncHelper extends AbstractQueuedSynchronizer {
    private static final long serialVersionUID = -7666580981453962426L;

    /**
     * 第一个线程进来，拿到锁就返回true；后面的线程进来，拿不到锁就返回false
     */
    @Override
    protected boolean tryAcquire(int arg) {
        // 获取资源状态
        int state = getState();
        if (0 == state) {// 如果没有线程拿到资源的锁
            if (compareAndSetState(0, arg)) {
                // 保存当前持有同步锁的线程
                setExclusiveOwnerThread(Thread.currentThread());
                return true;
            }
        } else if (Thread.currentThread() == getExclusiveOwnerThread()) {
            // 如果当前线程再次进来，state + 1，可重入
            // 如果这里没有这个判断，那么程序会卡死
            setState(state + arg);
            return true;
        }
        return false;
    }

    /**
     * 锁的获取和释放需要一一对应
     */
    @Override
    protected boolean tryRelease(int arg) {
        // 获取资源状态
        int state = getState();
        // 返回最后一个通过setExclusiveOwnerThread()方法设置过的线程，或者null
        if (Thread.currentThread() != getExclusiveOwnerThread()) {
            throw new RuntimeException();
        }
        setState(state - arg);
        if (0 == state) {
            setExclusiveOwnerThread(null);
            return true;
        }
        return false;
    }

    protected Condition newCondition() {
        return new ConditionObject();
    }
}

 

 

然后再用AQS实现lock接口的方法：

/**
 * 利用AQS实现自定义独占锁
 *
 * @author 湘王
 */
public class MyExclusiveLock implements Lock {
    private final SyncHelper synchepler = new SyncHelper();

    @Override
    public void lock() {
        synchepler.acquire(1);
    }

    @Override
    public void lockInterruptibly() throws InterruptedException {
        synchepler.acquireInterruptibly(1);
    }

    @Override
    public boolean tryLock() {
        return synchepler.tryAcquire(1);
    }

    @Override
    public boolean tryLock(long time, TimeUnit unit) throws InterruptedException {
        return synchepler.tryAcquireNanos(1, unit.toNanos(time));
    }

    @Override
    public void unlock() {
        synchepler.release(1);
    }

    @Override
    public Condition newCondition() {
        return synchepler.newCondition();
    }

    /**
     * 内部类继承AQS的方式，实现排他锁
     */
    private static class SyncHelper extends AbstractQueuedSynchronizer {
        private static final long serialVersionUID = -7666580981453962426L;
    
        /**
         * 第一个线程进来，拿到锁就返回true；后面的线程进来，拿不到锁就返回false
         */
        @Override
        protected boolean tryAcquire(int arg) {
            // 获取资源状态
            int state = getState();
            if (0 == state) {// 如果没有线程拿到资源的锁
                if (compareAndSetState(0, arg)) {
                    // 保存当前持有同步锁的线程
                    setExclusiveOwnerThread(Thread.currentThread());
                    return true;
                }
            } else if (Thread.currentThread() == getExclusiveOwnerThread()) {
                // 如果当前线程再次进来，state + 1，可重入
                // 如果这里没有这个判断，那么程序会卡死
                setState(state + arg);
                return true;
            }
            return false;
        }
    
        /**
         * 锁的获取和释放需要一一对应
         */
        @Override
        protected boolean tryRelease(int arg) {
            // 获取资源状态
            int state = getState();
            // 返回最后一个通过setExclusiveOwnerThread()方法设置过的线程，或者null
            if (Thread.currentThread() != getExclusiveOwnerThread()) {
                throw new RuntimeException();
            }
            setState(state - arg);
            if (0 == state) {
                setExclusiveOwnerThread(null);
                return true;
            }
            return false;
        }
    
        protected Condition newCondition() {
            return new ConditionObject();
        }
    }
}

 

 

然后再运行测试：

/**
 * 实现Lock接口方法并运行排他锁测试
 *
 * @author 湘王
 */
public class MyExclusiveLockTester {
    // 用自定义AQS独占锁实现
    private Lock lock = new MyExclusiveLock();
    private int value;

    public int accmulator() {
        lock.lock();
        try {
            ++value;
        } finally {
            lock.unlock();
        }

        return value;
    }

    public static void main(String[] args) throws InterruptedException {
        MyExclusiveLockTester test = new MyExclusiveLockTester();
        for (int i = 0; i < 5; i++) {
            new Thread(new Runnable() {
                @Override
                public void run() {
                    for (int i = 0; i < 5; i++) {
                        System.out.println(test.accmulator());
                    }
                }
            }).start();
        }
    }
}

 

 

可以看到，结果无论怎么样都不会再重复了。

 

这个只是简单的累加，接下来用AQS来实现一个实际的生活场景。比如周末带女票或男票去步行街吃饭，这时候人特别多，需要摇号，而且一次只能进去三张号（不按人头算，按叫到的号来算），该怎么实现呢？

可以顺着这个思路：摇号机虽有很多号，但它本质上是个共享资源，很多人可以共享，但是每次共享的数量有限。这其实就是个可以指定数量的共享锁而已。

既然有了思路，那接下来就好办了。

/**
 * 利用AQS实现自定义共享锁
 *
 * @author 湘王
 */
public class MyShareLock implements Lock {
    @Override
    public void lock() {
    }

    @Override
    public void lockInterruptibly() throws InterruptedException {
    }

    @Override
    public boolean tryLock() {
        return false;
    }

    @Override
    public boolean tryLock(long time, TimeUnit unit) throws InterruptedException {
        return false;
    }

    @Override
    public void unlock() {
    }

    @Override
    public Condition newCondition() {
        return null;
    }
}

 

 

还是一样实现Lock接口，但这次是用AQS实现共享锁。

/**
 * 内部类继承AQS实现共享锁
 *
 */
private static class SyncHelper extends AbstractQueuedSynchronizer {
    private static final long serialVersionUID = -7357716912664213942L;

    /**
     * count表示允许几个线程能同时获得锁
     */
    public SyncHelper(int count) {
        if (count <= 0) {
            throw new IllegalArgumentException("锁资源数量必须大于0");
        }
        // 设置资源总数
        setState(count);
    }

    /**
     * 一次允许多少个线程进来，允许数量的线程都能拿到锁，其他的线程进入队列
     */
    @Override
    protected int tryAcquireShared(int acquires) {
        // 自旋
        for (;;) {
            int state = getState();
            int remain = state - acquires;
            // 判断剩余锁资源是否已小于0或者CAS执行是否成功
            if (remain < 0 || compareAndSetState(state, remain)) {
                return remain;
            }
        }
    }

    /**
     * 锁资源的获取和释放要一一对应
     */
    @Override
    protected boolean tryReleaseShared(int releases) {
        // 自旋
        for (;;) {
            // 获取当前state
            int current = getState();
            // 释放状态state增加releases
            int next = current + releases;
            if (next < current) {// 溢出
                throw new Error("Maximum permit count exceeded");
            }
            // 通过CAS更新state的值
            // 这里不能用setState()
            if (compareAndSetState(current, next)) {
                return true;
            }
        }
    }

    protected Condition newCondition() {
        return new ConditionObject();
    }
}

 

 

然后再来改造之前实现的接口：

/**
 * 利用AQS实现自定义共享锁
 *
 * @author 湘王
 */
public class MyShareLock implements Lock {
    public static int count;
    private final SyncHelper synchepler = new SyncHelper(count);

    @Override
    public void lock() {
        synchepler.acquireShared(1);
    }

    @Override
    public void lockInterruptibly() throws InterruptedException {
        synchepler.acquireSharedInterruptibly(1);
    }

    @Override
    public boolean tryLock() {
        return synchepler.tryAcquireShared(1) > 0;
    }

    @Override
    public boolean tryLock(long time, TimeUnit unit) throws InterruptedException {
        return synchepler.tryAcquireSharedNanos(1, unit.toNanos(time));
    }

    @Override
    public void unlock() {
        synchepler.releaseShared(1);
    }

    @Override
    public Condition newCondition() {
        return synchepler.newCondition();
    }

    /**
     * 内部类继承AQS实现共享锁
     *
     */
    private static class SyncHelper extends AbstractQueuedSynchronizer {
        private static final long serialVersionUID = -7357716912664213942L;

        /**
         * count表示允许几个线程能同时获得锁
         */
        public SyncHelper(int count) {
            if (count <= 0) {
                throw new IllegalArgumentException("锁资源数量必须大于0");
            }
            // 设置资源总数
            setState(count);
        }

        /**
         * 一次允许多少个线程进来，允许数量的线程都能拿到锁，其他的线程进入队列
         */
        @Override
        protected int tryAcquireShared(int acquires) {
            // 自旋
            for (;;) {
                int state = getState();
                int remain = state - acquires;
                // 判断剩余锁资源是否已小于0或者CAS执行是否成功
                if (remain < 0 || compareAndSetState(state, remain)) {
                    return remain;
                }
            }
        }

        /**
         * 锁资源的获取和释放要一一对应
         */
        @Override
        protected boolean tryReleaseShared(int releases) {
            // 自旋
            for (;;) {
                // 获取当前state
                int current = getState();
                // 释放状态state增加releases
                int next = current + releases;
                if (next < current) {// 溢出
                    throw new Error("Maximum permit count exceeded");
                }
                // 通过CAS更新state的值
                // 这里不能用setState()
                if (compareAndSetState(current, next)) {
                    return true;
                }
            }
        }

        protected Condition newCondition() {
            return new ConditionObject();
        }
    }
}

 

 

接下来就该测试咱们需要的效果是否能实现了：

public class MyShareLockTester {
    public static void main(String[] args) throws InterruptedException {
        // 用自定义AQS共享锁实现
        // 一次允许发放三把锁
        MyShareLock.count = 3;
        final Lock lock = new MyShareLock();

        // 模拟20个客户端访问
        for (int i = 0; i < 20; i++) {
            new Thread(new Runnable() {
                @Override
                public void run() {
                    try {
                        lock.lock();
                        System.out.println("持有 " + Thread.currentThread().getName() + " 的客人可以进餐厅就餐");
                        // 每两次叫号之间间隔一段时间，模拟真实场景
                        Thread.sleep(3000);
                    } catch (InterruptedException e) {
                        e.printStackTrace();
                    } finally {
                        // 使用完成释放锁
                        lock.unlock();
                    }
                }
            }).start();
        }
    }
}

 

这里有20个号，每次只能发放3张，运行之后就可以看到确实如此。

AQS是个很神奇也很好玩的东西，就像它的作者（也是除了高司令就是对Java影响最大的那个人，整个Java的多线程juc包代码就是他编写的）Doug Lea在AbstractQueuedSynchronizer的注释中所说：AQS只是一个框架，至于怎么玩，就是你的事了！

 

 

 

 

感谢您的大驾光临！咨询技术、产品、运营和管理相关问题，请关注后留言。欢迎骚扰，不胜荣幸～

 

标签：return,AQS,int,void,Override,Java,多线程,public
From： https://www.cnblogs.com/xiangwang1111/p/16842277.html

相关文章

  • javax.validation 请求参数校验小问题: @Min 不校验是否为空
    想用@Min来校验是否为空, 结果发现不行, 看来是必须用到@NotNull 注解了.如下. 我的代码是:@Min(value=0,message=MsgCdConstant.AMOUNT_MUST_BE_POSITIVE)h......
  • 什么是JAVA内存模型
    前言在并发编程中，当多个线程同时访问同一个共享的可变变量时，会产生不确定的结果，所以要编写线程安全的代码，其本质上是对这些可变的共享变量的访问操作进行管理。导致这种不......
  • JavaWeb
    客户端给服务器发数据叫请求服务器给客户端回传数据叫响应Web资源的分类：静态资源：htmlcssjstxtmp4视频jpg图片动态资源：jsp页面Serviet程序常用的Web服务器：Tomc......
  • java锁之初相见
    废话不多说，先上锁的分类图1、乐观锁&悲观锁悲观锁悲观锁认为自己在使用数据的时候一定有别的线程来修改数据，因此在获取数据的时候会先加锁，把别的线程阻塞住，最终确......
  • Java锁
    Java锁乐观锁：乐观锁是一种乐观思想，即认为读多写少，遇到并发写的可能性低，每次去拿数据的时候都认为别人不会修改，所以不会上锁，但是在更新的时候会判断一下在此......
  • Java面试题（一）
    1、什么是自动装配，有哪些方式spring会自动中上下文中寻找并给bean装配与其相关的属性自动装配方式1、byName：把与bean的属性具有相同名字的其他b......
  • ES6与JavaScript学习总结
    菜鸟的ES6与JavaScript学习总结说明一、概述（1）什么是javascript?（2）什么是ES6（3）ES6和JavaScript的关系（4）ES6的浏览器兼容情况二、语法（1）变量（2）箭头函数（3）解构赋值（4）数组操作（5）字符串......
  • java 获取远程PDF文件并批量下载
    packagepdf;importcn.hutool.core.date.DateUtil;importcn.hutool.core.io.IoUtil;importcn.hutool.core.util.CharsetUtil;importlombok.extern.slf4j.Slf4j;......
  • JavaWeb之连接数据库操作
    亲爱的学弟学妹，我知道你们是因为什么搜到这篇博客的，我也很懂你们现在的心情，此时此刻举目无亲面对着建民老师的高要求，你们或许十分无助，所以我写下这篇博客，因为我淋过雨所以......
  • 学习记录23java拼图小游戏
    拼图目标GUI(GraphicalUserInterface,图形用户接口)这是指采用图形化的方式显示操作界面，几乎所有的语言都有GUI的知识java中有两套完整的体系：AWT包(出现的比较早，可能......

赞助商

阅读排行

网站主页关于我们联系我们网站地图

本网站内容转载自其他媒体，侵权联系[admin##ips99.com]。

Copyright © 2020-2023 IPS99 版权所有 IPS99