多线程等待所有子线程执行完使用总结（2）——CountDownLatch使用和源码初步分析

baorant24 2023-03-28 原文

问题背景

我们在日常开发和学习过程中，经常会使用到多线程的场景，其中我们经常会碰到，我们代码需要等待某个或者多个线程执行完再开始执行，上一篇文章中（参考 https://blog.51cto.com/baorant24/6059489 ），我们介绍了object的wait()和notify（），以及线程的join()方法来实现，本文将介绍一种新的方案，CountDownLatch类的使用。

问题分析

大家在日常开发和学习过程中，或多或少都使用过CountDownLatch，知道CountDownLatch的一般用法，CountDownLatch类是用在同步，允许一个或多个线程去等待直到另外的线程完成了一组操作。它通过count进行初始化，await方法会阻塞直到当前的count为0,之后所有的线程将被释放。了解CountDownLatch的一般作用后，我们先来一起看看CountDownLatch的源码相关分析。重点源码梳理如下：

public class CountDownLatch {
    // 内部类Sync，继承自AbstractQueuedSynchronizer，并且重写了tryAcquireShared（）和tryReleaseShared方法。
    private static final class Sync extends AbstractQueuedSynchronizer {
        private static final long serialVersionUID = 4982264981922014374L;

        Sync(int count) {
            setState(count);
        }

        int getCount() {
            return getState();
        }

        protected int tryAcquireShared(int acquires) {
            return (getState() == 0) ? 1 : -1;
        }

        protected boolean tryReleaseShared(int releases) {
            // Decrement count; signal when transition to zero
            for (;;) {
                int c = getState();
                if (c == 0)
                    return false;
                int nextc = c - 1;
                if (compareAndSetState(c, nextc))
                    return nextc == 0;
            }
        }
    }

    private final Sync sync;

    // 构造方法传入count计数值，然后构造内部的sync对象
    public CountDownLatch(int count) {
        if (count < 0) throw new IllegalArgumentException("count < 0");
        this.sync = new Sync(count);
    }
    
    
    public void await() throws InterruptedException {
        sync.acquireSharedInterruptibly(1);
    }
    ...

    public void countDown() {
        sync.releaseShared(1);
    }
    
    // 获取计数值
    public long getCount() {
        return sync.getCount();
    }
    ...
}

核心代码分析：（1）await()方法分析 java.util.concurrent.CountDownLatch#await()


    public void await() throws InterruptedException {
       sync.acquireSharedInterruptibly(1);
   }

java.util.concurrent.locks.AbstractQueuedSynchronizer#acquireSharedInterruptibly


    public final void acquireSharedInterruptibly(int arg)
           throws InterruptedException {
       if (Thread.interrupted())
           throw new InterruptedException();
       // 前面提到CountDownLatch类中Sync内部类对tryAcquireShared进行了重写
       if (tryAcquireShared(arg) < 0)
           doAcquireSharedInterruptibly(arg);
   }

java.util.concurrent.CountDownLatch.Sync#tryAcquireShared

              
        // 判断state值是否为0,0的话返回1，不然返回-1。
        protected int tryAcquireShared(int acquires) {
           return (getState() == 0) ? 1 : -1;
       }

所有上面 java.util.concurrent.locks.AbstractQueuedSynchronizer#acquireSharedInterruptibly方法会走doAcquireSharedInterruptibly(arg)，继续分析该方法。 java.util.concurrent.locks.AbstractQueuedSynchronizer#doAcquireSharedInterruptibly


    private void doAcquireSharedInterruptibly(int arg)
       throws InterruptedException {
       final Node node = addWaiter(Node.SHARED);
       try {
           for (;;) {
               final Node p = node.predecessor();
               if (p == head) {
                   int r = tryAcquireShared(arg);
                   // 根据前面的分析知道，这块r在state不等于0时，返回-1，所有state未减少到0时，这块回事for循环自旋一直等待。
                   if (r >= 0) {
                       setHeadAndPropagate(node, r);
                       p.next = null; // help GC
                       return;
                   }
               }
               if (shouldParkAfterFailedAcquire(p, node) &&
                   parkAndCheckInterrupt())
                   throw new InterruptedException();
           }
       } catch (Throwable t) {
           cancelAcquire(node);
           throw t;
       }
   }

（2）countDown()方法分析 java.util.concurrent.CountDownLatch#countDown


    public void countDown() {
       sync.releaseShared(1);
   }

java.util.concurrent.locks.AbstractQueuedSynchronizer#releaseShared


   public final boolean releaseShared(int arg) {
       // 同样，前面提到，tryReleaseShared方法在CountDownLatch类中Sync内部类中进行了重写
       if (tryReleaseShared(arg)) {
           doReleaseShared();
           return true;
       }
       return false;
   }

