File: retry_with_delay.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 (32 lines) | stat: -rw-r--r-- 664 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
# frozen_string_literal: true

module BitbucketServer
  module RetryWithDelay
    extend ActiveSupport::Concern

    MAXIMUM_DELAY = 20

    def retry_with_delay(&block)
      run_retry_with_delay(&block)
    end

    private

    def run_retry_with_delay
      response = yield

      if response.code == 429 && response.headers.has_key?('retry-after')
        retry_after = response.headers['retry-after'].to_i

        if retry_after <= MAXIMUM_DELAY
          logger.info(message: "Retrying in #{retry_after} seconds due to 429 Too Many Requests")
          sleep retry_after

          response = yield
        end
      end

      response
    end
  end
end