BitBucket refactoring. Iteration 3

This commit is contained in:
Valery Sizov 2016-12-07 15:54:32 +02:00
parent b12d654183
commit 98c0eb0f75
4 changed files with 15 additions and 21 deletions

View file

@ -6,35 +6,26 @@ module Bitbucket
def issues(repo)
path = "/repositories/#{repo}/issues"
paginator = Paginator.new(connection, path, :issue)
Collection.new(paginator)
get_collection(path, :issue)
end
def issue_comments(repo, issue_id)
path = "/repositories/#{repo}/issues/#{issue_id}/comments"
paginator = Paginator.new(connection, path, :comment)
Collection.new(paginator)
get_collection(path, :comment)
end
def pull_requests(repo)
path = "/repositories/#{repo}/pullrequests?state=ALL"
paginator = Paginator.new(connection, path, :pull_request)
Collection.new(paginator)
get_collection(path, :pull_request)
end
def pull_request_comments(repo, pull_request)
path = "/repositories/#{repo}/pullrequests/#{pull_request}/comments"
paginator = Paginator.new(connection, path, :pull_request_comment)
Collection.new(paginator)
get_collection(path, :pull_request_comment)
end
def pull_request_diff(repo, pull_request)
path = "/repositories/#{repo}/pullrequests/#{pull_request}/diff"
connection.get(path)
end
@ -45,9 +36,7 @@ module Bitbucket
def repos
path = "/repositories/#{user.username}"
paginator = Paginator.new(connection, path, :repo)
Collection.new(paginator)
get_collection(path, :repo)
end
def user
@ -60,5 +49,10 @@ module Bitbucket
private
attr_reader :connection
def get_collection(path, type)
paginator = Paginator.new(connection, path, type)
Collection.new(paginator)
end
end
end

View file

@ -3,7 +3,7 @@ module Bitbucket
def initialize(paginator)
super() do |yielder|
loop do
paginator.next.each { |item| yielder << item }
paginator.items.each { |item| yielder << item }
end
end

View file

@ -11,7 +11,7 @@ module Bitbucket
connection.set_default_query_parameters(pagelen: PAGE_LENGTH, sort: :created_on)
end
def next
def items
raise StopIteration unless has_next_page?
@page = fetch_next_page

View file

@ -50,7 +50,7 @@ module Gitlab
if issue.persisted?
client.issue_comments(repo, issue.iid).each do |comment|
# The note can be blank for issue service messages like "Chenged title: ..."
# The note can be blank for issue service messages like "Changed title: ..."
# We would like to import those comments as well but there is no any
# specific parameter that would allow to process them, it's just an empty comment.
# To prevent our importer from just crashing or from creating useless empty comments
@ -70,8 +70,8 @@ module Gitlab
end
end
end
rescue ActiveRecord::RecordInvalid
nil
rescue ActiveRecord::RecordInvalid => e
Rails.logger.error("Bitbucket importer ERROR in #{project.path_with_namespace}: Couldn't import record properly #{e.message}")
end
def import_pull_requests