22 lines
592 B
Ruby
22 lines
592 B
Ruby
|
# frozen_string_literal: true
|
||
|
|
|
||
|
|
module Gitlab
|
||
|
|
module SidekiqMiddleware
|
||
|
|
module WorkerContext
|
||
|
|
class Server
|
||
|
|
include Gitlab::SidekiqMiddleware::WorkerContext
|
||
|
|
|
||
|
|
def call(worker, job, _queue, &block)
|
||
|
|
worker_class = worker.class
|
||
|
|
|
||
|
|
# This is not a worker we know about, perhaps from a gem
|
||
|
|
return yield unless worker_class.respond_to?(:get_worker_context)
|
||
|
|
|
||
|
|
# Use the context defined on the class level as a base context
|
||
|
|
wrap_in_optional_context(worker_class.get_worker_context, &block)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|