summaryrefslogtreecommitdiff
path: root/lib/gitlab/sidekiq_middleware/duplicate_jobs/strategies/until_executing.rb
blob: 674e436b7143293c4a37375989d861521527127b (plain)
1
2
3
4
5
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
# frozen_string_literal: true

module Gitlab
  module SidekiqMiddleware
    module DuplicateJobs
      module Strategies
        # This strategy takes a lock before scheduling the job in a queue and
        # removes the lock before the job starts allowing a new job to be queued
        # while a job is still executing.
        class UntilExecuting
          def initialize(duplicate_job)
            @duplicate_job = duplicate_job
          end

          def schedule(job)
            if duplicate_job.check! && duplicate_job.duplicate?
              job['duplicate-of'] = duplicate_job.existing_jid
            end

            if duplicate_job.droppable?
              Gitlab::SidekiqLogging::DeduplicationLogger.instance.log(job, "dropped until executing")
              return false
            end

            yield
          end

          def perform(_job)
            duplicate_job.delete!

            yield
          end

          private

          attr_reader :duplicate_job
        end
      end
    end
  end
end