Class: Dependabot::Clients::GithubWithRetries

Inherits:
Object
  • Object
show all
Defined in:
lib/dependabot/clients/github_with_retries.rb

Constant Summary collapse

DEFAULT_CLIENT_ARGS = 
{
  connection_options: {
    request: {
      open_timeout: 2,
      timeout: 5
    }
  }
}.freeze
RETRYABLE_ERRORS = 
[
  Faraday::ConnectionFailed,
  Faraday::TimeoutError,
  Octokit::InternalServerError,
  Octokit::BadGateway
].freeze

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(max_retries: 3, **args) ⇒ GithubWithRetries

Proxying #



67
68
69
70
71
72
73
74
75
76
77
78
79
 
# File 'lib/dependabot/clients/github_with_retries.rb', line 67

def initialize(max_retries: 3, **args)
  args = DEFAULT_CLIENT_ARGS.merge(args)

  access_tokens = args.delete(:access_tokens) || []
  access_tokens << args[:access_token] if args[:access_token]
  access_tokens << nil if access_tokens.empty?
  access_tokens.uniq!

  @max_retries = max_retries || 3
  @clients = access_tokens.map do |token|
    Octokit::Client.new(args.merge(access_token: token))
  end
end

Dynamic Method Handling

This class handles dynamic methods through the method_missing method

#method_missing(method_name, *args, &block) ⇒ Object 



81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
 
# File 'lib/dependabot/clients/github_with_retries.rb', line 81

def method_missing(method_name, *args, &block)
  untried_clients = @clients.dup
  client = untried_clients.pop

  begin
    retry_connection_failures do
      if client.respond_to?(method_name)
        mutatable_args = args.map(&:dup)
        client.public_send(method_name, *mutatable_args, &block)
      else
        super
      end
    end
  rescue Octokit::NotFound, Octokit::Unauthorized, Octokit::Forbidden
    raise unless (client = untried_clients.pop)

    retry
  end
end

Class Method Details

.for_github_dot_com(credentials:) ⇒ Object 



41
42
43
44
45
46
47
48
49
 
# File 'lib/dependabot/clients/github_with_retries.rb', line 41

def self.for_github_dot_com(credentials:)
  access_tokens =
    credentials.
    select { |cred| cred["type"] == "git_source" }.
    select { |cred| cred["host"] == "github.com" }.
    map { |cred| cred.fetch("password") }

  new(access_tokens: access_tokens)
end

.for_source(source:, credentials:) ⇒ Object

Constructor methods #



28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/dependabot/clients/github_with_retries.rb', line 28

def self.for_source(source:, credentials:)
  access_tokens =
    credentials.
    select { |cred| cred["type"] == "git_source" }.
    select { |cred| cred["host"] == source.hostname }.
    map { |cred| cred.fetch("password") }

  new(
    access_tokens: access_tokens,
    api_endpoint: source.api_endpoint
  )
end

Instance Method Details

#fetch_commit(repo, branch) ⇒ Object

VCS Interface #



55
56
57
 
# File 'lib/dependabot/clients/github_with_retries.rb', line 55

def fetch_commit(repo, branch)
  ref(repo, "heads/#{branch}").object.sha
end

#fetch_default_branch(repo) ⇒ Object 



59
60
61
 
# File 'lib/dependabot/clients/github_with_retries.rb', line 59

def fetch_default_branch(repo)
  repository(repo).default_branch
end

#respond_to_missing?(method_name, include_private = false) ⇒ Boolean

Returns:

  • (Boolean) 


101
102
103
 
# File 'lib/dependabot/clients/github_with_retries.rb', line 101

def respond_to_missing?(method_name, include_private = false)
  @clients.first.respond_to?(method_name) || super
end

#retry_connection_failures ⇒ Object 



105
106
107
108
109
110
111
112
113
114
 
# File 'lib/dependabot/clients/github_with_retries.rb', line 105

def retry_connection_failures
  retry_attempt = 0

  begin
    yield
  rescue *RETRYABLE_ERRORS
    retry_attempt += 1
    retry_attempt <= @max_retries ? retry : raise
  end
end