Class: Wurk::Middleware::CurrentAttributes::Load

Inherits:
Object
  • Object
show all
Includes:
ServerMiddleware
Defined in:
lib/wurk/middleware/current_attributes.rb

Overview

Restores each registered CurrentAttributes class for the duration of the inner block, then puts back whatever was there before — not a blanket reset. On a server thread the prior state is empty, so this is equivalent to resetting; but Load also runs on the CLIENT chain (persist registers it on both), where an enqueue happens mid-request with request-scoped attributes already set. Resetting there would wipe the caller’s state right after ‘perform_async`. Save/restore is what Sidekiq does and is correct on both chains. Restore runs in `ensure` to survive raises and Skip.

Registered on BOTH chains, so ‘call` takes an optional 4th arg: the client chain passes a `redis_pool`, the server chain stops at `queue`.

Instance Attribute Summary

Attributes included from ServerMiddleware

#config

Instance Method Summary collapse

Methods included from ServerMiddleware

#logger, #redis, #redis_pool

Constructor Details

#initialize(classes) ⇒ Load

Returns a new instance of Load.



85
86
87
 
# File 'lib/wurk/middleware/current_attributes.rb', line 85

def initialize(classes)
  @classes = classes
end

Instance Method Details

#call(_job_or_class, job, _queue, _redis_pool = nil) ⇒ Object 



89
90
91
92
93
94
95
96
97
98
99
100
 
# File 'lib/wurk/middleware/current_attributes.rb', line 89

def call(_job_or_class, job, _queue, _redis_pool = nil)
  previous = @classes.map { |klass| CurrentAttributes.snapshot(klass) }
  @classes.each_with_index do |klass, idx|
    CurrentAttributes.restore(klass, job[CurrentAttributes.key_for(idx)])
  end
  yield
ensure
  previous&.each_with_index do |attrs, idx|
    @classes[idx].reset
    CurrentAttributes.restore(@classes[idx], attrs)
  end
end