/
mutex.rb
75 lines (64 loc) · 1.35 KB
/
mutex.rb
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
class Mutex
def initialize
@owner = nil
end
# Check and only allow it to be marshal'd if there are no waiters.
def marshal_dump
raise "Unable to dump locked mutex" unless @waiters.empty?
1
end
# Implemented because we must since we use marshal_load PLUS we need
# to create AND prime @lock. If we didn't do this, then Marshal
# wouldn't prime the lock anyway.
def marshal_load(bunk)
initialize
end
def locked?
Rubinius.locked?(self)
end
def try_lock
# Locking implies a memory barrier, so we don't need to use
# one explicitly.
if Rubinius.try_lock(self)
return false if @owner == Thread.current
@owner = Thread.current
true
else
false
end
end
def lock
Rubinius.memory_barrier
if @owner == Thread.current
raise ThreadError, "Recursively locking not allowed"
end
Rubinius.lock self
@owner = Thread.current
Rubinius.memory_barrier
return self
end
def unlock
Rubinius.memory_barrier
if @owner != Thread.current
raise ThreadError, "Not owner, #{@owner.inspect} is"
end
@owner = nil
Rubinius.unlock self
end
def synchronize
lock
begin
yield
ensure
unlock
end
end
def sleep(duration=undefined)
unlock
begin
Kernel.sleep(duration)
ensure
lock
end
end
end