diff options
Diffstat (limited to 'lib/timeout.rb')
-rw-r--r-- | lib/timeout.rb | 41 |
1 files changed, 27 insertions, 14 deletions
diff --git a/lib/timeout.rb b/lib/timeout.rb index 8c6ccb7128..00e60aff8c 100644 --- a/lib/timeout.rb +++ b/lib/timeout.rb @@ -23,25 +23,38 @@ # # The time in seconds to wait for block teminatation. # +# : [exception] +# +# The exception classs to be raised on timeout. +# #=end -class TimeoutError<Interrupt -end +module Timeout + class Error<Interrupt + end -def timeout(sec, exception=TimeoutError) - return yield if sec == nil - begin - x = Thread.current - y = Thread.start { - sleep sec - x.raise exception, "execution expired" if x.alive? - } - yield sec -# return true - ensure - y.kill if y and y.alive? + def timeout(sec, exception=Error) + return yield if sec == nil + begin + x = Thread.current + y = Thread.start { + sleep sec + x.raise exception, "execution expired" if x.alive? + } + yield sec + # return true + ensure + y.kill if y and y.alive? + end end + module_function :timeout +end + +# compatible +def timeout(n, &block) + Timeout::timeout(n, &block) end +TimeoutError = Timeout::Error if __FILE__ == $0 p timeout(5) { |