зеркало из https://github.com/github/ruby.git
80 строки
1.6 KiB
Ruby
80 строки
1.6 KiB
Ruby
# frozen_string_literal: true
|
|
##
|
|
# The BestSet chooses the best available method to query a remote index.
|
|
#
|
|
# It combines IndexSet and APISet
|
|
|
|
class Gem::Resolver::BestSet < Gem::Resolver::ComposedSet
|
|
|
|
##
|
|
# Creates a BestSet for the given +sources+ or Gem::sources if none are
|
|
# specified. +sources+ must be a Gem::SourceList.
|
|
|
|
def initialize sources = Gem.sources
|
|
super()
|
|
|
|
@sources = sources
|
|
end
|
|
|
|
##
|
|
# Picks which sets to use for the configured sources.
|
|
|
|
def pick_sets # :nodoc:
|
|
@sources.each_source do |source|
|
|
@sets << source.dependency_resolver_set
|
|
end
|
|
end
|
|
|
|
def find_all req # :nodoc:
|
|
pick_sets if @remote and @sets.empty?
|
|
|
|
super
|
|
rescue Gem::RemoteFetcher::FetchError => e
|
|
replace_failed_api_set e
|
|
|
|
retry
|
|
end
|
|
|
|
def prefetch reqs # :nodoc:
|
|
pick_sets if @remote and @sets.empty?
|
|
|
|
super
|
|
end
|
|
|
|
def pretty_print q # :nodoc:
|
|
q.group 2, '[BestSet', ']' do
|
|
q.breakable
|
|
q.text 'sets:'
|
|
|
|
q.breakable
|
|
q.pp @sets
|
|
end
|
|
end
|
|
|
|
##
|
|
# Replaces a failed APISet for the URI in +error+ with an IndexSet.
|
|
#
|
|
# If no matching APISet can be found the original +error+ is raised.
|
|
#
|
|
# The calling method must retry the exception to repeat the lookup.
|
|
|
|
def replace_failed_api_set error # :nodoc:
|
|
uri = error.uri
|
|
uri = URI uri unless URI === uri
|
|
uri.query = nil
|
|
|
|
raise error unless api_set = @sets.find { |set|
|
|
Gem::Resolver::APISet === set and set.dep_uri == uri
|
|
}
|
|
|
|
index_set = Gem::Resolver::IndexSet.new api_set.source
|
|
|
|
@sets.map! do |set|
|
|
next set unless set == api_set
|
|
index_set
|
|
end
|
|
end
|
|
|
|
end
|
|
|