switchman-inst-jobs 3.0.0 → 3.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7d0284b8288ca18b825002cb7df51b1f4f6469f3f62202d997934a42507f91f4
-  data.tar.gz: 7a4c11dca853c6d5aad2731f5aa9bc0e0af19cb2be149f9c22e57ebc6fd6e4b9
+  metadata.gz: 7507e39f0e9a097dd11cc1389edad14b92d77b959d00deb3cd8610e43b65e042
+  data.tar.gz: 1d6709c61b221f6e1798dfceff1a180507f905a6b398bbd609060fd202c2e8d6
 SHA512:
-  metadata.gz: e9a126073f7a1904eadf82f265f8c052c9fd5291b05760099f5bbc12dfd8714f4c8f8b2636a9c534568897cac676430b7741b1c36685d5d7c2291568795f9c80
-  data.tar.gz: f14a7b3ceb47de8c0e9f0b0d8f3a34e73e230259a94430c24c9b429d4be34722e4a7f96b5840c36566cbc054d164e777ed3a5d68e36fe84b184b50b34824646c
+  metadata.gz: 5e617b32fbda0fdaa7898bfda6b0d399792be12d03be8d0caa82ac9ad1a4bc9c29143dcca1b79de197eb0be637b61d91e9e06350ae264f02fb949baf4df132b3
+  data.tar.gz: 81b00f0665470553ae32bd88304b4e88a5a8ead5e8fc03cf728161715054a85523fb44a3796252e2a136a2cd58477d05ebb27f1cc13634390e17c3dbbee3561d

data/db/migrate/20110610213249_optimize_delayed_jobs.rb

@@ -29,7 +29,7 @@ class OptimizeDelayedJobs < ActiveRecord::Migration[4.2]
     add_index :delayed_jobs, %w[strand id], name: 'index_delayed_jobs_on_strand'
 
     # move all failed jobs to the new failed table
-    Delayed::Backend::ActiveRecord::Job.where('failed_at IS NOT NULL').find_each do |job|
+    Delayed::Backend::ActiveRecord::Job.where.not(failed_at: nil).find_each do |job|
       job.fail! unless job.on_hold?
     end
   end

data/lib/switchman_inst_jobs/delayed/backend/base.rb

@@ -43,8 +43,8 @@ module SwitchmanInstJobs
           end
 
           def configured_shard_ids
-            (::Delayed::Settings.worker_config.try(:[], 'workers') || [])
-              .map { |w| w['shard'] }.compact.uniq
+            (::Delayed::Settings.worker_config.try(:[], 'workers') || []).
+              map { |w| w['shard'] }.compact.uniq
           end
 
           def processes_locked_locally
@@ -93,7 +93,7 @@ module SwitchmanInstJobs
           # likely a missing shard with a stale cache
           current_shard.send(:clear_cache)
           ::Switchman::Shard.clear_cache
-          raise ShardNotFoundError, shard_id unless ::Switchman::Shard.where(id: shard_id).exists?
+          raise ShardNotFoundError, shard_id unless ::Switchman::Shard.exists?(id: shard_id)
 
           raise
         end

data/lib/switchman_inst_jobs/delayed/message_sending.rb

@@ -1,8 +1,9 @@
 module SwitchmanInstJobs
   module Delayed
     module MessageSending
-      def delay(**enqueue_args)
-        return self if ::Switchman::DatabaseServer.creating_new_shard
+      def delay(sender: nil, synchronous: false, **enqueue_args)
+        sender ||= __calculate_sender_for_delay
+        synchronous ||= ::Switchman::DatabaseServer.creating_new_shard
 
         super
       end

data/lib/switchman_inst_jobs/delayed/pool.rb

@@ -17,7 +17,7 @@ module SwitchmanInstJobs
           # We purposely don't .compact to remove nils here, since if any
           # workers are on the default jobs shard we want to unlock against
           # that shard too.
-          shard_ids = @config[:workers].map { |c| c[:shard] }.uniq
+          shard_ids = @config[:workers].pluck(:shard).uniq
           shards = shard_ids.map { |shard_id| ::Delayed::Worker.shard(shard_id) }
         end
         ::Switchman::Shard.with_each_shard(shards, [:delayed_jobs]) do

data/lib/switchman_inst_jobs/delayed/worker/health_check.rb

@@ -19,7 +19,7 @@ module SwitchmanInstJobs
           end
 
           def reschedule_abandoned_jobs(call_super: false)
-            shards = ::Switchman::Shard.delayed_jobs_shards
+            shards = ::Switchman::Shard.delayed_jobs_shards.to_a
             call_super = shards.first if shards.length == 1
             unless call_super == false
               call_super.activate(:delayed_jobs) do

data/lib/switchman_inst_jobs/jobs_migrator.rb

