mirror of
https://github.com/ruby/ruby.git
synced 2022-11-09 12:17:21 -05:00
df96f994f1
expression interpolation. * pack.c (pack_unpack): should give length to utf8_to_uv(). * pack.c (utf8_to_uv): add length check. * massages: replace "wrong #" by "wrong number". * marshal.c (w_float): output Infinity and NaN explicitly. * marshal.c (r_object): support new explicit float format. * eval.c (rb_thread_wait_for): select may cause ERESTART on Solaris. * eval.c (rb_thread_select): ditto. * array.c (rb_ary_join): dumped core if sep is not T_STRING nor T_NIL. * array.c (rb_ary_join): buffer size calculattion was wrong. * array.c (rb_ary_to_s): if rb_output_fs is nil, insert newlines between array elements (use rb_default_rs as newline litral) [experimental]. * gc.c (init_mark_stack): no need to clear mark_stack. * gc.c (gc_mark_all): need to handle finalizer mark. * gc.c (gc_mark_rest): use MEMCPY instead of memcpy. * gc.c (rb_gc_mark): earlier const check to avoid pusing special constants into mark stack. * numeric.c (fix_to_s): 'to_s' now takes optional argument to specify radix. [new] * bignum.c (rb_big_to_s): ditto. [new] git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@1842 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
159 lines
3 KiB
Ruby
159 lines
3 KiB
Ruby
#
|
|
# 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 "ftools"
|
|
require "md5"
|
|
|
|
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
|
|
unless @table.key? name
|
|
raise PStore::Error, format("undefined root name `%s'", name)
|
|
end
|
|
@table[name]
|
|
end
|
|
def []=(name, value)
|
|
in_transaction
|
|
@table[name] = value
|
|
end
|
|
def delete(name)
|
|
in_transaction
|
|
@table.delete name
|
|
end
|
|
|
|
def roots
|
|
in_transaction
|
|
@table.keys
|
|
end
|
|
def root?(name)
|
|
in_transaction
|
|
@table.key? name
|
|
end
|
|
def path
|
|
@filename
|
|
end
|
|
|
|
def commit
|
|
in_transaction
|
|
@abort = false
|
|
throw :pstore_abort_transaction
|
|
end
|
|
def abort
|
|
in_transaction
|
|
@abort = true
|
|
throw :pstore_abort_transaction
|
|
end
|
|
|
|
def transaction(read_only=false)
|
|
raise PStore::Error, "nested transaction" if @transaction
|
|
begin
|
|
@transaction = true
|
|
value = nil
|
|
backup = @filename+"~"
|
|
begin
|
|
file = File::open(@filename, "rb+")
|
|
orig = true
|
|
rescue Errno::ENOENT
|
|
raise if read_only
|
|
file = File::open(@filename, "wb+")
|
|
end
|
|
file.flock(read_only ? File::LOCK_SH : File::LOCK_EX)
|
|
if read_only
|
|
@table = Marshal::load(file)
|
|
elsif orig
|
|
content = file.read
|
|
@table = Marshal::load(content)
|
|
size = content.size
|
|
md5 = MD5.new(content).digest
|
|
content = nil # unreference huge data
|
|
else
|
|
@table = {}
|
|
end
|
|
begin
|
|
catch(:pstore_abort_transaction) do
|
|
value = yield(self)
|
|
end
|
|
rescue Exception
|
|
@abort = true
|
|
raise
|
|
ensure
|
|
if !read_only && !@abort
|
|
file.rewind
|
|
content = Marshal::dump(@table)
|
|
if !md5 || size != content.size || md5 != MD5.new(content).digest
|
|
File::copy @filename, backup
|
|
begin
|
|
file.write(content)
|
|
file.truncate(file.pos)
|
|
content = nil # unreference huge data
|
|
rescue
|
|
File::rename backup, @filename if File::exist?(backup)
|
|
raise
|
|
end
|
|
end
|
|
end
|
|
@abort = false
|
|
end
|
|
ensure
|
|
@table = nil
|
|
@transaction = false
|
|
file.close if file
|
|
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[1] = [1,1.5]
|
|
end
|
|
|
|
1000.times do
|
|
db.transaction do
|
|
db["root"][0] += 1
|
|
p db["root"][0]
|
|
end
|
|
end
|
|
|
|
db.transaction(true) do
|
|
p db["root"]
|
|
end
|
|
end
|