Class: SidekiqStatus::ClientMiddleware

Inherits:
Object
  • Object
show all
Defined in:
lib/sidekiq_status/client_middleware.rb

Instance Method Summary collapse

Instance Method Details

#call(worker, item, queue, redis_pool = nil) ⇒ Object 



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 
# File 'lib/sidekiq_status/client_middleware.rb', line 5

def call(worker, item, queue, redis_pool = nil)
  worker = worker.constantize if worker.is_a?(String)
  return yield unless worker < SidekiqStatus::Worker

  # Don't start reporting status if the job is scheduled for the future
  # When perform_at/perform_in is called this middleware is invoked within the client process
  # and job arguments have 'at' parameter. If all middlewares pass the job
  # Sidekiq::Client#raw_push puts the job into 'schedule' sorted set.
  #
  # Later, Sidekiq server ruby process periodically polls this sorted sets and repushes all
  # scheduled jobs which are due to run. This repush invokes all client middlewares, but within
  # sidekiq server ruby process.
  #
  # Luckily for us, when job is repushed, it doesn't have 'at' argument.
  # So we can distinguish the condition of the middleware invokation: we don't create SidekiqStatus::Container
  # when job is scheduled to run in the future, but we create status container when previously scheduled
  # job is due to run.
  return yield if item['at']

  jid  = item['jid']
  args = item['args']

  SidekiqStatus::Container.create(
      'jid'    => jid,
      'worker' => worker.name,
      'queue'  => queue,
      'args'   => args
  )

  yield
rescue Exception => exc
  SidekiqStatus::Container.load(jid).delete rescue nil
  raise exc
end