mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
43 lines
749 B
Ruby
43 lines
749 B
Ruby
|
#
|
||
|
# timeout.rb -- execution timeout
|
||
|
#
|
||
|
#= SYNOPSIS
|
||
|
#
|
||
|
# require 'timeout'
|
||
|
# status = timeout(5) {
|
||
|
# # something may take time
|
||
|
# }
|
||
|
#
|
||
|
#= DESCRIPTION
|
||
|
#
|
||
|
# timeout executes the block. If the block execution terminates successfully
|
||
|
# before timeout, it returns true. If not, it terminates the execution and
|
||
|
# raise TimeoutError exception.
|
||
|
#
|
||
|
#== Parameters
|
||
|
#
|
||
|
# : timout
|
||
|
#
|
||
|
# The time in seconds to wait for block teminatation.
|
||
|
#
|
||
|
#=end
|
||
|
|
||
|
class TimeoutError<StandardError
|
||
|
end
|
||
|
|
||
|
Thread.abort_on_exception = true
|
||
|
|
||
|
def timeout(sec)
|
||
|
begin
|
||
|
x = Thread.current
|
||
|
y = Thread.start {
|
||
|
sleep sec
|
||
|
x.raise TimeoutError, "execution expired" if x.status
|
||
|
}
|
||
|
yield sec
|
||
|
return true
|
||
|
ensure
|
||
|
Thread.kill y if y.status
|
||
|
end
|
||
|
end
|