Lock 那点事儿
博客专区 > 黄勇 的博客 > 博客详情
Lock 那点事儿
黄勇 发表于4年前
Lock 那点事儿
  • 发表于 4年前
  • 阅读 4782
  • 收藏 64
  • 点赞 25
  • 评论 32

华为云·免费上云实践>>>   

项目经理今天又接了一个客户需求,又要折磨我们这些程序员屌丝了。这个需求说起来很简单,做起来非常容易出错。我先简单描述一下:

这是一个在线文件编辑器。同一份文件,一个人在读的时候,其他人不能写;同理,一个人在写的时候,其他人也不能读。也就是说,要么读,要么写,这两件事情不能同时进行。

项目经理跟客户讲,“这个很容易实现的,我们是可以做的。”。什么都可以做,做不出来说是我们程序员能力不行,他一点责任都没有。领导发话了,不管怎么样,事情还是要做的。

看了一下需求,有两个问题,我得先问清楚,否则到时候做得不对,他又把负责推给我,我们项目经理经常搞这些让我背黑锅的事情。

“多人同时读可以吗?”

“当然可以啦!多少人来读都没关系,文件的内容不要变就行。”。

“多人同时写可以吗?”

“当然不行啦!你写别人也会写,文件不知道以哪份数据为准了。”。

他态度极其恶劣,算了,不跟他计较了,我的项目奖金还在他手里。赶紧完工,下班了还要回家抱小孩。

根据多年的项目实战经验,我写了一个超牛逼的 Data 类,来封装文件的数据。看起来是这样的:

public class Data {

    private final char[] buffer;

    public Data(int size) {
        this.buffer = new char[size];
        for (int i = 0; i < size; i++) {
            buffer[i] = '*';
        }
    }

    public String read() {
        StringBuilder result = new StringBuilder();
        for (char c : buffer) {
            result.append(c);
        }
        sleep(100);
        return result.toString();
    }

    public void write(char c) {
        for (int i = 0; i < buffer.length; i++) {
            buffer[i] = c;
            sleep(100);
        }
    }

