Class: Sidekiq::Statistic::Middleware

Inherits:
Object
  • Object
show all
Defined in:
lib/sidekiq/statistic/middleware.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#msg ⇒ Object

Returns the value of attribute msg.



4
5
6
 
# File 'lib/sidekiq/statistic/middleware.rb', line 4

def msg
  @msg
end

Instance Method Details

#call(worker, msg, queue) ⇒ Object 



6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
 
# File 'lib/sidekiq/statistic/middleware.rb', line 6

def call(worker, msg, queue)
  worker_status = { last_job_status: 'passed'.freeze }
  start = Time.now

  yield
rescue => e
  worker_status[:last_job_status] = 'failed'.freeze

  raise e
ensure
  finish = Time.now
  worker_status[:queue] = msg['queue'.freeze]
  worker_status[:last_runtime] = finish.utc
  worker_status[:time] = (finish - start).to_f.round(3)
  worker_status[:class] = msg['wrapped'.freeze] || worker.class.to_s
  if worker_status[:class] == 'ActionMailer::DeliveryJob'.freeze
    worker_status[:class] = msg['args'.freeze].first['arguments'.freeze].first
  end

  save_entry_for_worker worker_status
end

#save_entry_for_worker(worker_status) ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
 
# File 'lib/sidekiq/statistic/middleware.rb', line 28

def save_entry_for_worker(worker_status)
  status = worker_status.dup
  time = worker_status[:last_runtime]
  realtime_hash = "#{REDIS_HASH}:realtime:#{time.sec}"
  worker_key = "#{time.to_date}:#{status.delete :class}"

  Sidekiq.redis do |redis|
    redis.pipelined do
      redis.hincrby REDIS_HASH, "#{worker_key}:#{status[:last_job_status]}", 1
      redis.hmset REDIS_HASH, "#{worker_key}:last_job_status", status[:last_job_status],
                              "#{worker_key}:last_time", status[:last_runtime],
                              "#{worker_key}:queue", status[:queue]
      redis.lpush "#{worker_key}:timeslist", status[:time]

      redis.hincrby realtime_hash, "#{status[:last_job_status]}:#{worker_status[:class]}", 1
      redis.expire realtime_hash, 2
    end
  end
end