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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
#
# How to use:
#
# db = PStore.new("/tmp/foo")
# db.transaction do
# p db.roots
# ary = db["root"] = [1,2,3,4]
# ary[0] = [1,1.5]
# end
# db.transaction do
# p db["root"]
# end
require "marshal"
class PStore
class Error < StandardError
end
def initialize(file)
dir = File::dirname(file)
unless File::directory? dir
raise PStore::Error, format("directory %s does not exist", dir)
end
unless File::writable? dir
raise PStore::Error, format("directory %s not writable", dir)
end
if File::exist? file and not File::readable? file
raise PStore::Error, format("file %s not readable", file)
end
@transaction = false
@filename = file
@abort = false
end
def in_transaction
raise PStore::Error, "not in transaction" unless @transaction
end
private :in_transaction
def [](name)
in_transaction
value = @table[name]
if value == nil
raise PStore::Error, format("undefined root name `%s'", name)
end
value
end
def []=(name, value)
in_transaction
@table[name] = value
end
def roots
in_transaction
@table.keys
end
def root?(name)
in_transaction
@table.key? name
end
def path
@filename
end
def commit
@abort = false
throw :pstore_abort_transaction
end
def abort
@abort = true
throw :pstore_abort_transaction
end
def transaction
raise PStore::Error, "nested transaction" if @transaction
begin
@transaction = true
value = file = nil
begin
File::open(@filename, "r") do |file|
@table = Marshal.load(file)
end
rescue Errno::ENOENT
@table = {}
end
begin
catch(:pstore_abort_transaction) do
value = yield(self)
end
ensure
unless @abort
begin
File::rename @filename, @filename+"~"
rescue Errno::ENOENT
no_orig = true
end
begin
File::open(@filename, "w") do |file|
Marshal::dump(@table, file)
end
rescue
File::rename @filename+"~", @filename unless no_orig
end
end
@abort = false
end
ensure
@transaction = false
end
value
end
end
if __FILE__ == $0
db = PStore.new("/tmp/foo")
db.transaction do
p db.roots
ary = db["root"] = [1,2,3,4]
ary[0] = [1,1.5]
end
db.transaction do
p db["root"]
end
end
|