Module: CanvasSync::JobBatches::Sidekiq

Defined in:
lib/canvas_sync/job_batches/sidekiq.rb

Defined Under Namespace

Modules: WorkerExtension Classes: ClientMiddleware, ServerMiddleware

Class Method Summary collapse

Class Method Details

.configure ⇒ Object 



57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
 
# File 'lib/canvas_sync/job_batches/sidekiq.rb', line 57

def self.configure
  if defined?(::Sidekiq::Batch) && ::Sidekiq::Batch != JobBatches::Batch
    print "WARNING: Detected Sidekiq Pro or sidekiq-batch. CanvasSync JobBatches may not be fully compatible!"
  end

  ::Sidekiq.configure_client do |config|
    config.client_middleware do |chain|
      chain.remove ::Sidekiq::Batch::Middleware::ClientMiddleware if defined?(::Sidekiq::Batch::Middleware::ClientMiddleware)
      chain.add JobBatches::Sidekiq::ClientMiddleware
    end
  end
  ::Sidekiq.configure_server do |config|
    config.client_middleware do |chain|
      chain.remove ::Sidekiq::Batch::Middleware::ClientMiddleware if defined?(::Sidekiq::Batch::Middleware::ClientMiddleware)
      chain.add JobBatches::Sidekiq::ClientMiddleware
    end

    config.server_middleware do |chain|
      chain.remove ::Sidekiq::Batch::Middleware::ServerMiddleware if defined?(::Sidekiq::Batch::Middleware::ServerMiddleware)
      chain.add JobBatches::Sidekiq::ServerMiddleware
    end

    config.death_handlers << ->(job, ex) do
      return unless job['bid'].present?

      if defined?(::Apartment)
        ::Apartment::Tenant.switch(job['apartment'] || 'public') do
          Sidekiq::Batch.process_dead_job(job['bid'], job['jid'])
        end
      else
        Sidekiq::Batch.process_dead_job(job['bid'], job['jid'])
      end
    end
  end
  ::Sidekiq.const_set(:Batch, CanvasSync::JobBatches::Batch)
  # This alias helps apartment-sidekiq set itself up correctly
  ::Sidekiq::Batch.const_set(:Server, CanvasSync::JobBatches::Sidekiq::ServerMiddleware)
  ::Sidekiq::Worker.send(:include, JobBatches::Sidekiq::WorkerExtension)
end