sidekiq-compress 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b4d9769835661bd5651a58174d78d714f712b3b9399bd691bac5d43fc4f46580
-  data.tar.gz: 150c2f965dd735b7fbac79343ac69497a00b12c098d7e8bf51e116971f90a81d
+  metadata.gz: 180a44d5decc2d22318cf71c60c5ac901dcb28fec9c367b57717541b020e55d8
+  data.tar.gz: 6f6051e96f54be6212dd8cf9bb71f1518d10e02eb85fc927b50dcfb0a41bb589
 SHA512:
-  metadata.gz: a3c1592109bfb48d1f32af96d5628e3090fe403cfde4431f0861e2bb752c8360734c7ef94d857561d8463355c389e8d78dae832766204623819dcbf742bec03f
-  data.tar.gz: 2b59e2ef7a29c63ac5dd13af7e4fe815b4f84fded50f2564ef67c14a869888385149e800019ad9e4e66e43241d10bf8cf7ab6f5e7357928c8be9ac0f9ed544f8
+  metadata.gz: d939824efa350bcae5cd959106726d23969949770f55f3e7f6807acd2fc45f3e96a1e46ada19ed5b3998a60375ea13d4b3210ac78657e2f408f2d86ebd15b3c0
+  data.tar.gz: 6ba209c6cd11a6962a5f904d3bd000695b434f927218cadc345df800f2137f17eb541490a8f3dd62685dbaa7532fb10d3560665a5202615890fce7cd569bd2f5

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    sidekiq-compress (0.0.2)
+    sidekiq-compress (0.0.3)
 
 GEM
   remote: https://rubygems.org/

data/lib/sidekiq-compress.rb

@@ -1,6 +1,7 @@
 require 'sidekiq-compress/version'
 require 'sidekiq-compress/client_middleware'
 require 'sidekiq-compress/server_middleware'
+require 'sidekiq-compress/worker'
 
 module Sidekiq::Compress
 end

data/lib/sidekiq-compress/client_middleware.rb

@@ -3,31 +3,20 @@ require 'sidekiq-compress/compress_params'
 module Sidekiq::Compress
 # Should be in the client middleware chain
   class ClientMiddleware
-    # Uses msg['jid'] id and puts :queued status in the job's Redis hash
-    # @param [Class] worker_class if includes Sidekiq::Status::Worker, the job gets processed with the plugin
-    # @param [Array] msg job arguments
-    # @param [String] queue the queue's name
-    # @param [ConnectionPool] redis_pool optional redis connection pool
     def call(worker, msg, queue, redis_pool)
-      # return false/nil to stop the job from going to redis
       klass = msg["args"][0]["job_class"] || msg["class"] rescue msg["class"]
       job_class = klass.is_a?(Class) ? klass : Module.const_get(klass)
-      unless job_class.ancestors.include?(Sidekiq::Compress)
+      unless job_class.ancestors.include?(Sidekiq::Compress::Worker)
         yield
         return
       end
 
-      msg["args"] = Sidekiq::Compress::CompressParams.call(msg["args"])
-
+      compress_indexes = job_class.sidekiq_compress_params[0][:index] rescue []
+      msg["args"] = Sidekiq::Compres::CompressParams.call(msg["args"], compress_indexes)
       yield
     end
   end
 
-  # Helper method to easily configure sidekiq-status client middleware
-  # whatever the Sidekiq version is.
-  # @param [Sidekiq] sidekiq_config the Sidekiq config
-  # @param [Hash] client_middleware_options client middleware initialization options
-  # @option client_middleware_options [Fixnum] :expiration ttl for complete jobs
   def self.configure_client_middleware(sidekiq_config)
     sidekiq_config.client_middleware do |chain|
       chain.add Sidekiq::Compress::ClientMiddleware

data/lib/sidekiq-compress/compress_params.rb

@@ -1,14 +1,23 @@
 module Sidekiq::Compress
   class CompressParams
