From 80400362a47f079afbe6fb613dc62a81cbb8ec4d Mon Sep 17 00:00:00 2001
From: matz <matz@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Thu, 10 Apr 2008 14:10:19 +0000
Subject: * lib/pstore.rb (PStore::dump, PStore::load): allow subclass  
 overriding.  [ruby-dev:34305]

* lib/yaml/store.rb (YAML::Store::marshal_dump_supports_canonical_option?):
  add a method to support faster PStore.

git-svn-id: http://svn.ruby-lang.org/repos/ruby/trunk@15964 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 lib/pstore.rb     | 19 ++++++++++++++++---
 lib/yaml/store.rb |  4 ++--
 2 files changed, 18 insertions(+), 5 deletions(-)

(limited to 'lib')

diff --git a/lib/pstore.rb b/lib/pstore.rb
index 2f94a045f..54d4e7c3a 100644
--- a/lib/pstore.rb
+++ b/lib/pstore.rb
@@ -399,7 +399,7 @@ class PStore
   def load_data(file, read_only)
     if read_only
       begin
-        table = Marshal.load(file)
+        table = load(file)
         if !table.is_a?(Hash)
           raise Error, "PStore file seems to be corrupted."
         end
@@ -416,7 +416,7 @@ class PStore
         checksum = EMPTY_MARSHAL_CHECKSUM
         size = EMPTY_MARSHAL_DATA.size
       else
-        table = Marshal.load(data)
+        table = load(data)
         checksum = Digest::MD5.digest(data)
         size = data.size
         if !table.is_a?(Hash)
@@ -461,7 +461,7 @@ class PStore
     if marshal_dump_supports_canonical_option?
       new_data = Marshal.dump(@table, -1, true)
     else
-      new_data = Marshal.dump(@table)
+      new_data = dump(@table)
     end
     new_checksum = Digest::MD5.digest(new_data)
     
@@ -498,6 +498,19 @@ class PStore
     file.truncate(0)
     file.write(data)
   end
+
+
+  # This method is just a wrapped around Marshal.dump
+  # to allow subclass overriding used in YAML::Store.
+  def dump(table)  # :nodoc:
+    Marshal::dump(table)
+  end
+
+  # This method is just a wrapped around Marshal.load.
+  # to allow subclass overriding used in YAML::Store.
+  def load(content)  # :nodoc:
+    Marshal::load(content)
+  end
 end
 
 # :enddoc:
diff --git a/lib/yaml/store.rb b/lib/yaml/store.rb
index 2ffa9554b..4f0384cc5 100644
--- a/lib/yaml/store.rb
+++ b/lib/yaml/store.rb
@@ -23,7 +23,7 @@ class YAML::Store < PStore
     YAML::load(content)
   end
 
-  def load_file(file)
-    YAML::load(file)
+  def marshal_dump_supports_canonical_option?
+    false
   end
 end
-- 
cgit