如何为我的函数引入超时?

时间:2012-10-04 23:18:57

标签: ruby

我可以使用此代码为我的函数引入超时

require 'timeout'
Timeout::timeout(timeout_period) do
  run_some_code
rescue => err
  do_something_with err
  # and maybe the below?
  raise
end

如何创建另一个我可以为此目的重用的函数my_timeout?我希望能够指定在超时“需要”的情况下被调用的函数再次运行。

然后我会像my_timeout

那样致电my_timeout('function name',int_how_many_times_to repeat_if times_out)
  • my_timeout( “restart_db2_server”,5)
  • my_timeout( “restart_eduserver”,3)
  • my_timeout( “RUN_TEST”,0)

我在Windows7上使用ruby 1.9.3p194 (2012-04-20) [i386-mingw32]

1 个答案:

答案 0 :(得分:1)

def my_timeout repeat_n_times, timeout_period, &block
  Timeout::timeout timeout_period, &block
rescue Timeout::Error => timeout_error
  repeat_n_times -= 1

  if repeat_n_times > 0
    retry
  else
    raise timeout_error
  end
end

# Example

# execute the block with a timeout of 30 and repeat up to 5 times
my_timeout 5, 30 do
  # do the heavy work
end