summaryrefslogtreecommitdiff
path: root/lib/gitlab/sidekiq_middleware/duplicate_jobs/strategies/until_executed.rb
blob: 8c7e15364f86b4bf25c88d50e081476cc67960f0 (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
# 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 after the job has executed preventing a new job to be queued
        # while a job is still executing.
        class UntilExecuted < DeduplicatesWhenScheduling
          override :perform
          def perform(job)
            super

            yield

            should_reschedule = duplicate_job.should_reschedule?
            # Deleting before rescheduling to make sure we don't deduplicate again.
            duplicate_job.delete!
            duplicate_job.reschedule if should_reschedule
          end
        end
      end
    end
  end
end