Gecko [ nxtxgen.com ]

Name	Size	Permission
bigdecimal	[ DIR ]	drwxr-xr-x
cgi	[ DIR ]	drwxr-xr-x
date	[ DIR ]	drwxr-xr-x
digest	[ DIR ]	drwxr-xr-x
dl	[ DIR ]	drwxr-xr-x
drb	[ DIR ]	drwxr-xr-x
io	[ DIR ]	drwxr-xr-x
irb	[ DIR ]	drwxr-xr-x
net	[ DIR ]	drwxr-xr-x
openssl	[ DIR ]	drwxr-xr-x
optparse	[ DIR ]	drwxr-xr-x
racc	[ DIR ]	drwxr-xr-x
rdoc	[ DIR ]	drwxr-xr-x
rexml	[ DIR ]	drwxr-xr-x
rinda	[ DIR ]	drwxr-xr-x
rss	[ DIR ]	drwxr-xr-x
runit	[ DIR ]	drwxr-xr-x
shell	[ DIR ]	drwxr-xr-x
soap	[ DIR ]	drwxr-xr-x
test	[ DIR ]	drwxr-xr-x
uri	[ DIR ]	drwxr-xr-x
webrick	[ DIR ]	drwxr-xr-x
wsdl	[ DIR ]	drwxr-xr-x
x86_64-linux	[ DIR ]	drwxr-xr-x
xmlrpc	[ DIR ]	drwxr-xr-x
xsd	[ DIR ]	drwxr-xr-x
yaml	[ DIR ]	drwxr-xr-x
English.rb	5.6 KB	-rw-r--r--
Env.rb	274 B	-rw-r--r--
abbrev.rb	2.5 KB	-rw-r--r--
base64.rb	3.37 KB	-rw-r--r--
benchmark.rb	17.73 KB	-rw-r--r--
cgi-lib.rb	6.89 KB	-rw-r--r--
cgi.rb	73.74 KB	-rw-r--r--
complex.rb	12.84 KB	-rw-r--r--
csv.rb	24.46 KB	-rw-r--r--
date.rb	53.02 KB	-rw-r--r--
date2.rb	128 B	-rw-r--r--
debug.rb	20.61 KB	-rw-r--r--
delegate.rb	8.81 KB	-rw-r--r--
digest.rb	1.12 KB	-rw-r--r--
drb.rb	19 B	-rw-r--r--
e2mmap.rb	4.04 KB	-rw-r--r--
erb.rb	21.38 KB	-rw-r--r--
eregex.rb	487 B	-rw-r--r--
expect.rb	633 B	-rw-r--r--
fileutils.rb	42.23 KB	-rw-r--r--
finalize.rb	5.38 KB	-rw-r--r--
find.rb	1.84 KB	-rw-r--r--
forwardable.rb	6.16 KB	-rw-r--r--
ftools.rb	6.17 KB	-rw-r--r--
generator.rb	8.1 KB	-rw-r--r--
getoptlong.rb	14.88 KB	-rw-r--r--
getopts.rb	2.25 KB	-rw-r--r--
gserver.rb	6.43 KB	-rw-r--r--
importenv.rb	590 B	-rw-r--r--
ipaddr.rb	21.96 KB	-rw-r--r--
irb.rb	7.43 KB	-rw-r--r--
jcode.rb	4.3 KB	-rw-r--r--
kconv.rb	8.12 KB	-rw-r--r--
logger.rb	17.59 KB	-rw-r--r--
mailread.rb	1.28 KB	-rw-r--r--
mathn.rb	5.42 KB	-rw-r--r--
matrix.rb	27.21 KB	-rw-r--r--
md5.rb	411 B	-rw-r--r--
mkmf.rb	50.65 KB	-rw-r--r--
monitor.rb	7.93 KB	-rw-r--r--
mutex_m.rb	2.07 KB	-rw-r--r--
observer.rb	5.15 KB	-rw-r--r--
open-uri.rb	20.49 KB	-rw-r--r--
open3.rb	2.1 KB	-rw-r--r--
openssl.rb	575 B	-rw-r--r--
optparse.rb	47.12 KB	-rw-r--r--
ostruct.rb	3.35 KB	-rw-r--r--
parsearg.rb	1.55 KB	-rw-r--r--
parsedate.rb	1.33 KB	-rw-r--r--
pathname.rb	29.39 KB	-rw-r--r--
ping.rb	1.48 KB	-rw-r--r--
pp.rb	15.97 KB	-rw-r--r--
prettyprint.rb	18.33 KB	-rw-r--r--
profile.rb	90 B	-rw-r--r--
profiler.rb	1.59 KB	-rw-r--r--
pstore.rb	11.15 KB	-rw-r--r--
rational.rb	12.05 KB	-rw-r--r--
readbytes.rb	835 B	-rw-r--r--
resolv-replace.rb	1.55 KB	-rw-r--r--
resolv.rb	56.83 KB	-rw-r--r--
rss.rb	504 B	-rw-r--r--
rubyunit.rb	180 B	-rw-r--r--
scanf.rb	20.63 KB	-rw-r--r--
securerandom.rb	4.27 KB	-rw-r--r--
set.rb	27.08 KB	-rw-r--r--
sha1.rb	418 B	-rw-r--r--
shell.rb	4.66 KB	-rw-r--r--
shellwords.rb	3.99 KB	-rw-r--r--
singleton.rb	8.08 KB	-rw-r--r--
sync.rb	6.09 KB	-rw-r--r--
tempfile.rb	4.86 KB	-rw-r--r--
thread.rb	104 B	-rw-r--r--
thwait.rb	4.32 KB	-rw-r--r--
time.rb	31.58 KB	-rw-r--r--
timeout.rb	3 KB	-rw-r--r--
tmpdir.rb	3.69 KB	-rw-r--r--
tracer.rb	2.73 KB	-rw-r--r--
tsort.rb	7.99 KB	-rw-r--r--
un.rb	4.54 KB	-rw-r--r--
uri.rb	710 B	-rw-r--r--
weakref.rb	2.68 KB	-rw-r--r--
webrick.rb	811 B	-rw-r--r--
yaml.rb	12.36 KB	-rw-r--r--