-    def self.call(params = [])
-      result = []
-      params.each do |param|
-        if param.class == String
-          param = Base64.encode64(Zstd.compress(param))
-        end
-        result << param
+    def self.call(params = [], compress_indexes = [])
+      compress_indexes.empty? ? compress_all(params) : compress_by_index(params, compress_indexes)
+    end
+
+    private
+
+    def self.compress_all(params)
+      params.map! { |param| param.class == String ? Base64.encode64(Zstd.compress(param)) : param }
+    end
+
+    def self.compress_by_index(params, compress_indexes)
+      compress_indexes.each do |index|
+        raise "Parameter at index #{index} must be a string!" unless params[index].class == String
+
+        params[index] = Base64.encode64(Zstd.compress(params[index]))
       end
-      result
+
+      params
     end
   end
 end

data/lib/sidekiq-compress/decompress_params.rb

@@ -1,14 +1,34 @@
 module Sidekiq::Compress
   class DecompressParams
-    def self.call(params = [])
+    def self.call(params = [], decompress_indexes = [])
       result = []
-      params.each do |param|
+      if decompress_indexes.empty?
+        result = decompress_all(params)
+      else  
+        result = decompress_by_index(params, decompress_indexes)
+      end
+
+      result
+    end
+
+    private
+
+    def self.decompress_all(params)
+      params.map! do |param|
         if param.class == String
-          param = Zstd.decompress(Base64.decode64(param)).force_encoding('UTF-8')
+          Zstd.decompress(Base64.decode64(param)).force_encoding('UTF-8')
+        else
+          param
         end
-        result << param
       end
-      result
+    end
+
+    def self.decompress_by_index(params, decompress_indexes)
+      decompress_indexes.each do |index|
+        params[index] = Zstd.decompress(Base64.decode64(params[index])).force_encoding('UTF-8')
+      end
+
+      params
     end
   end
 end

data/lib/sidekiq-compress/server_middleware.rb

@@ -6,22 +6,19 @@ module Sidekiq::Compress
     def call(worker, msg, queue)
       klass = msg["args"][0]["job_class"] || msg["class"] rescue msg["class"]
       job_class = klass.is_a?(Class) ? klass : Module.const_get(klass)
-      unless job_class.ancestors.include?(Sidekiq::Compress)
+      unless job_class.ancestors.include?(Sidekiq::Compress::Worker)
         yield
         return
       end
 
-      msg["args"] = Sidekiq::Compress::DecompressParams.call(msg["args"])
+      decompress_indexes = job_class.sidekiq_compress_params[0][:index]
+
+      msg["args"] = Sidekiq::Compress::DecompressParams.call(msg["args"], decompress_indexes)
 
       yield
     end
   end
 
-  # Helper method to easily configure sidekiq-status server middleware
-  # whatever the Sidekiq version is.
-  # @param [Sidekiq] sidekiq_config the Sidekiq config
-  # @param [Hash] server_middleware_options server middleware initialization options
-  # @option server_middleware_options [Fixnum] :expiration ttl for complete jobs
   def self.configure_server_middleware(sidekiq_config)
     sidekiq_config.server_middleware do |chain|
       chain.add Sidekiq::Compress::ServerMiddleware

data/lib/sidekiq-compress/version.rb

@@ -1,5 +1,5 @@
 module Sidekiq
   module Compress
-    VERSION = "0.0.2"
+    VERSION = "0.0.3"
   end
 end

data/lib/sidekiq-compress/worker.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Sidekiq::Compress
+  module Worker
+    def self.included(base)
+      base.extend(ClassMethods)
+    end
+
+    module ClassMethods
+      attr_reader :sidekiq_compress_params
+
+      def compress_params(*params)
+        @sidekiq_compress_params = params
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sidekiq-compress
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Oğuzhan Göller
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-04-14 00:00:00.000000000 Z
+date: 2020-04-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -75,6 +75,7 @@ files:
 - lib/sidekiq-compress/decompress_params.rb
 - lib/sidekiq-compress/server_middleware.rb
 - lib/sidekiq-compress/version.rb
+- lib/sidekiq-compress/worker.rb
 - sidekiq-compress.gemspec
 homepage: https://github.com/oguzhangoller/sidekiq-compress
 licenses: