1
0
Fork 0
mirror of https://github.com/rails/rails.git synced 2022-11-09 12:12:34 -05:00

Differentiate between io streams and read data

This commit is contained in:
David Heinemeier Hansson 2017-07-04 16:05:06 +02:00
parent 8da081c36f
commit ccaba581c0
7 changed files with 16 additions and 16 deletions

View file

@ -11,16 +11,16 @@ class ActiveFile::Blob < ActiveRecord::Base
class_attribute :site
class << self
def build_after_upload(data:, filename:, content_type: nil, metadata: nil)
def build_after_upload(io:, filename:, content_type: nil, metadata: nil)
new.tap do |blob|
blob.filename = name
blob.content_type = content_type # Marcel::MimeType.for(data, name: name, declared_type: content_type)
blob.upload data
blob.upload io
end
end
def create_after_upload!(data:, filename:, content_type: nil, metadata: nil)
build_after_upload(data: data, filename: filename, content_type: content_type, metadata: metadata).tap(&:save!)
def create_after_upload!(io:, filename:, content_type: nil, metadata: nil)
build_after_upload(io: io, filename: filename, content_type: content_type, metadata: metadata).tap(&:save!)
end
end
@ -38,8 +38,8 @@ class ActiveFile::Blob < ActiveRecord::Base
end
def upload(data)
site.upload(key, data)
def upload(io)
site.upload(key, io)
self.checksum = site.checksum(key)
self.byte_size = site.byte_size(key)

View file

@ -3,7 +3,7 @@ class ActiveFile::Site
def initialize
end
def upload(key, data)
def upload(key, io)
raise NotImplementedError
end

View file

@ -9,9 +9,9 @@ class ActiveFile::Sites::DiskSite < ActiveFile::Site
end
def upload(key, data)
def upload(key, io)
File.open(make_path_for(key), "wb") do |file|
while chunk = data.read(65536)
while chunk = io.read(65536)
file.write(chunk)
end
end

View file

@ -8,8 +8,8 @@ class ActiveFile::Sites::GCSSite < ActiveFile::Site
@bucket = @client.bucket(bucket)
end
def upload(key, data)
bucket.create_file(data, key)
def upload(key, io)
bucket.create_file(io, key)
end
def download(key)

View file

@ -5,8 +5,8 @@ class ActiveFile::Sites::MirrorSite < ActiveFile::Site
@sites = sites
end
def upload(key, data)
perform_across_sites :upload, key, data
def upload(key, io)
perform_across_sites :upload, key, io
end
def download(key)

View file

@ -8,8 +8,8 @@ class ActiveFile::Sites::S3Site < ActiveFile::Site
@bucket = @client.bucket(bucket)
end
def upload(key, data)
object_for(key).put(body: data)
def upload(key, io)
object_for(key).put(body: io)
end
def download(key)

View file

@ -22,6 +22,6 @@ class ActiveFile::BlobTest < ActiveSupport::TestCase
private
def create_blob(data: "Hello world!", filename: "hello.txt", content_type: "text/plain")
ActiveFile::Blob.create_after_upload! data: StringIO.new(data), filename: filename, content_type: content_type
ActiveFile::Blob.create_after_upload! io: StringIO.new(data), filename: filename, content_type: content_type
end
end