Code Editor : pstore.rb

# = PStore -- Transactional File Storage for Ruby Objects
#
# pstore.rb -
#   originally by matz
#   documentation by Kev Jackson and James Edward Gray II
#
# See PStore for documentation.

require "fileutils"
require "digest/md5"

#
# PStore implements a file based persistence mechanism based on a Hash.  User
# code can store hierarchies of Ruby objects (values) into the data store file
# by name (keys).  An object hierarchy may be just a single object.  User code 
# may later read values back from the data store or even update data, as needed.
# 
# The transactional behavior ensures that any changes succeed or fail together.
# This can be used to ensure that the data store is not left in a transitory
# state, where some values were updated but others were not.
# 
# Behind the scenes, Ruby objects are stored to the data store file with 
# Marshal.  That carries the usual limitations.  Proc objects cannot be 
# marshalled, for example.
#
# == Usage example:
# 
#  require "pstore"
#  
#  # a mock wiki object...
#  class WikiPage
#    def initialize( page_name, author, contents )
#      @page_name = page_name
#      @revisions = Array.new
#      
#      add_revision(author, contents)
#    end
#    
#    attr_reader :page_name
#    
#    def add_revision( author, contents )
#      @revisions << { :created  => Time.now,
#                      :author   => author,
#                      :contents => contents }
#    end
#    
#    def wiki_page_references
#      [@page_name] + @revisions.last[:contents].scan(/\b(?:[A-Z]+[a-z]+){2,}/)
#    end
#    
#    # ...
#  end
#  
#  # create a new page...
#  home_page = WikiPage.new( "HomePage", "James Edward Gray II",
#                            "A page about the JoysOfDocumentation..." )
#  
#  # then we want to update page data and the index together, or not at all...
#  wiki = PStore.new("wiki_pages.pstore")
#  wiki.transaction do  # begin transaction; do all of this or none of it
#    # store page...
#    wiki[home_page.page_name] = home_page
#    # ensure that an index has been created...
#    wiki[:wiki_index] ||= Array.new
#    # update wiki index...
#    wiki[:wiki_index].push(*home_page.wiki_page_references)
#  end                   # commit changes to wiki data store file
#  
#  ### Some time later... ###
#  
#  # read wiki data...
#  wiki.transaction(true) do  # begin read-only transaction, no changes allowed
#    wiki.roots.each do |data_root_name|
#      p data_root_name
#      p wiki[data_root_name]
#    end
#  end
#
class PStore
  binmode = defined?(File::BINARY) ? File::BINARY : 0
  RDWR_ACCESS = File::RDWR | File::CREAT | binmode
  RD_ACCESS = File::RDONLY | binmode
  WR_ACCESS = File::WRONLY | File::CREAT | File::TRUNC | binmode

