canvas_sync 0.17.23.beta7 → 0.17.23.beta8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4ffd1eae28ffc49cdba7673f1648fbf82853a57768697df38d2a100c05c7ab3f
-  data.tar.gz: 4ab9d6f8c7ce264b69fae4da982d16dfedf40e5b1d1f64fbfd40831f16465bdf
+  metadata.gz: 9bed2b1c5be095c602aca19968b0155ff48026a1c5136cb65b5f31e5decbcb40
+  data.tar.gz: ae6112e2694f0118a43619cf3de053d7eee7ff1873e0dface27b0b3558727132
 SHA512:
-  metadata.gz: 9619fbc749887158498c00761798cd0b10018ce5962262f27c9ff69c0615c66ffe1ebfa004bdc2ca7374308e53598ee81175f2372e2cde69527ce4672c288626
-  data.tar.gz: a9459bd07489e9c68abd692977e6ca1bc01b06d7dd960560f85e7f38bff4240ac8334aff15b91fb7f7a84897c7fcbd828e623d3b098fa34126b1b4e9be255222
+  metadata.gz: 6261b328341a102a9ea6c5f91315444604a3556ccbc040fb38122d18db557bd2c5d2cbc7e5b3c74a52d0d48eb0315793232a3cdff0be21964af1c2617d4ae22d
+  data.tar.gz: 458a2c7b422c2a769ec961bd75bd8f43e5dde63c9231bbd4aab6e2a400ec79ee74d9dc12d93cdbd2cc26ef69bb3611972b84956f55cfc2e2a89a4fc22ec2eafb

data/lib/canvas_sync/job_batches/batch.rb

@@ -76,14 +76,12 @@ module CanvasSync
       def on(event, callback, options = {})
         return unless Callback::VALID_CALLBACKS.include?(event.to_s)
         callback_key = "#{@bidkey}-callbacks-#{event}"
-        redis do |r|
-          r.multi do
-            r.sadd(callback_key, JSON.unparse({
-              callback: callback,
-              opts: options
-            }))
-            r.expire(callback_key, BID_EXPIRE_TTL)
-          end
+        redis.multi do |r|
+          r.sadd(callback_key, JSON.unparse({
+            callback: callback,
+            opts: options
+          }))
+          r.expire(callback_key, BID_EXPIRE_TTL)
         end
       end
 
@@ -93,16 +91,14 @@ module CanvasSync
         if !@existing && !@initialized
           parent_bid = Thread.current[:batch]&.bid
 
-          redis do |r|
-            r.multi do
-              r.hset(@bidkey, "parent_bid", parent_bid.to_s) if parent_bid
-              r.expire(@bidkey, BID_EXPIRE_TTL)
+          redis.multi do |r|
+            r.hset(@bidkey, "parent_bid", parent_bid.to_s) if parent_bid
+            r.expire(@bidkey, BID_EXPIRE_TTL)
 
-              if parent_bid
-                r.hincrby("BID-#{parent_bid}", "children", 1)
-                r.expire("BID-#{parent_bid}", BID_EXPIRE_TTL)
-                r.zadd("BID-#{parent_bid}-bids", created_at, bid)
-              end
+            if parent_bid
+              r.hincrby("BID-#{parent_bid}", "children", 1)
+              r.expire("BID-#{parent_bid}", BID_EXPIRE_TTL)
+              r.zadd("BID-#{parent_bid}-bids", created_at, bid)
             end
           end
 
@@ -131,15 +127,11 @@ module CanvasSync
       end
 
       def invalidate_all
-        redis do |r|
-          r.setex("invalidated-bid-#{bid}", BID_EXPIRE_TTL, 1)
-        end
+        redis.setex("invalidated-bid-#{bid}", BID_EXPIRE_TTL, 1)
       end
 
       def parent_bid
-        redis do |r|
-          r.hget(@bidkey, "parent_bid")
-        end
+        redis.hget(@bidkey, "parent_bid")
       end
 
       def parent
@@ -149,7 +141,7 @@ module CanvasSync
       end
 
       def valid?(batch = self)
-        valid = !redis { |r| r.exists?("invalidated-bid-#{batch.bid}") }
+        valid = !redis.exists?("invalidated-bid-#{batch.bid}")
         batch.parent ? valid && valid?(batch.parent) : valid
       end
 
@@ -162,23 +154,19 @@ module CanvasSync
             let_close!
           end
         else
