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
80
81
82
83
84
85
86
87
88
89
90
91
|
# $Id$
module Puppet
# a class for storing state
class Storage
include Singleton
def initialize
self.class.load
end
def self.clear
@@state = nil
Storage.init
end
def self.init
Puppet.debug "Initializing Storage"
@@state = Hash.new { |hash,key|
hash[key] = Hash.new(nil)
}
@@splitchar = "\t"
end
self.init
def self.load
if Puppet[:checksumfile].nil?
raise "Somehow the statefile is nil"
end
unless File.exists?(Puppet[:checksumfile])
Puppet.info "Statefile %s does not exist" % Puppet[:checksumfile]
return
end
#Puppet.debug "Loading statefile %s" % Puppet[:checksumfile]
File.open(Puppet[:checksumfile]) { |file|
file.each { |line|
myclass, key, value = line.split(@@splitchar)
begin
@@state[eval(myclass)][key] = Marshal::load(value)
rescue => detail
raise RuntimeError,
"Failed to load value for %s::%s => %s" % [
myclass,key,detail
], caller
end
}
}
#Puppet.debug "Loaded state is %s" % @@state.inspect
end
def self.state(myclass)
unless myclass.is_a? Class
myclass = myclass.class
end
result = @@state[myclass]
return result
end
def self.store
unless FileTest.directory?(File.dirname(Puppet[:checksumfile]))
begin
Puppet.recmkdir(File.dirname(Puppet[:checksumfile]))
Puppet.info "Creating state directory %s" %
File.dirname(Puppet[:checksumfile])
rescue => detail
Puppet.err "Could not create state file: %s" % detail
return
end
end
unless FileTest.exist?(Puppet[:checksumfile])
Puppet.info "Creating state file %s" % Puppet[:checksumfile]
end
File.open(Puppet[:checksumfile], File::CREAT|File::WRONLY, 0600) { |file|
@@state.each { |klass, thash|
thash.each { |key,value|
mvalue = Marshal::dump(value)
file.puts([klass,key,mvalue].join(@@splitchar))
}
}
}
#Puppet.debug "Stored state is %s" % @@state.inspect
end
end
end
|
