From 3f0387830924822fde321e3613ced1607ffc27c6 Mon Sep 17 00:00:00 2001
From: matz <matz@b2dd03c8-39d4-4d8f-98ff-823fe69b080e>
Date: Fri, 11 Apr 2008 07:43:31 +0000
Subject: * lib/yaml/store.rb (YAML::load): modified to support empty  
 database.

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@15970 b2dd03c8-39d4-4d8f-98ff-823fe69b080e
---
 lib/pstore.rb     | 11 +++++++++--
 lib/yaml/store.rb | 16 +++++++++++++++-
 2 files changed, 24 insertions(+), 3 deletions(-)

(limited to 'lib')

diff --git a/lib/pstore.rb b/lib/pstore.rb
index 54d4e7c3a6..a1ad768fbf 100644
--- a/lib/pstore.rb
+++ b/lib/pstore.rb
@@ -413,8 +413,8 @@ class PStore
       if data.empty?
         # This seems to be a newly-created file.
         table = {}
-        checksum = EMPTY_MARSHAL_CHECKSUM
-        size = EMPTY_MARSHAL_DATA.size
+        checksum = empty_marshal_checksum
+        size = empty_marshal_data.size
       else
         table = load(data)
         checksum = Digest::MD5.digest(data)
@@ -511,6 +511,13 @@ class PStore
   def load(content)  # :nodoc:
     Marshal::load(content)
   end
+
+  def empty_marshal_data
+    EMPTY_MARSHAL_DATA
+  end
+  def empty_marshal_checksum
+    EMPTY_MARSHAL_CHECKSUM
+  end
 end
 
 # :enddoc:
diff --git a/lib/yaml/store.rb b/lib/yaml/store.rb
index 4f0384cc5e..e3a8e9fcdd 100644
--- a/lib/yaml/store.rb
+++ b/lib/yaml/store.rb
@@ -20,10 +20,24 @@ class YAML::Store < PStore
   end
 
   def load(content)
-    YAML::load(content)
+    table = YAML::load(content)
+    if table == false
+      {}
+    else
+      table
+    end
   end
 
   def marshal_dump_supports_canonical_option?
     false
   end
+
+  EMPTY_MARSHAL_DATA = {}.to_yaml
+  EMPTY_MARSHAL_CHECKSUM = Digest::MD5.digest(EMPTY_MARSHAL_DATA)
+  def empty_marshal_data
+    EMPTY_MARSHAL_DATA
+  end
+  def empty_marshal_checksum
+    EMPTY_MARSHAL_CHECKSUM
+  end
 end
-- 
cgit v1.2.3