@@ -29,26 +29,41 @@ module SwitchmanInstJobs
 
       def migrate_shards(shard_map)
         source_shards = Set[]
+        target_shards = Hash.new([])
         shard_map.each do |(shard, target_shard)|
           shard = ::Switchman::Shard.find(shard) unless shard.is_a?(::Switchman::Shard)
           source_shards << shard.delayed_jobs_shard.id
-          # If target_shard is an int, it won't have an id, but we can just use it as is
-          shard.update(delayed_jobs_shard_id: target_shard.try(:id) || target_shard, block_stranded: true)
+          target_shard = target_shard.try(:id) || target_shard
+          target_shards[target_shard] += [shard.id]
         end
 
+        # Do the updates in batches and then just clear redis instead of clearing them one at a time
+        target_shards.each do |target_shard, shards|
+          ::Switchman::Shard.where(id: shards).update_all(delayed_jobs_shard_id: target_shard, block_stranded: true)
+        end
+        clear_shard_cache
+
         # Wait a little over the 60 second in-process shard cache clearing
         # threshold to ensure that all new stranded jobs are now being
         # enqueued with next_in_strand: false
-        Rails.logger.debug("Waiting for caches to clear (#{source_shard.id} -> #{target_shard.id})")
+        Rails.logger.debug('Waiting for caches to clear')
         sleep(65) unless @skip_cache_wait
 
-        # TODO: 4 has been picked completely out of a hat.  We should make it configurable or something
-        Parallel.each(source_shards, in_processes: 4) do |s|
-          # Ensure the child processes don't share connections with the parent
-          Delayed::Pool.on_fork.call
-          ActiveRecord::Base.clear_all_connections!
-          s.activate(:delayed_jobs) { run }
+        ::Switchman::Shard.clear_cache
+        # rubocop:disable Style/CombinableLoops
+        # We first migrate strands so that we can stop blocking strands before we migrate unstranded jobs
+        source_shards.each do |s|
+          ::Switchman::Shard.lookup(s).activate(:delayed_jobs) { migrate_strands }
+        end
+
+        source_shards.each do |s|
+          ::Switchman::Shard.lookup(s).activate(:delayed_jobs) { migrate_everything }
         end
+        # rubocop:enable Style/CombinableLoops
+      end
+
+      def clear_shard_cache
+        ::Switchman.cache.clear
       end
 
       # This method expects that all relevant shards already have block_stranded: true
@@ -71,7 +86,7 @@ module SwitchmanInstJobs
         #    those (= do nothing since it should already be true)
 
         source_shard = ::Switchman::Shard.current(:delayed_jobs)
-        strand_scope = ::Delayed::Job.shard(source_shard).where('strand IS NOT NULL')
+        strand_scope = ::Delayed::Job.shard(source_shard).where.not(strand: nil)
         shard_map = build_shard_map(strand_scope, source_shard)
         shard_map.each do |(target_shard, source_shard_ids)|
           shard_scope = strand_scope.where(shard_id: source_shard_ids)
@@ -92,7 +107,7 @@ module SwitchmanInstJobs
                 # We lock it to ensure that the jobs worker can't delete it until we are done moving the strand
                 # Since we only unlock it on the new jobs queue *after* deleting from the original
                 # the lock ensures the blocker always gets unlocked
-                first = this_strand_scope.where('locked_by IS NOT NULL').next_in_strand_order.lock.first
+                first = this_strand_scope.where.not(locked_by: nil).next_in_strand_order.lock.first
                 if first
                   first_job = ::Delayed::Job.create!(strand: strand, next_in_strand: false)
                   first_job.payload_object = ::Delayed::PerformableMethod.new(Kernel, :sleep, args: [0])
@@ -106,8 +121,8 @@ module SwitchmanInstJobs
                   first_job.save!
                   # the rest of 3) is taken care of here
                   # make sure that all the jobs moved over are NOT next in strand
-                  ::Delayed::Job.where(next_in_strand: true, strand: strand, locked_by: nil)
-                    .update_all(next_in_strand: false)
+                  ::Delayed::Job.where(next_in_strand: true, strand: strand, locked_by: nil).
+                    update_all(next_in_strand: false)
                 end
 
                 # 4) is taken care of here, by leaveing next_in_strand alone and
@@ -123,26 +138,37 @@ module SwitchmanInstJobs
               end
             end
 
-            ::Switchman::Shard.find(source_shard_ids).each do |shard|
-              shard.update(block_stranded: false)
+            updated = ::Switchman::Shard.where(id: source_shard_ids, block_stranded: true).
+              update_all(block_stranded: false)
+            # If this is being manually re-run for some reason to clean something up, don't wait for nothing to happen
+            unless updated.zero?
+              clear_shard_cache
+              # Wait a little over the 60 second in-process shard cache clearing
+              # threshold to ensure that all new stranded jobs are now being
+              # enqueued with next_in_strand: false
+              Rails.logger.debug("Waiting for caches to clear (#{source_shard.id} -> #{target_shard.id})")
+              # for spec usage only
+              sleep(65) unless @skip_cache_wait
             end