java.util.concurrent.CountDownLatch.Sync#tryReleaseShared

        protected boolean tryReleaseShared(int releases) {
           // Decrement count; signal when transition to zero
           for (;;) {
               int c = getState();
               if (c == 0)
                   return false;
               int nextc = c - 1;
               // 通过compareAndSet保证state-1的同步正确性。
               if (compareAndSetState(c, nextc))
   	    // 当改到0后return TRUE，不然 return FALSE。
                   return nextc == 0;
           }
       }

问题解决

大概梳理了CountDownLatch的源码后，我们实践一个demo具体再看下，代码如下：

import android.os.Bundle
import android.util.Log
import androidx.appcompat.app.AppCompatActivity
import java.util.concurrent.CountDownLatch

class TestCountDownLatchActivity : AppCompatActivity() {
    private var mCountDownLatch: CountDownLatch? = null

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        setContentView(R.layout.activity_test_count_down_latch)

        mCountDownLatch = CountDownLatch(3);

        Thread {
            try {
                Log.d("TAG", "Thread Main A start:" + mCountDownLatch?.count);
                mCountDownLatch?.await();
                Log.d("TAG", "Thread Main A end:" + mCountDownLatch?.count);
            } catch (e: InterruptedException) {
                e.printStackTrace();
            }
        }.start()

        Thread {
            try {
                Log.d("TAG", "Thread Main B start:" + mCountDownLatch?.count);
                mCountDownLatch?.await();
                Log.d("TAG", "Thread Main B end:" + mCountDownLatch?.count);
            } catch (e: InterruptedException) {
                e.printStackTrace();
            }
        }.start()

        Thread {
            Log.d("TAG", "Thread A start:" + mCountDownLatch?.count);
            mCountDownLatch?.countDown();
            Log.d("TAG", "Thread A end:" + mCountDownLatch?.count);
        }.start()

        Thread {
            Log.d("TAG", "Thread B start:" + mCountDownLatch?.count);
            mCountDownLatch?.countDown();
            Log.d("TAG", "Thread B end:" + mCountDownLatch?.count);
        }.start()

        Thread {
            Log.d("TAG", "Thread C start:" + mCountDownLatch?.count);
            mCountDownLatch?.countDown();
            Log.d("TAG", "Thread C end:" + mCountDownLatch?.count);
        }.start()
    }
}

运行结果如下：

结果分析：线程Main A和线程Main B在执行过程中调用mCountDownLatch?.await()方法，会等待mCountDownLatch的计数值减少到0后再继续执行。其余三个线程Thread A、Thread B、 Thread C执行过程将CountDownLatch分别减1，所以这三个线程都执行完后，线程Main A和线程Main B才会继续执行。

问题总结

我们在日常开发和学习过程中，经常会使用到多线程的场景，其中经常会碰到，我们代码需要等待某个或者多个线程执行完再开始执行，上一篇文章中，我们介绍了object的wait()和notify（），以及线程的join()方法来实现，本文介绍了一种新的方案，CountDownLatch类的使用和源码的初步分析，有兴趣的同学可以进一步深入研究。

多线 mdash quot CountDownLatch mCountDownLatch android java 多线程等待Android 移动开发yyds干货盘点

有关多线程等待所有子线程执行完使用总结（2）——CountDownLatch使用和源码初步分析的更多相关文章

ruby - 如何使用 Nokogiri 的 xpath 和 at_xpath 方法 - 2
我正在学习如何使用Nokogiri，根据这段代码我遇到了一些问题:require'rubygems'require'mechanize'post_agent=WWW::Mechanize.newpost_page=post_agent.get('http://www.vbulletin.org/forum/showthread.php?t=230708')puts"\nabsolutepathwithtbodygivesnil"putspost_page.parser.xpath('/html/body/div/div/div/div/div/table/tbody/tr/td/div
ruby - 使用 RubyZip 生成 ZIP 文件时设置压缩级别 - 2
我有一个Ruby程序，它使用rubyzip压缩XML文件的目录树。gem。我的问题是文件开始变得很重，我想提高压缩级别，因为压缩时间不是问题。我在rubyzipdocumentation中找不到一种为创建的ZIP文件指定压缩级别的方法。有人知道如何更改此设置吗？是否有另一个允许指定压缩级别的Ruby库？最佳答案这是我通过查看rubyzip内部创建的代码。level=Zlib::BEST_COMPRESSIONZip::ZipOutputStream.open(zip_file)do|zip|Dir.glob("**/*")d
ruby - 为什么我可以在 Ruby 中使用 Object#send 访问私有(private)/ protected 方法？ - 2
类classAprivatedeffooputs:fooendpublicdefbarputs:barendprivatedefzimputs:zimendprotecteddefdibputs:dibendendA的实例a=A.new测试a.foorescueputs:faila.barrescueputs:faila.zimrescueputs:faila.dibrescueputs:faila.gazrescueputs:fail测试输出failbarfailfailfail.发送测试[:foo,:bar,:zim,:dib,:gaz].each{|m|a.send(m)resc
ruby-on-rails - 使用 Ruby on Rails 进行自动化测试 - 最佳实践 - 2
很好奇，就使用rubyonrails自动化单元测试而言，你们正在做什么？您是否创建了一个脚本来在cron中运行rake作业并将结果邮寄给您？git中的预提交Hook？只是手动调用？我完全理解测试，但想知道在错误发生之前捕获错误的最佳实践是什么。让我们理所当然地认为测试本身是完美无缺的，并且可以正常工作。下一步是什么以确保他们在正确的时间将可能有害的结果传达给您？最佳答案不确定您到底想听什么，但是有几个级别的自动代码库控制:在处理某项功能时，您可以使用类似autotest的内容获得关于哪些有效，哪些无效的即时反馈。要确保您的提
ruby - 在 Ruby 中使用匿名模块 - 2
假设我做了一个模块如下:m=Module.newdoclassCendend三个问题:除了对m的引用之外，还有什么方法可以访问C和m中的其他内容？我可以在创建匿名模块后为其命名吗(就像我输入“module...”一样)？如何在使用完匿名模块后将其删除，使其定义的常量不再存在？最佳答案三个答案:是的，使用ObjectSpace.此代码使c引用你的类(class)C不引用m:c=nilObjectSpace.each_object{|obj|c=objif(Class===objandobj.name=~/::C$/)}当然这取决于
ruby - 使用 ruby 和 savon 的 SOAP 服务 - 2
我正在尝试使用ruby和Savon来使用网络服务。测试服务为http://www.webservicex.net/WS/WSDetails.aspx?WSID=9&CATID=2require'rubygems'require'savon'client=Savon::Client.new"http://www.webservicex.net/stockquote.asmx?WSDL"client.get_quotedo|soap|soap.body={:symbol=>"AAPL"}end返回SOAP异常。检查soap信封，在我看来soap请求没有正确的命名空间。任何人都可以建议我
python - 如何使用 Ruby 或 Python 创建一系列高音调和低音调的蜂鸣声？ - 2
关闭。这个问题是opinion-based.它目前不接受答案。想要改进这个问题？更新问题，以便editingthispost可以用事实和引用来回答它.关闭4年前。Improvethisquestion我想在固定时间创建一系列低音和高音调的哔哔声。例如:在150毫秒时发出高音调的蜂鸣声在151毫秒时发出低音调的蜂鸣声200毫秒时发出低音调的蜂鸣声250毫秒的高音调蜂鸣声有没有办法在Ruby或Python中做到这一点？我真的不在乎输出编码是什么(.wav、.mp3、.ogg等等)，但我确实想创建一个输出文件。
ruby-on-rails - 'compass watch' 是如何工作的/它是如何与 rails 一起使用的 - 2
我在我的项目目录中完成了compasscreate.和compassinitrails。几个问题:我已将我的.sass文件放在public/stylesheets中。这是放置它们的正确位置吗？当我运行compasswatch时，它不会自动编译这些.sass文件。我必须手动指定文件:compasswatchpublic/stylesheets/myfile.sass等。如何让它自动运行？文件ie.css、print.css和screen.css已放在stylesheets/compiled。如何在编译后不让它们重新出现的情况下删除它们？我自己编译的.sass文件编译成compiled/t
ruby - 使用 ruby 将 HTML 转换为纯文本并维护结构/格式 - 2
我想将html转换为纯文本。不过，我不想只删除标签，我想智能地保留尽可能多的格式。为插入换行符标签，检测段落并格式化它们等。输入非常简单，通常是格式良好的html(不是整个文档，只是一堆内容，通常没有anchor或图像)。我可以将几个正则表达式放在一起，让我达到80%，但我认为可能有一些现有的解决方案更智能。最佳答案首先，不要尝试为此使用正则表达式。很有可能你会想出一个脆弱/脆弱的解决方案，它会随着HTML的变化而崩溃，或者很难管理和维护。您可以使用Nokogiri快速解析HTML并提取文本:require'nokogiri'h
ruby - 在 64 位 Snow Leopard 上使用 rvm、postgres 9.0、ruby 1.9.2-p136 安装 pg gem 时出现问题 - 2
我想为Heroku构建一个Rails3应用程序。他们使用Postgres作为他们的数据库，所以我通过MacPorts安装了postgres9.0。现在我需要一个postgresgem并且共识是出于性能原因你想要pggem。但是我对我得到的错误感到非常困惑当我尝试在rvm下通过geminstall安装pg时。我已经非常明确地指定了所有postgres目录的位置可以找到但仍然无法完成安装:$envARCHFLAGS='-archx86_64'geminstallpg--\--with-pg-config=/opt/local/var/db/postgresql90/defaultdb/po