# The error type thrown by all PStore methods.
  class Error < StandardError
  end

# 
  # To construct a PStore object, pass in the _file_ path where you would like 
  # the data to be stored.
  # 
  def initialize(file)
    dir = File::dirname(file)
    unless File::directory? dir
      raise PStore::Error, format("directory %s does not exist", 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

# Raises PStore::Error if the calling code is not in a PStore#transaction.
  def in_transaction
    raise PStore::Error, "not in transaction" unless @transaction
  end
  # 
  # Raises PStore::Error if the calling code is not in a PStore#transaction or
  # if the code is in a read-only PStore#transaction.
  # 
  def in_transaction_wr()
    in_transaction()
    raise PStore::Error, "in read-only transaction" if @rdonly
  end
  private :in_transaction, :in_transaction_wr

#
  # Retrieves a value from the PStore file data, by _name_.  The hierarchy of 
  # Ruby objects stored under that root _name_ will be returned.
  # 
  # *WARNING*:  This method is only valid in a PStore#transaction.  It will
  # raise PStore::Error if called at any other time.
  #
  def [](name)
    in_transaction
    @table[name]
  end
  #
  # This method is just like PStore#[], save that you may also provide a 
  # _default_ value for the object.  In the event the specified _name_ is not 
  # found in the data store, your _default_ will be returned instead.  If you do 
  # not specify a default, PStore::Error will be raised if the object is not 
  # found.
  # 
  # *WARNING*:  This method is only valid in a PStore#transaction.  It will
  # raise PStore::Error if called at any other time.
  #
  def fetch(name, default=PStore::Error)
    in_transaction
    unless @table.key? name
      if default==PStore::Error
	raise PStore::Error, format("undefined root name `%s'", name)
      else
	return default
      end
    end
    @table[name]
  end
  #
  # Stores an individual Ruby object or a hierarchy of Ruby objects in the data
  # store file under the root _name_.  Assigning to a _name_ already in the data
  # store clobbers the old data.
  # 
  # == Example:
  # 
  #  require "pstore"
  #  
  #  store = PStore.new("data_file.pstore")
  #  store.transaction do  # begin transaction
  #    # load some data into the store...
  #    store[:single_object] = "My data..."
  #    store[:obj_heirarchy] = { "Kev Jackson" => ["rational.rb", "pstore.rb"],
  #                              "James Gray"  => ["erb.rb", "pstore.rb"] }
  #  end                   # commit changes to data store file
  # 
  # *WARNING*:  This method is only valid in a PStore#transaction and it cannot
  # be read-only.  It will raise PStore::Error if called at any other time.
  #
  def []=(name, value)
    in_transaction_wr()
    @table[name] = value
  end
  #
  # Removes an object hierarchy from the data store, by _name_.
  # 
  # *WARNING*:  This method is only valid in a PStore#transaction and it cannot
  # be read-only.  It will raise PStore::Error if called at any other time.
  #
  def delete(name)
    in_transaction_wr()
    @table.delete name
  end

#
  # Returns the names of all object hierarchies currently in the store.
  # 
  # *WARNING*:  This method is only valid in a PStore#transaction.  It will
  # raise PStore::Error if called at any other time.
  #
  def roots
    in_transaction
    @table.keys
  end
  #
  # Returns true if the supplied _name_ is currently in the data store.
  # 
  # *WARNING*:  This method is only valid in a PStore#transaction.  It will
  # raise PStore::Error if called at any other time.
  #
  def root?(name)
    in_transaction
    @table.key? name
  end
  # Returns the path to the data store file.
  def path
    @filename
  end

#
  # Ends the current PStore#transaction, committing any changes to the data
  # store immediately.
  # 
  # == Example:
  # 
  #  require "pstore"
  #   
  #  store = PStore.new("data_file.pstore")
  #  store.transaction do  # begin transaction
  #    # load some data into the store...
  #    store[:one] = 1
  #    store[:two] = 2
  #  
  #    store.commit        # end transaction here, committing changes
  #  
  #    store[:three] = 3   # this change is never reached
  #  end
  # 
  # *WARNING*:  This method is only valid in a PStore#transaction.  It will
  # raise PStore::Error if called at any other time.
  #
  def commit
    in_transaction
    @abort = false
    throw :pstore_abort_transaction
  end
  #
  # Ends the current PStore#transaction, discarding any changes to the data
  # store.
  # 
  # == Example:
  # 
  #  require "pstore"
  #   
  #  store = PStore.new("data_file.pstore")
  #  store.transaction do  # begin transaction
  #    store[:one] = 1     # this change is not applied, see below...
  #    store[:two] = 2     # this change is not applied, see below...
  #  
  #    store.abort         # end transaction here, discard all changes
  #  
  #    store[:three] = 3   # this change is never reached
  #  end
  # 
  # *WARNING*:  This method is only valid in a PStore#transaction.  It will
  # raise PStore::Error if called at any other time.
  #
  def abort
    in_transaction
    @abort = true
    throw :pstore_abort_transaction
  end

#
  # Opens a new transaction for the data store.  Code executed inside a block
  # passed to this method may read and write data to and from the data store 
  # file.
  # 
  # At the end of the block, changes are committed to the data store
  # automatically.  You may exit the transaction early with a call to either 
  # PStore#commit or PStore#abort.  See those methods for details about how
  # changes are handled.  Raising an uncaught Exception in the block is 
  # equivalent to calling PStore#abort.
  # 
  # If _read_only_ is set to +true+, you will only be allowed to read from the
  # data store during the transaction and any attempts to change the data will
  # raise a PStore::Error.
  # 
  # Note that PStore does not support nested transactions.
  #
  def transaction(read_only=false)  # :yields:  pstore
    raise PStore::Error, "nested transaction" if @transaction
    begin
      @rdonly = read_only
      @abort = false
      @transaction = true
      value = nil
      new_file = @filename + ".new"

content = nil
      unless read_only
        file = File.open(@filename, RDWR_ACCESS)
        file.flock(File::LOCK_EX)
        commit_new(file) if FileTest.exist?(new_file)
        content = file.read()
      else
        begin
          file = File.open(@filename, RD_ACCESS)
          file.flock(File::LOCK_SH)
          content = (File.open(new_file, RD_ACCESS) {|n| n.read} rescue file.read())
        rescue Errno::ENOENT
          content = ""
        end
      end

if content != ""
	@table = load(content)
        if !read_only
          size = content.size
          md5 = Digest::MD5.digest(content)
        end
      else
	@table = {}
      end
      content = nil		# unreference huge data

begin
	catch(:pstore_abort_transaction) do
	  value = yield(self)
	end
      rescue Exception
	@abort = true
	raise
      ensure
	if !read_only and !@abort
          tmp_file = @filename + ".tmp"
	  content = dump(@table)
	  if !md5 || size != content.size || md5 != Digest::MD5.digest(content)
            File.open(tmp_file, WR_ACCESS) {|t| t.write(content)}
            File.rename(tmp_file, new_file)
            commit_new(file)
          end
          content = nil		# unreference huge data
	end
      end
    ensure
      @table = nil
      @transaction = false
      file.close if file
    end
    value
  end

# This method is just a wrapped around Marshal.dump.
  def dump(table)  # :nodoc:
    Marshal::dump(table)
  end

# This method is just a wrapped around Marshal.load.
  def load(content)  # :nodoc:
    Marshal::load(content)
  end

# This method is just a wrapped around Marshal.load.
  def load_file(file)  # :nodoc:
    Marshal::load(file)
  end

private
  # Commits changes to the data store file.
  def commit_new(f)
    f.truncate(0)
    f.rewind
    new_file = @filename + ".new"
    File.open(new_file, RD_ACCESS) do |nf|
      FileUtils.copy_stream(nf, f)
    end
    File.unlink(new_file)
  end
end

# :enddoc:

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

${this.title}

Code Editor : pstore.rb