new_file: backwards compatibility for old args

1 files changed, 2 insertions, 1 deletions

diff --git a/lib/mogilefs/mogilefs.rb b/lib/mogilefs/mogilefs.rb
index 6059f5f..22dd39c 100644
--- a/lib/mogilefs/mogilefs.rb
+++ b/lib/mogilefs/mogilefs.rb
@@ -129,7 +129,7 @@ class MogileFS::MogileFS < MogileFS::Client
   # Consider using store_file instead of this method for large files.
   # This requires a block passed to it and operates like File.open.
   # This atomically replaces existing data stored as +key+ when
-  def new_file(key, args = nil, bytes = 0) # :yields: file
+  def new_file(key, args = nil, bytes = nil) # :yields: file
     raise MogileFS::ReadOnlyError if readonly?
     opts = { :key => key, :multi_dest => 1 }
     case args
@@ -157,6 +157,7 @@ class MogileFS::MogileFS < MogileFS::Client
     opts.merge!(args) if Hash === args
     opts[:backend] = @backend
     opts[:fid] = res['fid']
+    opts[:content_length] ||= bytes if bytes
 
     case (dests[0][1] rescue nil)
     when %r{\Ahttp://}