    private void sleep(long ms) {
        try {
            Thread.sleep(ms);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}

稍微解释一下:

  1. Data 类中封装了一个 char 数组类型的 buffer 成员变量。
  2. 在构造器中传入一个 size,表示 buffer 的长度,并在其中创建并初始化这个 buffer,使其每个字符都为“*”。
  3. 提供两个方法,一个负责读取,另一个负责写入。在读取方法中只需遍历 buffer,将结果不断 append 到一个 StringBuilder 中,最终将其转为 String 并返回。
  4. 在写入方法中传入一个字符,仍然是遍历 buffer,赋值 buffer 中的每个字符,这样可以使 buffer 中每个字符都是相同的。
  5. 故意在读写方法中加入了一个 sleep() 方法,让程序运行慢一点,模拟比较耗时的操作。而且故意让写入比读取慢一点,因为将 sleep() 方法放入了 write() 方法的循环体中,而 read() 方法却没有。

当然了,以上这个示例跑通了,我想项目经理那个需求也不难实现。这也是我们平时做开发的一种习惯,先快速地写个 Demo 出来,让领导们看看,技术上走通了,我们再实现具体的需求。

好了,不就是要同时读写吗?这不就是一个典型的多线程使用场景吗?于是我快速地写了一个读取线程,让它拼命地去读取 Data 中的数据。

public class ReaderThread extends Thread {

    private final Data data;

    public ReaderThread(Data data) {
        this.data = data;
    }

    @Override
    public void run() {
        while (true) {
            String result = data.read();
            System.out.println(Thread.currentThread().getName() + " => " + result);
        }
    }
}

在 ReaderThread 中通过一个死循环去不断地读取 Data 中的数据,并将结果打印出来。

再来一个写入线程,让它使劲地向 Data 中写入数据。

public class WriterThread extends Thread {

    private final Data data;
    private final String str;
    private int index = 0;

    public WriterThread(Data data, String str) {
        this.data = data;
        this.str = str;
    }

    @Override
    public void run() {
        while (true) {
            char c = next();
            data.write(c);
        }
    }

    private char next() {
        char c = str.charAt(index);
        index++;
        if (index >= str.length()) {
            index = 0;
        }
        return c;
    }
}

一次性可以传入一个字符串到 WriterThread 中,它将不断获取下一个字符(请见 next() 方法),并将该字符写入 Data 中。

如果让 ReaderThread 与 WriterThread 同时工作会怎样?不妨写了一个简单的 Client 类运行试试看。

public class Client {

    public static void main(String[] args) {
        Data data = new Data(10);

        new ReaderThread(data).start();
        new ReaderThread(data).start();
        new ReaderThread(data).start();
        new ReaderThread(data).start();
        new ReaderThread(data).start();

        new WriterThread(data, "ABCDEFGHI").start();
        new WriterThread(data, "012345789").start();
    }
}

我开启了 5 个 ReaderThread 与 2 个 WriterThread,模拟读得多写得少的情况,并将不同的数据写入 Data 中。

运行一下!

...
Thread-1 => AA0A0A00A0
Thread-4 => AA0A0A00A0
Thread-3 => AA0A0A00A0
Thread-2 => AA0A0A00A0
Thread-0 => AA0A0A00A0
...

为何每次读取出来的数据不一致呢?应该是输出 10 个相同的字符才对啊!Data 的 buffer 中每个字符不是应该相同吗?

如果把这个结果给项目经理看,他肯定要搞死我的。

哦!想到了!在多线程开发中,资源的访问一定要做到“共享互斥”,也就是说要“上锁”,这招还是架构师前几天才教我的,我怎能不用?

于是我用了 Java 多线程中超牛逼的 synchronized 关键字,将它放到了 read() 与 write() 方法上,这样就可以保证 synchronized 方法在同一时刻只能被一个线程调用了,其他线程将会阻挡在外。

废话少说,赶紧加两个 synchronized 运行看看吧。

public class Data {

    ...

    public synchronized String read() {
        ...
    }

    public synchronized void write(char c) {
        ...
    }

    ...
}

再运行一把!

...
Thread-0 => 1111111111
Thread-4 => CCCCCCCCCC
Thread-3 => CCCCCCCCCC
Thread-2 => CCCCCCCCCC
Thread-1 => CCCCCCCCCC
...

终于搞定啦!这下子项目经理应该满意了吧?

“不错!这效果很好啊,同时写同时读,而且每次读出来的数据都一样,技术上应该是走通了,这个需求应该可以实现了吧?” 项目经理问。

“没问题啊!小意思!” 我高兴的答。

“这是一个在线文件编辑器,你考虑过性能问题吗?” 架构师突然问了一句。

“性能很好啊!”

“你可以在 ReaderThread 中每调用 10 次 read() 方法,就打印 1 次所耗时间看看。”

“好啊!”

这还不简单,我快速地给 ReaderThread 的 run() 方法中加了几行代码,测试一下运行所消耗的时间。

public class ReaderThread extends Thread {

    ...

    @Override
    public void run() {
        while (true) {
            long begin = System.currentTimeMillis();
            for (int i = 0; i < 10; i++) {
                String result = data.read();
                System.out.println(Thread.currentThread().getName() + " => " + result);
            }
            long time = System.currentTimeMillis() - begin;
            System.out.println(Thread.currentThread().getName() + " -- " + time + "ms");
        }
    }
}

跑起来吧!

...
Thread-2 => IIIIIIIIII
Thread-2 -- 24802ms
Thread-3 => IIIIIIIIII
Thread-3 -- 24901ms
Thread-4 => IIIIIIIIII
Thread-4 -- 25001ms
Thread-0 => 3333333333
...
Thread-0 => 1111111111
Thread-0 -- 55305ms
Thread-4 => CCCCCCCCCC
Thread-3 => CCCCCCCCCC
Thread-2 => CCCCCCCCCC
Thread-1 => CCCCCCCCCC
Thread-1 -- 58705ms
Thread-2 => CCCCCCCCCC
...

我随意挑选了其中这 5 个 ReaderThread 所消耗的时间,平均值是:37742.8 毫秒,折合 37.8 秒。

我心里也没谱了,这性能到底是否需要优化呢?于是我带着测试结果,去向架构师请教。

他看到了这样的结果,微笑着摇了摇头。从他鄙视而又猥琐的表情上,我可以推测,这次他又要在我面前露一手了。

来吧,我给你写一个 ReadWriteLock,你自己去看吧。

随后,架构师用他熟练的手指,疯狂地在键盘上敲了一堆让我一知半解的东西。

public class ReadWriteLock {

    private int readThreadCounter = 0;      // 正在读取的线程数(0个或多个)
    private int waitingWriteCounter = 0;    // 等待写入的线程数(0个或多个)
    private int writeThreadCounter = 0;     // 正在写入的线程数(0个或1个)
    private boolean writeFlag = true;       // 是否对写入优先(默认为是)

    // 读取加锁
    public synchronized void readLock() throws InterruptedException {
        // 若存在正在写入的线程,或当写入优先时存在等待写入的线程,则将当前线程设置为等待状态
        while (writeThreadCounter > 0 || (writeFlag && waitingWriteCounter > 0)) {
            wait();
        }
        // 使正在读取的线程数加一
        readThreadCounter++;
    }

    // 读取解锁
    public synchronized void readUnlock() {
        // 使正在读取的线程数减一
        readThreadCounter--;
        // 读取结束,对写入优先
        writeFlag = true;
        // 通知所有处于 wait 状态的线程
        notifyAll();
    }

    // 写入加锁
    public synchronized void writeLock() throws InterruptedException {
        // 使等待写入的线程数加一
        waitingWriteCounter++;
        try {
            // 若存在正在读取的线程,或存在正在写入的线程,则将当前线程设置为等待状态
            while (readThreadCounter > 0 || writeThreadCounter > 0) {
                wait();
            }
        } finally {
            // 使等待写入的线程数减一
            waitingWriteCounter--;
        }
        // 使正在写入的线程数加一
        writeThreadCounter++;
    }

    // 写入解锁
    public synchronized void writeUnlock() {
        // 使正在写入的线程数减一
        writeThreadCounter--;
        // 写入结束,对读取优先
        writeFlag = false;
        // 通知所有处于等待状态的线程
        notifyAll();
    }
}

我看出来了,架构师特意写了很多注释,免得我总是去烦他。

代码不解释了,看看注释吧,有疑问可以给我留言哦!

此时,Data 类还需要稍作改写。

public class Data {

    ...

    private final ReadWriteLock lock = new ReadWriteLock(); // 创建读写锁

    ...

    public String read() throws InterruptedException {
        lock.readLock(); // 读取上锁
        try {
            return doRead(); // 执行读取操作
        } finally {
            lock.readUnlock(); // 读取解锁
        }
    }

    public void write(char c) throws InterruptedException {
        lock.writeLock(); // 写入上锁
        try {
            doWrite(c); // 执行写入操作
        } finally {
            lock.writeUnlock(); // 写入解锁
        }
    }

    private String doRead() {
        StringBuilder result = new StringBuilder();
        for (char c : buffer) {
            result.append(c);
        }
        sleep(100);
        return result.toString();
    }

    private void doWrite(char c) {
        for (int i = 0; i < buffer.length; i++) {
            buffer[i] = c;
            sleep(100);
        }
    }

    ...
}

同样的 Client 类,我再运行一把试试看,性能是否有提高呢?

...
Thread-1 => 4444444444
Thread-2 -- 14000ms
Thread-0 -- 14001ms
Thread-3 -- 14000ms
Thread-4 -- 14000ms
Thread-1 -- 14001ms
Thread-4 => IIIIIIIIII
...

平均下来是 14000.4 毫秒,折合 14.0 秒,比以前快了 63%,而且输出的结果都比以前平稳(以前忽高忽低的)。

果然是架构师,真让我们这些程序员崇拜啊!

最后架构师过来,看到我在那里得意地笑。他拍拍我的肩,对我说:“别乐了,其实 JDK 1.5 中已经有 ReadWriteLock 了,我这个只不过是一个精简版而已,去看看 java.util.concurrent.locks.ReadWriteLock 吧,你一定会震精!”。

看来我真是孤陋寡闻啊,打开 JDK API 看到了 ReadWriteLock:

public interface ReadWriteLock {

    Lock readLock();

    Lock writeLock();
}

可以通过 ReadWriteLock 接口来获取 ReadLock 与 WriteLock,它们都是 Lock 对象,这也是一个接口。

官方提供了一个 ReadWriteLock 接口的实现类 java.util.concurrent.locks.ReentrantReadWriteLock。

public interface Lock {

    void lock();

    void lockInterruptibly() throws InterruptedException;

    boolean tryLock();

    boolean tryLock(long time, TimeUnit unit) throws InterruptedException;

    void unlock();

    Condition newCondition();
}

该接口中,有两个非常重要的方法:lock() 与 unlock(),分别表示“上锁”与“解锁”。

尝试用一下 JDK 的 ReadWriteLock 吧。

public class Data {

    ...

    private final ReadWriteLock lock = new ReentrantReadWriteLock(); // 创建读写锁
    private final Lock readLock = lock.readLock();    // 获取读锁
    private final Lock writeLock = lock.writeLock();  // 获取写锁

    ...

    public String read() throws InterruptedException {
        readLock.lock(); // 读取上锁
        try {
            return doRead(); // 执行读取操作
        } finally {
            readLock.unlock(); // 读取解锁
        }
    }

    public void write(char c) throws InterruptedException {
        writeLock.lock(); // 写入上锁
        try {
            doWrite(c); // 执行写入操作
        } finally {
            writeLock.unlock(); // 写入解锁
        }
    }

    ...
}

再次运行一把看看效果。

使用了 JDK 的 ReadWriteLock,性能与自己实现的 ReadWriteLock 差不多,大家不妨自己试一下吧。

此外 JDK 还提供了一个更加简单的 ReentrantLock,它可以取代 synchronized,确保获取更高的吞吐率,一般可以这样来做:

以前的做法:

public synchronized void foo() {
    ...
}

现在的做法:

private final Lock lock = new ReentrantLock();

public void foo() {
    lock.lock();
    try {
        ...
    } finally {
        lock.unlock();
    }
}
这里提供两张 synchronized 与 Lock 的性能测试对比:

参考:http://www.ibm.com/developerworks/cn/java/j-jtp10264/index.html



总结

当系统中出现不同的读写线程同时访问某一资源时,需要考虑共享互斥问题,可使用 synchronized 解决次问题。若对性能要求较高的情况下,可考虑使用 ReadWriteLock 接口及其 ReentrantReadWriteLock 实现类,当然,自己实现一个 ReadWriteLock 也是一种解决方案。此外,为了在高并发情况下获取较高的吞吐率,建议使用 Lock 接口及其 ReentrantLock 实现类来替换以前的 synchronized 方法或代码块。

关于 Lock 那点事儿当然还不止这些,今天先写到这里吧,以上内容是否对大家有用,敬请点评!

共有 人打赏支持
黄勇
粉丝 5603
博文 114
码字总数 196279
作品 1
评论 (32)
V神
nice!!
一条大河波浪宽
真好,,,我正要做个类似的需求。。。楼主太好了,么么
anjero
hiahiahia
你要爪子
lz写文章越来越厉害了
laojiadingxi
我是菜鸟,表示没看懂
黄勇
经大量测试证实:JDK 提供的 ReadWriteLock 的性能并没有显著提升,与自己实现的 ReadWriteLock 的性能差不多。在项目中用哪种,完全取决于您。
黄勇

引用来自“laojiadingxi”的评论

我是菜鸟,表示没看懂

请问哪里没看懂呢?:)
黄勇

引用来自“雅典娜拉”的评论

真好,,,我正要做个类似的需求。。。楼主太好了,么么

希望对您有帮助!
RyanHoo
静候《勇哥戏说Java》~
大头圆又圆
嗯,学习,
暗夜的黑魔导
勇哥,问个问题,新线程在start()后,会不会有前面的线程轮到时间片执行了run(),之后这个新线程得以执行第一个run()?我自己写了个和你类似的例子,都没有能重现这种情况(我都尝试了十来次了,而且前面线程开了十个).我就是想说明不同步会出现的问题.
黄勇

引用来自“selwyn224”的评论

勇哥,问个问题,新线程在start()后,会不会有前面的线程轮到时间片执行了run(),之后这个新线程得以执行第一个run()?我自己写了个和你类似的例子,都没有能重现这种情况(我都尝试了十来次了,而且前面线程开了十个).我就是想说明不同步会出现的问题.

如果不用 同步 或 锁定 的方式,A 线程输出的内容可能会被 B 线程修改,所以导致每次输出的字符串里面夹杂这其他字符。
暗夜的黑魔导

引用来自“黄勇”的评论

引用来自“selwyn224”的评论

勇哥,问个问题,新线程在start()后,会不会有前面的线程轮到时间片执行了run(),之后这个新线程得以执行第一个run()?我自己写了个和你类似的例子,都没有能重现这种情况(我都尝试了十来次了,而且前面线程开了十个).我就是想说明不同步会出现的问题.

如果不用 同步 或 锁定 的方式,A 线程输出的内容可能会被 B 线程修改,所以导致每次输出的字符串里面夹杂这其他字符。

这个我知道啊,我问的是一个新线程启动后正要执行run()来进行修改时,在这之前的线程喧宾夺主读取了值.为什么我的例子出不了这种情况呢?你有时间不,要不我加你qq,代码发你帮忙看看?
优雅先生
Thead => Thread
在云端-看世间变幻
例子貌似是 Java多线程设计模式 那本书上的。不过写的不错,赞。
黄勇

引用来自“何以明真心”的评论

Thead => Thread

多谢指正!
黄勇

引用来自“在云端-看世间变幻”的评论

例子貌似是 Java多线程设计模式 那本书上的。不过写的不错,赞。

没错!这本书写得很好,小日本写的,可惜现在已经绝版了!
chape
像跟《火影、海贼王》一样,每次期待 《那些事系列》,通俗易懂又实用,谢谢勇哥
黄勇

引用来自“chape”的评论

像跟《火影、海贼王》一样,每次期待 《那些事系列》,通俗易懂又实用,谢谢勇哥

感谢您支持《Java 那点事儿》!
MattZhou
不错,例子精简,表达诙谐、到位,赞!
×
黄勇
如果觉得我的文章对您有用,请随意打赏。您的支持将鼓励我继续创作!
* 金额(元)
¥1 ¥5 ¥10 ¥20 其他金额
打赏人
留言
* 支付类型
微信扫码支付
打赏金额:
已支付成功
打赏金额: