python中实现闭锁

在java的java.util.concurrent工具类有闭锁的一个灵活的实现CountDownLatch
在Python中我们可以借助 threading 库中的Condition来实现CountDownLatch

python 版本的CountDownLatch

from threading import Condition
class CountDownLatch:
    def __init__(self, count):
        self.count = count
        self.condition = Condition()

    def await(self):
        self.condition.acquire()
        try:
            while self.count > 0:
                self.condition.wait()
        finally:
            self.condition.release()

    def count_down(self):
        self.condition.acquire()
        self.count -= 1
        if self.count <= 0:
            self.condition.notifyAll()
        self.condition.release()