remote_service.rb 2.49 KB
Newer Older
1 2 3
module Gitlab
  module GitalyClient
    class RemoteService
4 5
      MAX_MSG_SIZE = 128.kilobytes.freeze

6 7 8 9 10 11 12 13 14 15 16
      def self.exists?(remote_url)
        request = Gitaly::FindRemoteRepositoryRequest.new(remote: remote_url)

        response = GitalyClient.call(GitalyClient.random_storage,
                                     :remote_service,
                                     :find_remote_repository, request,
                                     timeout: GitalyClient.medium_timeout)

        response.exists
      end

17 18 19 20 21 22
      def initialize(repository)
        @repository = repository
        @gitaly_repo = repository.gitaly_repository
        @storage = repository.storage
      end

23
      def add_remote(name, url, mirror_refmaps)
24
        request = Gitaly::AddRemoteRequest.new(
25 26 27 28
          repository: @gitaly_repo,
          name: name,
          url: url,
          mirror_refmaps: Array.wrap(mirror_refmaps).map(&:to_s)
29 30
        )

31
        GitalyClient.call(@storage, :remote_service, :add_remote, request, timeout: GitalyClient.fast_timeout)
32 33 34 35 36
      end

      def remove_remote(name)
        request = Gitaly::RemoveRemoteRequest.new(repository: @gitaly_repo, name: name)

37
        response = GitalyClient.call(@storage, :remote_service, :remove_remote, request, timeout: GitalyClient.fast_timeout)
38 39 40

        response.result
      end
41 42 43 44 45 46 47 48 49 50 51 52 53

      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
54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78

      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
79 80 81
    end
  end
end