File: create_worker.rb

package info (click to toggle)
gitlab 17.6.5-19
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 629,368 kB
  • sloc: ruby: 1,915,304; javascript: 557,307; sql: 60,639; xml: 6,509; sh: 4,567; makefile: 1,239; python: 406
file content (48 lines) | stat: -rw-r--r-- 979 bytes parent folder | download
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
42
43
44
45
46
47
48
# frozen_string_literal: true

module ObjectPool
  class CreateWorker # rubocop:disable Scalability/IdempotentWorker
    include ApplicationWorker

    data_consistency :always

    sidekiq_options retry: 3
    include ObjectPoolQueue
    include ExclusiveLeaseGuard

    attr_reader :pool

    def perform(pool_id)
      @pool = PoolRepository.find_by_id(pool_id)
      return unless pool

      try_obtain_lease do
        perform_pool_creation
      end
    end

    private

    def perform_pool_creation
      return unless pool.failed? || pool.scheduled?

      # If this is a retry and the previous execution failed, deletion will
      # bring the pool back to a pristine state
      pool.delete_object_pool if pool.failed?

      pool.create_object_pool
      pool.mark_ready
    rescue StandardError => e
      pool.mark_failed
      raise e
    end

    def lease_key
      "object_pool:create:#{pool.id}"
    end

    def lease_timeout
      1.hour
    end
  end
end