canvas_sync 0.17.23.beta6 → 0.17.24

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 298b05c56dda9ca6d603c1242c25fce864b4d69d096f2559a22267546f2183cb
-  data.tar.gz: d130e60ee19646e3d6cb817d63875bfdbd9358772c1dca25e49286c18cd0ca0d
+  metadata.gz: ff1d01575608d114952bb12803cae01cc2f4ff61c8ffeed78f18b247973926fa
+  data.tar.gz: ba3629041f491bb7bac12e03da3c896563c9595c77b58cec0d0c728f97168fcf
 SHA512:
-  metadata.gz: d1b091aa6718f213dffc0a80985db5d736dcc4269365a8c7cdba713fc17ed256314b66183d050b952284374fc10f18b31018fcfee6fdb4fa9d185824fec73033
-  data.tar.gz: b822704460aef2ea10664ff02de96de7f9baa240e3c1fbd19092a86160cad364ec8f42305882502b3aede3ad4972bd3f168716cba1c65bdd6e3b46014776f7ff
+  metadata.gz: 73ef3c21b2bb55276da6d5bd96a13b089117e47a04a117d4533822dc790eb2a62b2f8ac88fbfe122a4292e801542c49facef12554fd9a7c80426f699125d42b2
+  data.tar.gz: 29be22e24cb562c9decae8986b5e8a2b7f09da39a80e019edfe6f3cc994a90a2d53da1c9bae4e3d56640d630d7fbd8abca43e8bf946d35c211dae0a00833ef91

data/lib/canvas_sync/job_batches/batch.rb

@@ -19,6 +19,8 @@ require_relative "./chain_builder"
 
 module CanvasSync
   module JobBatches
+    CURRENT_BATCH_THREAD_KEY = :job_batches_batch
+
     class Batch
       include RedisModel
 
@@ -76,14 +78,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
 
@@ -91,18 +91,16 @@ module CanvasSync
         raise NoBlockGivenError unless block_given?
 
         if !@existing && !@initialized
-          parent_bid = Thread.current[:batch]&.bid
+          parent_bid = Thread.current[CURRENT_BATCH_THREAD_KEY]&.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
 
@@ -115,11 +113,11 @@ module CanvasSync
         end
 
         begin
-          parent = Thread.current[:batch]
-          Thread.current[:batch] = self
+          parent = Thread.current[CURRENT_BATCH_THREAD_KEY]
+          Thread.current[CURRENT_BATCH_THREAD_KEY] = self
           yield
         ensure
-          Thread.current[:batch] = parent
+          Thread.current[CURRENT_BATCH_THREAD_KEY] = parent
         end
 
         nil
@@ -131,15 +129,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 +143,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 +156,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
@@ -191,11 +181,11 @@ module CanvasSync
 
       def self.with_batch(batch)
         batch = self.new(batch) if batch.is_a?(String)
-        parent = Thread.current[:batch]
-        Thread.current[:batch] = batch
+        parent = Thread.current[CURRENT_BATCH_THREAD_KEY]
+        Thread.current[CURRENT_BATCH_THREAD_KEY] = batch
         yield
       ensure
-        Thread.current[:batch] = parent
+        Thread.current[CURRENT_BATCH_THREAD_KEY] = parent
       end
 
       # Any Batches or Jobs created in the given block won't be assocaiated to the current batch
@@ -211,18 +201,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 +220,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)
@@ -244,6 +230,14 @@ module CanvasSync
       end
 
       class << self
+        def current
+          Thread.current[CURRENT_BATCH_THREAD_KEY]
+        end
+
+        def current_context
+          current&.context || {}
+        end
+
         def process_failed_job(bid, jid)
           _, pending, failed, children, complete, parent_bid = redis do |r|
             return unless r.exists?("BID-#{bid}")
@@ -283,11 +277,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 +405,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 +438,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/batch_aware_job.rb

@@ -6,17 +6,17 @@ module CanvasSync
       included do
         around_perform do |job, block|
           if (@bid) # This _must_ be @bid - not just bid
-            prev_batch = Thread.current[:batch]
+            prev_batch = Thread.current[CURRENT_BATCH_THREAD_KEY]
             begin