-          redis do |r|
-            r.hset(@bidkey, 'keep_open', true)
-          end
+          redis.hset(@bidkey, 'keep_open', true)
         end
       end
 
       def let_close!
-        _, failed, pending, children, complete, success = redis do |r|
-          r.multi do
-            r.hset(@bidkey, 'keep_open', false)
-
-            r.scard("BID-#{bid}-failed")
-            r.hincrby("BID-#{bid}", "pending", 0)
-            r.hincrby("BID-#{bid}", "children", 0)
-            r.scard("BID-#{bid}-batches-complete")
-            r.scard("BID-#{bid}-batches-success")
-          end
+        _, failed, pending, children, complete, success = redis.multi do |r|
+          r.hset(@bidkey, 'keep_open', false)
+
+          r.scard("BID-#{bid}-failed")
+          r.hincrby("BID-#{bid}", "pending", 0)
+          r.hincrby("BID-#{bid}", "children", 0)
+          r.scard("BID-#{bid}-batches-complete")
+          r.scard("BID-#{bid}-batches-success")
         end
 
         all_success = pending.to_i.zero? && children == success
@@ -211,18 +199,14 @@ module CanvasSync
 
       def flush_pending_attrs
         super
-        redis do |r|
-          r.zadd("batches", created_at, bid)
-        end
+        redis.zadd("batches", created_at, bid)
       end
 
       private
 
       def assert_batch_is_open
         unless defined?(@closed)
-          redis do |r|
-            @closed = r.hget(@bidkey, 'success') == 'true'
-          end
+          @closed = redis.hget(@bidkey, 'success') == 'true'
         end
         raise "Cannot add jobs to Batch #{} bid - it has already entered the callback-stage" if @closed
       end
@@ -234,7 +218,7 @@ module CanvasSync
         redis do |r|
           tme = Time.now.utc.to_f
           added = r.zadd(@bidkey + "-jids", jids.map{|jid| [tme, jid] }, nx: true)
-          r.multi do
+          r.multi do |r|
             r.hincrby(@bidkey, "pending", added)
             r.hincrby(@bidkey, "job_count", added)
             r.expire(@bidkey, BID_EXPIRE_TTL)
@@ -283,11 +267,9 @@ module CanvasSync
           end
 
           if parent_bid
-            redis do |r|
-              r.multi do
-                r.sadd("BID-#{parent_bid}-dead", jid)
-                r.expire("BID-#{parent_bid}-dead", BID_EXPIRE_TTL)
-              end
+            redis.multi do
+              r.sadd("BID-#{parent_bid}-dead", jid)
+              r.expire("BID-#{parent_bid}-dead", BID_EXPIRE_TTL)
             end
           end
 
@@ -413,8 +395,21 @@ module CanvasSync
           cleanup_redis(bid)
         end
 
-        def redis(*args, &blk)
-          defined?(::Sidekiq) ? ::Sidekiq.redis(*args, &blk) : nil # TODO
+        def redis(&blk)
+          return RedisProxy.new unless block_given?
+
+          if Thread.current[:job_batches_redis]
+            yield Thread.current[:job_batches_redis]
+          elsif defined?(::Sidekiq)
+            ::Sidekiq.redis do |r|
+              Thread.current[:job_batches_redis] = r
+              yield r
+            ensure
+              Thread.current[:job_batches_redis] = nil
+            end
+          else
+            # TODO
+          end
         end
 
         def logger
@@ -433,6 +428,42 @@ module CanvasSync
           end
         end
       end
+
+      class RedisProxy
+        def multi(*args, &block)
+          Batch.redis do |r|
+            if block.arity == 1
+              r.multi(*args) do
+                block.call(r)
+              end
+            else
+              r.multi(*args, &block)
+            end
+          end
+        end
+
+        def pipelined(*args, &block)
+          Batch.redis do |r|
+            if block.arity == 1
+              r.pipelined(*args) do
+                block.call(r)
+              end
+            else
+              r.pipelined(*args, &block)
+            end
+          end
+        end
+
+        def method_missing(method_name, *arguments, &block)
+          Batch.redis do |r|
+            r.send(method_name, *arguments, &block)
+          end
+        end
+
+        def respond_to_missing?(method_name, include_private = false)
+          super || Redis.method_defined?(method_name)
+        end
+      end
     end
 
     ActiveJob::Base.include BatchAwareJob

data/lib/canvas_sync/job_batches/pool.rb

@@ -64,18 +64,14 @@ module CanvasSync
             let_close!
           end
         else
