summaryrefslogtreecommitdiff
path: root/lib/gitlab/sidekiq_middleware/worker_context/client.rb
blob: 0eb52179db2a72bd0bd3d8334393261c2b184718 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# frozen_string_literal: true

module Gitlab
  module SidekiqMiddleware
    module WorkerContext
      class Client
        include Gitlab::SidekiqMiddleware::WorkerContext

        def call(worker_class_or_name, job, _queue, _redis_pool, &block)
          worker_class = worker_class_or_name.to_s.safe_constantize

          # Mailers can't be constantized like this
          return yield unless worker_class
          return yield unless worker_class.include?(::ApplicationWorker)

          context_for_args = worker_class.context_for_arguments(job['args'])

          wrap_in_optional_context(context_for_args, &block)
        end
      end
    end
  end
end