-            # Wait a little over the 60 second in-process shard cache clearing
-            # threshold to ensure that all new stranded jobs are now being
-            # enqueued with next_in_strand: false
-            Rails.logger.debug("Waiting for caches to clear (#{source_shard.id} -> #{target_shard.id})")
-            # for spec usage only
-            sleep(65) unless @skip_cache_wait
+            ::Switchman::Shard.clear_cache
             # At this time, let's unblock all the strands on the target shard that aren't being held by a blocker
             # but actually could have run and we just didn't know it because we didn't know if they had jobs
             # on the source shard
-            # rubocop:disable Layout/LineLength
-            strands_to_unblock = shard_scope.where.not(source: 'JobsMigrator::StrandBlocker')
-              .distinct
-              .where("NOT EXISTS (SELECT 1 FROM #{::Delayed::Job.quoted_table_name} dj2 WHERE delayed_jobs.strand=dj2.strand AND next_in_strand)")
-              .pluck(:strand)
-            # rubocop:enable Layout/LineLength
-            strands_to_unblock.each do |strand|
-              Delayed::Job.where(strand: strand).next_in_strand_order.first.update_attribute(:next_in_strand, true)
+            target_shard.activate(:delayed_jobs) do
+              loop do
+                # We only want to unlock stranded jobs where they don't belong to a blocked shard (if they *do* belong)
+                # to a blocked shard, they must be part of a concurrent jobs migration from a different source shard to
+                # this target shard, so we shouldn't unlock them yet.  We only ever unlock one job here to keep the
+                # logic cleaner; if the job is n-stranded, after the first one runs, the trigger will unlock larger
+                # batches
+                break if ::Delayed::Job.where(id: ::Delayed::Job.select('DISTINCT ON (strand) id').
+                  where.not(strand: nil).
+                  where.not(shard_id: ::Switchman::Shard.where(block_stranded: true).pluck(:id)).where(
+                    ::Delayed::Job.select(1).from("#{::Delayed::Job.quoted_table_name} dj2").
+                    where("dj2.next_in_strand = true OR dj2.source = 'JobsMigrator::StrandBlocker'").
+                    where('dj2.strand = delayed_jobs.strand').arel.exists.not
+                  ).order(:strand, :strand_order_override, :id)).limit(500).update_all(next_in_strand: true).zero?
+              end
             end
           end
         end
@@ -242,7 +268,7 @@ module SwitchmanInstJobs
 
         connection.execute "COPY #{::Delayed::Job.quoted_table_name} (#{quoted_keys}) FROM STDIN"
         records.map do |record|
-          connection.raw_connection.put_copy_data(keys.map { |k| quote_text(record[k]) }.join("\t") + "\n")
+          connection.raw_connection.put_copy_data("#{keys.map { |k| quote_text(record[k]) }.join("\t")}\n")
         end
         connection.clear_query_cache
         connection.raw_connection.put_copy_end

data/lib/switchman_inst_jobs/switchman/database_server.rb

@@ -11,7 +11,7 @@ module SwitchmanInstJobs
         # shard's delayed_jobs_shard
         if shard&.default?
           # first look for any shard that behaves like a jobs shard
-          dj_shard ||= ::Switchman::Shard.delayed_jobs_shards.first
+          dj_shard ||= ::Switchman::Shard.delayed_jobs_shards.find(&:database_server)
           # we're really truly out of options, use the default shard itself
           dj_shard ||= shard
         end

data/lib/switchman_inst_jobs/switchman/shard.rb

@@ -14,7 +14,7 @@ module SwitchmanInstJobs
           shard = ::Switchman::Shard.lookup(delayed_jobs_shard_id)
           return shard if shard
         end
-        database_server&.delayed_jobs_shard(self)
+        @delayed_jobs_shard ||= database_server&.delayed_jobs_shard(self)
       end
 
       # Adapted from hold/unhold methods in base delayed jobs base
@@ -28,9 +28,9 @@ module SwitchmanInstJobs
         return unless wait
 
         delayed_jobs_shard.activate(:delayed_jobs) do
-          while ::Delayed::Job.where(shard_id: id)
-              .where.not(locked_at: nil)
-              .where.not(locked_by: ::Delayed::Backend::Base::ON_HOLD_LOCKED_BY).exists?
+          while ::Delayed::Job.where(shard_id: id).
+              where.not(locked_at: nil).
+              where.not(locked_by: ::Delayed::Backend::Base::ON_HOLD_LOCKED_BY).exists?
             sleep 10
             lock_jobs_for_hold
           end
