73 lines
2.5 KiB
Ruby
73 lines
2.5 KiB
Ruby
|
module Gitlab
|
||
|
|
module GitalyClient
|
||
|
|
class ConflictsService
|
||
|
|
include Gitlab::EncodingHelper
|
||
|
|
|
||
|
|
MAX_MSG_SIZE = 128.kilobytes.freeze
|
||
|
|
|
||
|
|
def initialize(repository, our_commit_oid, their_commit_oid)
|
||
|
|
@gitaly_repo = repository.gitaly_repository
|
||
|
|
@repository = repository
|
||
|
|
@our_commit_oid = our_commit_oid
|
||
|
|
@their_commit_oid = their_commit_oid
|
||
|
|
end
|
||
|
|
|
||
|
|
def list_conflict_files
|
||
|
|
request = Gitaly::ListConflictFilesRequest.new(
|
||
|
|
repository: @gitaly_repo,
|
||
|
|
our_commit_oid: @our_commit_oid,
|
||
|
|
their_commit_oid: @their_commit_oid
|
||
|
|
)
|
||
|
|
response = GitalyClient.call(@repository.storage, :conflicts_service, :list_conflict_files, request)
|
||
|
|
|
||
|
|
GitalyClient::ConflictFilesStitcher.new(response)
|
||
|
|
end
|
||
|
|
|
||
|
|
def conflicts?
|
||
|
|
list_conflict_files.any?
|
||
|
|
rescue GRPC::FailedPrecondition
|
||
|
|
# The server raises this exception when it encounters ConflictSideMissing, which
|
||
|
|
# means a conflict exists but its `theirs` or `ours` data is nil due to a non-existent
|
||
|
|
# file in one of the trees.
|
||
|
|
true
|
||
|
|
end
|
||
|
|
|
||
|
|
def resolve_conflicts(target_repository, resolution, source_branch, target_branch)
|
||
|
|
reader = binary_stringio(resolution.files.to_json)
|
||
|
|
|
||
|
|
req_enum = Enumerator.new do |y|
|
||
|
|
header = resolve_conflicts_request_header(target_repository, resolution, source_branch, target_branch)
|
||
|
|
y.yield Gitaly::ResolveConflictsRequest.new(header: header)
|
||
|
|
|
||
|
|
until reader.eof?
|
||
|
|
chunk = reader.read(MAX_MSG_SIZE)
|
||
|
|
|
||
|
|
y.yield Gitaly::ResolveConflictsRequest.new(files_json: chunk)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
response = GitalyClient.call(@repository.storage, :conflicts_service, :resolve_conflicts, req_enum, remote_storage: target_repository.storage)
|
||
|
|
|
||
|
|
if response.resolution_error.present?
|
||
|
|
raise Gitlab::Git::Conflict::Resolver::ResolutionError, response.resolution_error
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
private
|
||
|
|
|
||
|
|
def resolve_conflicts_request_header(target_repository, resolution, source_branch, target_branch)
|
||
|
|
Gitaly::ResolveConflictsRequestHeader.new(
|
||
|
|
repository: @gitaly_repo,
|
||
|
|
our_commit_oid: @our_commit_oid,
|
||
|
|
target_repository: target_repository.gitaly_repository,
|
||
|
|
their_commit_oid: @their_commit_oid,
|
||
|
|
source_branch: source_branch,
|
||
|
|
target_branch: target_branch,
|
||
|
|
commit_message: resolution.commit_message,
|
||
|
|
user: Gitlab::Git::User.from_gitlab(resolution.user).to_gitaly
|
||
|
|
)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|