summaryrefslogtreecommitdiff
path: root/lib/gitlab/sidekiq_middleware/worker_context/server.rb
blob: d2d84742c17fda12e40fbbe3aec76865063e890a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
# 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