redis-rb で watch

redis_watch.rb

require 'redis'
require 'hiredis'

# usage:
# 端末(1) : 更新し続ける
#           watch -n 0.1 "echo incr r | redis-cli"
# 端末(2) : watch しつつ更新する
#           ruby redis_watch.rb 1000000

r = Redis.new(driver: :hiredis, timeout: 60) # デフォルトだと、5秒でタイムアウト
puts "challenge!"
r.watch 'x' do
  x = r.get 'x'
  res = r.multi do |m|
    (1..ARGV[0].to_i).each do |i|
      m.incr 'x'
    end
  end
  # 途中で書き換えられると res は nil になる
  # うまくいっていれば、multi の結果が res に入っている
  p res
end

# リトライする場合
loop do
  r = Redis.new(driver: :hiredis, timeout: 60)
  puts "challenge!"
  res = nil
  r.watch 'x' do
    x = r.get 'x'
    res = r.multi do |m|
      (1..ARGV[0].to_i).each do |i|
        m.incr 'x'
      end
    end
  end
  # 途中で書き換えられると res は nil のまま
  # watch のブロックの中から呼ぶと、break はブロックから抜けるだけで
  # loop から抜けられない
  if res
    p res
    break
  end
end