store_file/store_content: wire these up to new new_file opts

These allow us to specify Content-MD5 and checksums for use
with the new and improved new_file interface without breaking
existing apps.

1 files changed, 6 insertions, 4 deletions

diff --git a/lib/mogilefs/mogilefs.rb b/lib/mogilefs/mogilefs.rb
index 22dd39c..372eae0 100644
--- a/lib/mogilefs/mogilefs.rb
+++ b/lib/mogilefs/mogilefs.rb
@@ -181,18 +181,20 @@ class MogileFS::MogileFS < MogileFS::Client
   # either a path name (String or Pathname object) or an IO-like object that
   # responds to #read or #readpartial.  Returns size of +file+ stored.
   # This atomically replaces existing data stored as +key+
-  def store_file(key, klass, file)
+  def store_file(key, klass, file, opts = nil)
     raise MogileFS::ReadOnlyError if readonly?
+    (opts ||= {})[:class] = klass if String === klass
 
-    new_file(key, klass) { |mfp| mfp.big_io = file }
+    new_file(key, opts) { |mfp| mfp.big_io = file }
   end
 
   # Stores +content+ into +key+ in class +klass+, where +content+ is a String
   # This atomically replaces existing data stored as +key+
-  def store_content(key, klass, content)
+  def store_content(key, klass, content, opts = nil)
     raise MogileFS::ReadOnlyError if readonly?
+    (opts ||= {})[:class] = klass if String === klass
 
-    new_file key, klass do |mfp|
+    new_file(key, opts) do |mfp|
       if content.is_a?(MogileFS::Util::StoreContent)
         mfp.streaming_io = content
       else