RubyでJavaのCountDownLatchを実装してみる

テストコードで別スレッドの作業の完了待ち的な処理を行いたく、Rubyの標準ライブラリでJavaのCountDownLatch的なクラスは無いものかと探してみたものの見つからなかったので、試しにJava同じようなインターフェースでシンプル実装してみました。(実用には耐えないと思います)

class CountDownLatch
  def initialize(count)
    raise ArgumentError, 'count < 0' if count < 0
    @count = count
    @lock = Monitor.new
    @condition = @lock.new_cond
  end

  def count_down
    @lock.synchronize do
      @count -= 1 if @count > 0
      @condition.broadcast if @count == 0
    end
  end

  def await
    @lock.synchronize do
      @condition.wait if @count > 0
    end
  end
end

count = 10
latch = CountDownLatch.new(count)
t1 = Thread.new do
  latch.await
  puts 'Finished.'
end

t2 = Thread.new do
  count.times do |i|
    puts "Count #{i}"
    latch.count_down
  end
end

t1.join
t2.join

出力結果

Count 0
Count 1
Count 2
Count 3
Count 4
Count 5
Count 6
Count 7
Count 8
Count 9
Finished.