lib/gitlab/worker/unique.rb


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

require 'digest'

module Gitlab
  module Worker
    class Unique
      def initialize(worker, *args)
        @worker = worker
        @args = args
      end

      def uuid
        @uuid ||= Digest::SHA1
          .hexdigest(@worker.name + @args.to_json)
      end

      def lease
        @lease ||= Gitlab::ExclusiveLease
          .new(uuid, timeout: 1.hour.to_i)
      end

      def schedule!
        if lease.try_obtain
          @worker.perform_async(*@args)
        end
      end

      def release!
        lease.cancel!
      end
    end
  end
end