gitlab-org--gitlab-foss/lib/github/collection.rb

27 lines
512 B
Ruby
Raw Normal View History

2017-04-12 01:33:54 +00:00
module Github
class Collection
def initialize(url)
@url = url
end
def fetch(query = {})
return [] if @url.blank?
Enumerator.new do |yielder|
loop do
response = client.get(@url, query)
response.body.each { |item| yielder << item }
raise StopIteration unless response.rels.key?(:next)
@url = response.rels[:next]
end
end.lazy
end
private
def client
@client ||= Github::Client.new
end
end
end