module Gitlab module GitalyClient class RemoteService MAX_MSG_SIZE = 128.kilobytes.freeze def initialize(repository) @repository = repository @gitaly_repo = repository.gitaly_repository @storage = repository.storage end def add_remote(name, url, mirror_refmaps) request = Gitaly::AddRemoteRequest.new( repository: @gitaly_repo, name: name, url: url, mirror_refmaps: Array.wrap(mirror_refmaps).map(&:to_s) ) GitalyClient.call(@storage, :remote_service, :add_remote, request) end def remove_remote(name) request = Gitaly::RemoveRemoteRequest.new(repository: @gitaly_repo, name: name) response = GitalyClient.call(@storage, :remote_service, :remove_remote, request) response.result end def fetch_internal_remote(repository) request = Gitaly::FetchInternalRemoteRequest.new( repository: @gitaly_repo, remote_repository: repository.gitaly_repository ) response = GitalyClient.call(@storage, :remote_service, :fetch_internal_remote, request, remote_storage: repository.storage) response.result end def update_remote_mirror(ref_name, only_branches_matching) req_enum = Enumerator.new do |y| y.yield Gitaly::UpdateRemoteMirrorRequest.new( repository: @gitaly_repo, ref_name: ref_name ) current_size = 0 slices = only_branches_matching.slice_before do |branch_name| current_size += branch_name.bytesize next false if current_size < MAX_MSG_SIZE current_size = 0 end slices.each do |slice| y.yield Gitaly::UpdateRemoteMirrorRequest.new(only_branches_matching: slice) end end GitalyClient.call(@storage, :remote_service, :update_remote_mirror, req_enum) end end end end