timeout.rb

timeout.rb

class TimeoutError < StandardError; end
def timeout(sec)
  x = Thread.current
  t = Thread.new do
    begin
      sleep sec
      x.raise TimeoutError.new "Exceed maximum timeout value #{sec}s"
    rescue e
      x.raise e
    end
  end
  return yield(sec)
ensure
  if t
    t.kill
    t.join
  end
end

t1 = Thread.new do
  timeout(3) {
    p 'a'
    sleep 5
    p 'b'
  }
end

t2 = Thread.new do
  timeout(6) {
    p 'c'
    sleep 4
    raise StandardError.new 'error'
    p 'd'
  }
end

[t1, t2].each {|t|
  begin
    t.join.value
  rescue TimeoutError => e
    p e
  end
}