DGNum/demarches-normaliennes
14
1
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity

Implement streaming GET for Cellar

Browse source
This commit is contained in:
Frederic Merizen 2018-02-14 16:38:22 +01:00
parent 0e74c55d48
commit e2c583480d
1 changed files with 21 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
lib/active_storage/service/cellar_service.rb
View file

@ -12,13 +12,27 @@ module ActiveStorage
end
def download(key)
instrument :download, key: key do
http_start do |http|
request = Net::HTTP::Get.new(URI::join(@endpoint, "/#{key}"))
sign(request, key)
response = http.request(request)
if response.is_a?(Net::HTTPSuccess)
response.body
if block_given?
instrument :streaming_download, key: key do
http_start do |http|
request = Net::HTTP::Get.new(URI::join(@endpoint, "/#{key}"))
sign(request, key)
http.request(request) do |response|
response.read_body do |chunk|
yield(chunk.force_encoding(Encoding::BINARY))
end
end
end
end
else
instrument :download, key: key do
http_start do |http|
request = Net::HTTP::Get.new(URI::join(@endpoint, "/#{key}"))
sign(request, key)
response = http.request(request)
if response.is_a?(Net::HTTPSuccess)
response.body.force_encoding(Encoding::BINARY)
end
end
end
end