@@ -41,9 +41,9 @@ module SwitchmanInstJobs
         self.jobs_held = false
         save! if changed?
         delayed_jobs_shard.activate(:delayed_jobs) do
-          ::Delayed::Job.where(locked_by: ::Delayed::Backend::Base::ON_HOLD_LOCKED_BY, shard_id: id)
-            .in_batches(of: 10_000)
-            .update_all(
+          ::Delayed::Job.where(locked_by: ::Delayed::Backend::Base::ON_HOLD_LOCKED_BY, shard_id: id).
+            in_batches(of: 10_000).
+            update_all(
               locked_by: nil,
               locked_at: nil,
               attempts: 0,
@@ -111,32 +111,27 @@ module SwitchmanInstJobs
         end
 
         def delayed_jobs_shards
-          unless instance_variable_defined?(:@delayed_jobs_shards)
-            # re-entrancy protection
-            @delayed_jobs_shards = begin
-              shard_dj_shards = [] unless ::Switchman::Shard.columns_hash.key?('delayed_jobs_shard_id')
-              shard_dj_shards ||= begin
-                ::Switchman::Shard
-                  .where.not(delayed_jobs_shard_id: nil)
-                  .distinct
-                  .pluck(:delayed_jobs_shard_id)
-                  .map { |id| ::Switchman::Shard.lookup(id) }
-                  .compact
-              end
-              # set it temporarily, to avoid the default shard falling back to itself
-              # if other shards are usable
-              @delayed_jobs_shards = shard_dj_shards.uniq.sort
-
-              db_dj_shards = ::Switchman::DatabaseServer.all.map do |db|
-                next db.shards.to_a if db.config[:delayed_jobs_shard] == 'self'
-
-                db.delayed_jobs_shard
-              end.compact.flatten.uniq # yes, all three
-
-              (db_dj_shards + shard_dj_shards).uniq.sort
-            end
+          return none unless ::Switchman::Shard.columns_hash.key?('delayed_jobs_shard_id')
+
+          scope = ::Switchman::Shard.unscoped.
+            where(id: ::Switchman::Shard.unscoped.distinct.where.not(delayed_jobs_shard_id: nil).
+            select(:delayed_jobs_shard_id))
+          db_jobs_shards = ::Switchman::DatabaseServer.all.map { |db| db.config[:delayed_jobs_shard] }.uniq
+          db_jobs_shards.delete(nil)
+          has_self = db_jobs_shards.delete('self')
+          scope = scope.or(::Switchman::Shard.unscoped.where(id: db_jobs_shards)) unless db_jobs_shards.empty?
+
+          if has_self
+            self_dbs = ::Switchman::DatabaseServer.all.
+              select { |db| db.config[:delayed_jobs_shard] == 'self' }.map(&:id)
+            scope = scope.or(::Switchman::Shard.unscoped.
+              where(id: ::Switchman::Shard.unscoped.where(delayed_jobs_shard_id: nil, database_server_id: self_dbs).
+              select(:id)))
           end
-          @delayed_jobs_shards
+          @jobs_scope_empty = !scope.exists? unless instance_variable_defined?(:@jobs_scope_empty)
+          return [::Switchman::Shard.default] if @jobs_scope_empty
+
+          ::Switchman::Shard.merge(scope)
         end
       end
     end

data/lib/switchman_inst_jobs/timed_cache.rb

@@ -6,7 +6,7 @@ module SwitchmanInstJobs
       @cached_at = Time.zone.now
     end
 
-    def clear(force = false)
+    def clear(force: false)
       if force || @cached_at < @timeout.call
         @block.call
         @cached_at = Time.zone.now

data/lib/switchman_inst_jobs/version.rb

@@ -1,3 +1,3 @@
 module SwitchmanInstJobs
-  VERSION = '3.0.0'.freeze
+  VERSION = '3.0.5'.freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: switchman-inst-jobs
 version: !ruby/object:Gem::Version
-  version: 3.0.0
+  version: 3.0.5
 platform: ruby
 authors:
 - Bryan Petty
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2020-10-26 00:00:00.000000000 Z
+date: 2020-11-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: inst-jobs
@@ -17,6 +17,9 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.3
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
@@ -24,6 +27,9 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '1.0'
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.0.3
 - !ruby/object:Gem::Dependency
   name: parallel
   requirement: !ruby/object:Gem::Requirement
@@ -204,28 +210,28 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.79.0
+        version: 1.3.1
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.79.0
+        version: 1.3.1
 - !ruby/object:Gem::Dependency
   name: rubocop-rails
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.4.2
+        version: 2.8.1
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 2.4.2
+        version: 2.8.1
 - !ruby/object:Gem::Dependency
   name: simplecov
   requirement: !ruby/object:Gem::Requirement