-              Thread.current[:batch] = Batch.new(@bid)
+              Thread.current[CURRENT_BATCH_THREAD_KEY] = Batch.new(@bid)
               block.call
-              Thread.current[:batch].save_context_changes
+              Thread.current[CURRENT_BATCH_THREAD_KEY].save_context_changes
               Batch.process_successful_job(@bid, job_id)
             rescue
               Batch.process_failed_job(@bid, job_id)
               raise
             ensure
-              Thread.current[:batch] = prev_batch
+              Thread.current[CURRENT_BATCH_THREAD_KEY] = prev_batch
             end
           else
             block.call
@@ -24,7 +24,7 @@ module CanvasSync
         end
 
         around_enqueue do |job, block|
-          if (batch = Thread.current[:batch])
+          if (batch = Thread.current[CURRENT_BATCH_THREAD_KEY])
             @bid = batch.bid
             batch.increment_job_queue(job_id) if @bid
           end
@@ -33,11 +33,11 @@ module CanvasSync
       end
 
       def bid
-        @bid || Thread.current[:batch]&.bid
+        @bid || Thread.current[CURRENT_BATCH_THREAD_KEY]&.bid
       end
 
       def batch
-        Thread.current[:batch]
+        Thread.current[CURRENT_BATCH_THREAD_KEY]
       end
 
       def batch_context

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/job_batches/sidekiq.rb

@@ -8,11 +8,11 @@ module CanvasSync
     module Sidekiq
       module WorkerExtension
         def bid
-          Thread.current[:batch].bid
+          Thread.current[CURRENT_BATCH_THREAD_KEY].bid
         end
 
         def batch
-          Thread.current[:batch]
+          Thread.current[CURRENT_BATCH_THREAD_KEY]
         end
 
         def batch_context
@@ -42,7 +42,7 @@ module CanvasSync
 
       class ClientMiddleware
         def call(_worker, msg, _queue, _redis_pool = nil)
-          if (batch = Thread.current[:batch]) && should_handle_batch?(msg)
+          if (batch = Thread.current[CURRENT_BATCH_THREAD_KEY]) && should_handle_batch?(msg)
             batch.increment_job_queue(msg['jid']) if (msg[:bid] = batch.bid)
           end
           yield
@@ -57,17 +57,17 @@ module CanvasSync
       class ServerMiddleware
         def call(_worker, msg, _queue)
           if (bid = msg['bid'])
-            prev_batch = Thread.current[:batch]
+            prev_batch = Thread.current[CURRENT_BATCH_THREAD_KEY]
             begin
-              Thread.current[:batch] = Batch.new(bid)
+              Thread.current[CURRENT_BATCH_THREAD_KEY] = Batch.new(bid)
               yield
-              Thread.current[:batch].save_context_changes
+              Thread.current[CURRENT_BATCH_THREAD_KEY].save_context_changes
               Batch.process_successful_job(bid, msg['jid'])
             rescue
               Batch.process_failed_job(bid, msg['jid'])
               raise
             ensure
-              Thread.current[:batch] = prev_batch
+              Thread.current[CURRENT_BATCH_THREAD_KEY] = prev_batch
             end
           else
             yield

data/lib/canvas_sync/jobs/sync_terms_job.rb

@@ -9,7 +9,7 @@ module CanvasSync
       #   will be started for each term in that scope. :models should be an array of
       #   models to sync.
       def perform(options)
-        CanvasSync.get_canvas_sync_client(batch_context).terms("self").all_pages!.each do |term_params|
+        CanvasSync.get_canvas_sync_client(batch_context).terms("self", workflow_state: ['all']).all_pages!.each do |term_params|
           if account_id = batch_context[:account_id]
             # These branches are primarily to support Legacy apps
             if Term.respond_to?(:create_or_update) && Term.method(:create_or_update).arity.abs == 2

data/lib/canvas_sync/version.rb

@@ -1,3 +1,3 @@
 module CanvasSync
-  VERSION = "0.17.23.beta6".freeze
+  VERSION = "0.17.24".freeze
 end