blob: 024fedf3de2422ae15eeee866834e4f9720eac27 (
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
|
require 'thread'
require 'sync'
# Gotten from:
# http://path.berkeley.edu/~vjoel/ruby/solaris-bug.rb
# Extensions to the File class for exception-safe file locking in a
# environment with multiple user threads.
# This is here because closing a file on solaris unlocks any locks that
# other threads might have. So we have to make sure that only the last
# reader thread closes the file.
#
# The hash maps inode number to a count of reader threads
$reader_count = Hash.new(0)
class File
# Get an exclusive (i.e., write) lock on the file, and yield to the block.
# If the lock is not available, wait for it without blocking other ruby
# threads.
def lock_exclusive
if Thread.list.size == 1
flock(LOCK_EX)
else
# ugly hack because waiting for a lock in a Ruby thread blocks the
# process
period = 0.001
until flock(LOCK_EX|LOCK_NB)
sleep period
period *= 2 if period < 1
end
end
yield self
ensure
flush
flock(LOCK_UN)
end
# Get a shared (i.e., read) lock on the file, and yield to the block.
# If the lock is not available, wait for it without blocking other ruby
# threads.
def lock_shared
if Thread.list.size == 1
flock(LOCK_SH)
else
# ugly hack because waiting for a lock in a Ruby thread blocks the
# process
period = 0.001
until flock(LOCK_SH|LOCK_NB)
sleep period
period *= 2 if period < 1
end
end
yield self
ensure
Thread.exclusive {flock(LOCK_UN) if $reader_count[self.stat.ino] == 1}
## for solaris, no need to unlock here--closing does it
## but this has no effect on the bug
end
end
|
