blob: 7e3ee81ed004a48598d6f389de067ac208d33747 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#
# timeout.rb -- execution timeout
#
# Copyright (C) 2000 Network Applied Communication Laboratory, Inc.
# Copyright (C) 2000 Information-technology Promotion Agency, Japan
#
#= 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 termination.
#
# : [exception]
#
# The exception class to be raised on timeout.
#
#=end
module Timeout
class Error<Interrupt
end
def timeout(sec, exception=Error)
return yield if sec == nil or sec.zero?
raise ThreadError, "timeout within critical session" if Thread.critical
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, e=Timeout::Error, &block)
Timeout::timeout(n, e, &block)
end
TimeoutError = Timeout::Error
if __FILE__ == $0
p timeout(5) {
45
}
p timeout(5, TimeoutError) {
45
}
p timeout(nil) {
54
}
p timeout(0) {
54
}
p timeout(5) {
loop {
p 10
sleep 1
}
}
end
|