-          redis do |r|
-            r.hset(redis_key, 'keep_open', true)
-          end
+          redis.hset(redis_key, 'keep_open', true)
         end
       end
 
       def let_close!
-        _, active_count = redis do |r|
-          r.multi do
-            r.hset(redis_key, 'keep_open', false)
-            r.hincrby(redis_key, "active_count", 0)
-          end
+        _, active_count = redis.multi do |r|
+          r.hset(redis_key, 'keep_open', false)
+          r.hincrby(redis_key, "active_count", 0)
         end
 
         if active_count == 0 && pending_count == 0
@@ -95,35 +91,38 @@ module CanvasSync
       end
 
       def active_count
-        redis do |r|
-          r.hincrby(redis_key, "active_count", 0)
-        end
+        redis.hincrby(redis_key, "active_count", 0)
       end
 
       def pending_count
         jobs_key = "#{redis_key}-jobs"
         order = self.order || 'fifo'
-        redis do |r|
-          case order.to_sym
-          when :fifo, :lifo
-            r.llen(jobs_key)
-          when :random
-            r.scard(jobs_key)
-          when :priority
-            r.zcard(jobs_key)
-          end
+        case order.to_sym
+        when :fifo, :lifo
+          redis.llen(jobs_key)
+        when :random
+          redis.scard(jobs_key)
+        when :priority
+          redis.zcard(jobs_key)
         end
       end
 
       def job_checked_in(status, options)
-        active_count = redis do |r|
+        active_count, pending_count = redis do |r|
           return unless r.exists?(redis_key)
-          r.hincrby(redis_key, "active_count", -1)
+
+          # Make sure this is loaded outside of the pipeline
+          self.order
+
+          redis.multi do
+            r.hincrby(redis_key, "active_count", -1)
+            self.pending_count
+          end
         end
 
         added_count = refill_allotment
-        if active_count == 0 && added_count == 0
-          if clean_when_empty && redis {|r| r.hget(redis_key, 'keep_open') } != 'true'
+        if active_count == 0 && added_count == 0 && pending_count == 0
+          if clean_when_empty && redis.hget(redis_key, 'keep_open') != 'true'
             cleanup_redis
           end
         end
@@ -175,18 +174,16 @@ module CanvasSync
         job_json = JSON.unparse(ActiveJob::Arguments.serialize([job_desc]))
         order = self.order
 
-        redis do |r|
-          r.multi do
-            case order.to_sym
-            when :fifo, :lifo
-              r.rpush(jobs_key, job_json)
-            when :random
-              r.sadd(jobs_key, job_json)
-            when :priority
-              r.zadd(jobs_key, job_desc[:priority] || 0, job_json)
-            end
-            r.expire(jobs_key, Batch::BID_EXPIRE_TTL)
+        redis.multi do |r|
+          case order.to_sym
+          when :fifo, :lifo
+            r.rpush(jobs_key, job_json)
+          when :random
+            r.sadd(jobs_key, job_json)
+          when :priority
+            r.zadd(jobs_key, job_desc[:priority] || 0, job_json)
           end
+          r.expire(jobs_key, Batch::BID_EXPIRE_TTL)
         end
       end
 
@@ -194,18 +191,15 @@ module CanvasSync
         jobs_key = "#{redis_key}-jobs"
         order = self.order
 
-        job_json = nil
-        redis do |r|
-          job_json = case order.to_sym
-          when :fifo
-            r.lpop(jobs_key)
-          when :lifo
-            r.rpop(jobs_key)
-          when :random
-            r.spop(jobs_key)
-          when :priority
-            r.zpopmax(jobs_key)
-          end
+        job_json = case order.to_sym
+        when :fifo
+          redis.lpop(jobs_key)
+        when :lifo
+          redis.rpop(jobs_key)
+        when :random
+          redis.spop(jobs_key)
+        when :priority
+          redis.zpopmax(jobs_key)
         end
 
         return nil unless job_json.present?
@@ -218,13 +212,9 @@ module CanvasSync
       end
       delegate :redis, to: :class
 
-      protected
-
       def flush_pending_attrs
         super
-        redis do |r|
-          r.zadd("pools", created_at, pid)
-        end
+        redis.zadd("pools", created_at, pid)
       end
 
       private

data/lib/canvas_sync/version.rb

@@ -1,3 +1,3 @@
 module CanvasSync
-  VERSION = "0.17.23.beta7".freeze
+  VERSION = "0.17.23.beta8".freeze
 end