summaryrefslogtreecommitdiffstats
path: root/lib/puppet/util/file_locking.rb
blob: 101ece3085785f2446f272dc2cb8a01cfdef3443 (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
require 'puppet/util'

module Puppet::Util::FileLocking
    module_function

    # Create a shared lock for reading
    def readlock(file)
        Puppet::Util.sync(file).synchronize(Sync::SH) do
            File.open(file) { |f|
                f.lock_shared { |lf| yield lf }
            }
        end
    end

    # Create an exclusive lock for writing, and do the writing in a
    # tmp file.
    def writelock(file, mode = nil)
        unless FileTest.directory?(File.dirname(file))
            raise Puppet::DevError, "Cannot create %s; directory %s does not exist" % [file, File.dirname(file)]
        end
        tmpfile = file + ".tmp"

        unless mode
            # It's far more likely that the file will be there than not, so it's
            # better to stat once to check for existence and mode.
            # If we can't stat, it's most likely because the file's not there,
            # but could also be because the directory isn't readable, in which case
            # we won't be able to write anyway.
            begin
                mode = File.stat(file).mode
            rescue
                mode = 0600
            end
        end

        Puppet::Util.sync(file).synchronize(Sync::EX) do
            File.open(file, "w", mode) do |rf|
                rf.lock_exclusive do |lrf|
                    yield lrf
                end
            end
        end
    end
end