toptal/chewy

module Chewy
  class Strategy
    # The strategy works the same way as atomic, but performs
    # async index update driven by sidekiq
    #
    #   Chewy.strategy(:sidekiq) do
    #     User.all.map(&:save) # Does nothing here
    #     Post.all.map(&:save) # And here
    #     # It imports all the changed users and posts right here
    #   end
    #
    class Sidekiq < Atomic
      class Worker
        include ::Sidekiq::Worker

        def perform(type, ids, options = {})
          options[:refresh] = !Chewy.disable_refresh_async if Chewy.disable_refresh_async
          type.constantize.import!(ids, **options)
        end
      end

      def leave
        @stash.each do |type, ids|
          next if ids.empty?

          ::Sidekiq::Client.push(
            'queue' => sidekiq_queue,
            'class' => Chewy::Strategy::Sidekiq::Worker,
            'args'  => [type.name, ids]
          )
        end
      end

    private

      def sidekiq_queue
        Chewy.settings.dig(:sidekiq, :queue) || 'chewy'
      end
    end
  end
end