Contents

# frozen_string_literal: true

module Gouda
  def self.in_bulk(&blk)
    if Thread.current[:gouda_bulk_buffer].nil?
      Thread.current[:gouda_bulk_buffer] = []
      retval = yield
      buf, Thread.current[:gouda_bulk_buffer] = Thread.current[:gouda_bulk_buffer], nil
      enqueue_jobs_via_their_adapters(buf)
      retval
    else # There already is an open bulk
      yield
    end
  end

  # This method exists in edge Rails so probably can be replaced later:
  # https://github.com/rails/rails/commit/9b62f88a2fde0d2bf8c4f6e3bcd06ecba7ca9d8d
  def self.enqueue_jobs_via_their_adapters(active_jobs)
    jobs_per_adapter = active_jobs.compact.group_by { |aj| aj.class.queue_adapter }
    jobs_per_adapter.each_pair do |adapter, active_jobs|
      if adapter.respond_to?(:enqueue_all)
        adapter.enqueue_all(active_jobs)
      else
        active_jobs.each { |aj| adapter.enqueue(aj) }
      end
    end
  end

  module BulkAdapterExtension
    def enqueue_all(active_jobs)
      if Thread.current[:gouda_bulk_buffer]
        Thread.current[:gouda_bulk_buffer].append(*active_jobs)
        active_jobs
      else
        super
      end
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
gouda-0.1.4	lib/gouda/bulk.rb
gouda-0.1.3	lib/gouda/bulk.rb
gouda-0.1.2	lib/gouda/bulk.rb
gouda-0.1.1	lib/gouda/bulk.rb
gouda-0.1.0	lib/gouda/bulk.rb