Contents

# frozen_string_literal: true

module AdvancedSneakersActiveJob
  # Sneakers uses Sneakers::Worker::Classes array to track all workers.
  # WorkersRegistry mocks original array to track ActiveJob workers separately.
  class WorkersRegistry
    attr_reader :sneakers_workers, :activejob_workers

    delegate :activejob_workers_strategy, to: :'AdvancedSneakersActiveJob.config'

    delegate :empty?, to: :call

    def initialize
      @sneakers_workers = []
      @activejob_workers = []
    end

    def <<(worker)
      if worker <= ActiveJob::QueueAdapters::AdvancedSneakersAdapter::JobWrapper
        activejob_workers << worker
      else
        sneakers_workers << worker
      end
    end

    # Sneakers workergroup supports callable objects.
    # https://github.com/jondot/sneakers/pull/210/files
    # https://github.com/jondot/sneakers/blob/7a972d22a58de8a261a738d9a1e5fb51f9608ede/lib/sneakers/workergroup.rb#L28
    def call
      case activejob_workers_strategy
      when :only    then activejob_workers
      when :exclude then sneakers_workers
      when :include then sneakers_workers + activejob_workers
      else
        raise "Unknown activejob_workers_strategy '#{activejob_workers_strategy}'"
      end
    end

    # For cleaner output on inspecting Sneakers::Worker::Classes in console.
    def inspect
      {
        sneakers_workers: sneakers_workers,
        activejob_workers: activejob_workers
      }
    end
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
advanced-sneakers-activejob-0.2.2	lib/advanced_sneakers_activejob/workers_registry.rb
advanced-sneakers-activejob-0.2.1	lib/advanced_sneakers_activejob/workers_registry.rb
advanced-sneakers-activejob-0.2.0	lib/advanced_sneakers_activejob/workers_registry.rb
advanced-sneakers-activejob-0.1.0	lib/advanced_sneakers_activejob/workers_registry.rb