response_bank 1.3.6 → 1.3.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 59df27c0628e98bd29533d141d2209e03ab602ccc7a87f7cb4b74b6a6a671852
-  data.tar.gz: 955d472d29e0827203634443e9341f67f24cfcdf61b8f152bb0e8564732ede7f
+  metadata.gz: 0a5c7a799e1272fffad4878273bf154c09f809059944dbf47bf6c6c83d1b4acb
+  data.tar.gz: ff7a669443427dcee35c7d2a590258fc0ac0591bd156c1316a42198cacc4bd0e
 SHA512:
-  metadata.gz: 6cbc80643e1eaed0a3f7a039b3b30f02cb56a245ece87549dd29948c3e959c1aaefb4ef870a163ebab20bc6dd4a959a790881f4441703f1ee60cf11ca16f35cf
-  data.tar.gz: f949480d70681d20537ab03b766c7dfdf4cfbaa6f70aa2965a87e219e47bc6aab532d26e32675d3fb5f45f20f41822a8878519f6e020b905ff1a477fbd839fc7
+  metadata.gz: 5ddaeeca4364360ce6f2392d5b53d2f34421e44a62f1184d594add38dc45ebf23f72184bc9aa097667404c83d6b64c450b91b1dd25726a307ba41149b8a9509d
+  data.tar.gz: 4bfc7ebc29beab0c67f72e24e635d9fa2a0ca59a06fd92027d00172acc31a1974288adee1a97baf005e2ae64751a49e205a7005c70881f27e44b03a6827bf66a

data/lib/response_bank/middleware.rb

@@ -37,12 +37,21 @@ module ResponseBank
           body_compressed = nil
           if body_string && body_string != ""
             headers['Content-Encoding'] = content_encoding
-            body_compressed = ResponseBank.compress(body_string, content_encoding)
+            env["cacheable.compression_level"] = ResponseBank.compression_level_for_request(env, headers)
+            time = ResponseBank.measure do
+              body_compressed = ResponseBank.compress(
+                body_string,
+                content_encoding,
+                compression_level: env["cacheable.compression_level"],
+              )
+            end
+            ResponseBank.log("Compression time: #{time}ms")
+            env["cacheable.compression_time"] = time
           end
 
           cached_headers = headers.slice(*CACHEABLE_HEADERS)
           # Store result
-          cache_data = [status, cached_headers, body_compressed, timestamp]
+          cache_data = [status, cached_headers, body_compressed, timestamp, env["cacheable.compression_level"]]
 
           ResponseBank.write_to_cache(env['cacheable.key']) do
             payload = MessagePack.dump(cache_data)

data/lib/response_bank/response_cache_handler.rb

@@ -119,7 +119,9 @@ module ResponseBank
         @env['cacheable.miss']  = false
         @env['cacheable.store'] = 'server'
 
-        status, headers, body, timestamp = hit
+        status, headers, body, timestamp, compression_level = hit
+
+        @env['cacheable.compression_level'] = compression_level
 
         @env['cacheable.locked'] ||= false
 

data/lib/response_bank/version.rb

@@ -1,4 +1,4 @@
 # frozen_string_literal: true
 module ResponseBank
-  VERSION = "1.3.6"
+  VERSION = "1.3.7"
 end

data/lib/response_bank.rb

@@ -4,11 +4,31 @@ require 'response_bank/railtie' if defined?(Rails)
 require 'response_bank/response_cache_handler'
 require 'msgpack'
 require 'brotli'
+require 'benchmark'
 
 module ResponseBank
   class << self
     attr_accessor :cache_store
-    attr_writer :logger
+    attr_writer :logger, :compression_level
+
+    DEFAULT_BROTLI_COMPRESSION_LEVEL = 7
+
+    DEFAULT_COMPRESSION_LEVEL = -> (_env, headers) {
+      case headers['Content-Encoding']
+      when 'br'
+        DEFAULT_BROTLI_COMPRESSION_LEVEL
+      when 'gzip'
+        Zlib::BEST_COMPRESSION
+      end
+    }
+
+    def compression_level_for_request(env, headers)
+      if @compression_level
+        return @compression_level.respond_to?(:call) ? @compression_level.call(env, headers) : @compression_level
+      end
+
+      DEFAULT_COMPRESSION_LEVEL.call(env, headers)
+    end
 
     def log(message)
       @logger.info("[ResponseBank] #{message}")
@@ -30,13 +50,19 @@ module ResponseBank
       backing_cache_store.read(cache_key, raw: true)
     end
 
-    def compress(content, encoding = "br")
+    def measure
+      Benchmark.realtime do
+        yield
+      end * 1000 # milliseconds
+    end
+
+    def compress(content, encoding = "br", compression_level: nil)
       case encoding
       when 'gzip'
         attempts = 0
 
         begin
-          Zlib.gzip(content, level: Zlib::BEST_COMPRESSION)
+          Zlib.gzip(content, level: compression_level || Zlib::BEST_COMPRESSION)
         rescue Zlib::BufError
           # We get sporadic Zlib::BufError, so we retry once (https://github.com/ruby/zlib/issues/49)
           attempts += 1
@@ -48,7 +74,7 @@ module ResponseBank
           end
         end
       when 'br'
-        Brotli.deflate(content, mode: :text, quality: 7)
+        Brotli.deflate(content, mode: :text, quality: compression_level || DEFAULT_BROTLI_COMPRESSION_LEVEL)
       else
         raise ArgumentError, "Unsupported encoding: #{encoding}"
       end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: response_bank
 version: !ruby/object:Gem::Version
-  version: 1.3.6
+  version: 1.3.7
 platform: ruby
 authors:
 - Tobias Lütke
@@ -143,7 +143,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.9
+rubygems_version: 3.7.2
 specification_version: 4
 summary: Simple response caching for Ruby applications
 test_files: []