lib/sidekiq/statistic/middleware.rb in sidekiq-statistic-1.0.0 vs lib/sidekiq/statistic/middleware.rb in sidekiq-statistic-1.1.0

@@ -12,13 +12,17 @@
         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
 
       def save_entry_for_worker(worker_status)
@@ -29,10 +33,11 @@
 
         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}: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