1999-08-13 01:45:20 -04:00
|
|
|
#
|
|
|
|
# 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
|
|
|
|
|
|
|
|
def timeout(sec)
|
|
|
|
begin
|
|
|
|
x = Thread.current
|
|
|
|
y = Thread.start {
|
|
|
|
sleep sec
|
2000-01-04 23:41:21 -05:00
|
|
|
x.raise TimeoutError, "execution expired" if x.alive?
|
1999-08-13 01:45:20 -04:00
|
|
|
}
|
|
|
|
yield sec
|
|
|
|
return true
|
|
|
|
ensure
|
2000-01-04 23:41:21 -05:00
|
|
|
Thread.kill y if y.alive?
|
1999-08-13 01:45:20 -04:00
|
|
|
end
|
|
|
|
end
|