Module: Sidekiq::Batch::Middleware

Defined in:
lib/sidekiq/batch/middleware.rb

Defined Under Namespace

Classes: ClientMiddleware, ServerMiddleware

Class Method Summary collapse

Class Method Details

.configure ⇒ Object 



33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
 
# File 'lib/sidekiq/batch/middleware.rb', line 33

def self.configure
  Sidekiq.configure_client do |config|
    config.client_middleware do |chain|
      chain.add Sidekiq::Batch::Middleware::ClientMiddleware
    end
  end
  Sidekiq.configure_server do |config|
    config.client_middleware do |chain|
      chain.add Sidekiq::Batch::Middleware::ClientMiddleware
    end
    config.server_middleware do |chain|
      chain.add Sidekiq::Batch::Middleware::ServerMiddleware
    end
  end
  Sidekiq::Worker.send(:define_method, 'bid') do
    Thread.current[:bid]
  end
  Sidekiq::Worker.send(:define_method, 'batch') do
    Sidekiq::Batch.new(Thread.current[:bid]) if Thread.current[:bid]
  end
end