activestorage-backblaze 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7d7c0eb8bb1cab9146cb415bf6b16357b7816fdbe32821c8fffda0281d2fbc5b
-  data.tar.gz: 62b0ce9a3a4bb6816f9ba49b69ef6192c6c2d90043ffc5c1ae1d5aff5b18f08c
+  metadata.gz: 91def17491b8ea4a04f7c03cac167a712e633d7297e679ec42d0149674cdcc68
+  data.tar.gz: 8b31f27d7574a7e70b863d5341aabb79084a7668c8a6af4ff6eb649f42065964
 SHA512:
-  metadata.gz: ec5fe0d075775943a5dc676fe01cbe98e56dbd4cba2deecc085309ea61aee60bca05790aacd35521cf5877d1c430f57bd5ebf32a00057c3f4f7cb55c2d80c4e1
-  data.tar.gz: 27d5a9803fdd4d3ab3422e06613913a9c0de3983343dc0995d3c3bea78618cf721344dcee8648d60e4b5b6d04720a5dc26dc8906b8b5af9749de56142753da47
+  metadata.gz: 1d7b8c31aa2b7da9ea211f32a65b61239b065e7984ca2f5e444049585b87cf8b3026725ba931679476c838cf79af880fcd56eba2ed2eee578be69692c04eeb68
+  data.tar.gz: 79dbff0c379557b22cf7f8ab76bb1140d43ce21e7348026fbe88a0dda39b455e3908e04d9744694d258312ada353fc66cc139267a0c48b96a302a6db8566f702

data/lib/active_storage/service/backblaze_service.rb

@@ -12,7 +12,7 @@ class ActiveStorage::Service::BackblazeService < ActiveStorage::Service
       b2_key_token: @key_token,
       b2_bucket_name: @bucket_name,
       b2_bucket_id: @bucket_id,
-      logger: Logger.new(STDOUT)
+      logger: Rails.logger
     )
   end
 
@@ -26,9 +26,17 @@ class ActiveStorage::Service::BackblazeService < ActiveStorage::Service
     end
   end
 
-  def download(key)
-    instrument :download, { key: key } do
-      io = @connection.get_object(@bucket_name, key)
+  def download(key, &block)
+    if block_given?
+      instrument :streaming_download, { key: key } do
+        stream(key, &block)
+      end
+    else
+      instrument :download, { key: key } do
+        resp = @connection.get_object(@bucket_name, key)
+        io = StringIO.new(resp.body)
+        io
+      end
     end
   end
 
@@ -76,4 +84,17 @@ class ActiveStorage::Service::BackblazeService < ActiveStorage::Service
     raise NotImpletementedError
   end
 
+  private
+    def stream(key, options = {}, &block)
+      resp = @connection.get_object(@bucket_name, key)
+      io = StringIO.new(resp.body)
+      io.binmode
+
+      chunk_size = 5.megabytes
+
+      while chunk = io.read(chunk_size)
+        yield chunk
+      end
+    end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: activestorage-backblaze
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Jey
@@ -33,7 +33,8 @@ files:
 - lib/active_storage/service/backblaze_service.rb
 - lib/active_storage_backblaze.rb
 homepage: https://github.com/gten/activestorage-backblaze
-licenses: []
+licenses:
+- MIT
 metadata:
   source_code_uri: https://github.com/gten/activestorage-backblaze
 post_install_message: