Class: Sidekiq::AssuredJobs::Middleware

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

Instance Method Summary collapse

Instance Method Details

#call(worker, job, queue) ⇒ Object 



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
39
40
41
42
43
44
45
46
47
48
49
50
51
 
# File 'lib/sidekiq/assured_jobs/middleware.rb', line 6

def call(worker, job, queue)
  # Only track jobs that have assured_jobs: true option
  should_track = should_track_job?(worker, job)
  
  return yield unless should_track

  jid = job["jid"]
  instance_id = AssuredJobs.instance_id
  logger = AssuredJobs.logger

  # Create tracking keys (using custom namespacing)
  job_tracking_key = AssuredJobs.send(:namespaced_key, "jobs:#{instance_id}")
  job_data_key = AssuredJobs.send(:namespaced_key, "job:#{jid}")

  begin
    # Add job to tracking set and store job payload
    begin
      AssuredJobs.redis_sync do |conn|
        conn.multi do |multi|
          multi.sadd(job_tracking_key, jid)
          multi.set(job_data_key, job.to_json)
        end
      end
      logger.debug "AssuredJobs started tracking job #{jid} on instance #{instance_id}"
    rescue => e
      logger.error "AssuredJobs failed to start tracking job #{jid}: #{e.message}"
      logger.error e.backtrace.join("\n")
    end

    # Execute the job
    yield
  ensure
    # Remove job from tracking
    begin
      AssuredJobs.redis_sync do |conn|
        conn.multi do |multi|
          multi.srem(job_tracking_key, jid)
          multi.del(job_data_key)
        end
      end
      logger.debug "AssuredJobs stopped tracking job #{jid} on instance #{instance_id}"
    rescue => e
      logger.error "AssuredJobs failed to stop tracking job #{jid}: #{e.message}"
    end
  end
end