summaryrefslogtreecommitdiff
path: root/lib/timeout.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/timeout.rb')
-rw-r--r--lib/timeout.rb41
1 files changed, 27 insertions, 14 deletions
diff --git a/lib/timeout.rb b/lib/timeout.rb
index 8c6ccb71289..00e60aff8c3 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) {