Welcome to mirror list, hosted at ThFree Co, Russian Federation.

object_pool_service.rb « gitaly_client « gitlab « lib - gitlab.com/gitlab-org/gitlab-foss.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: e07bf3fbccc58ba3e7b794275611871d4de59ed1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
# frozen_string_literal: true

module Gitlab
  module GitalyClient
    class ObjectPoolService
      include WithFeatureFlagActors

      attr_reader :object_pool, :storage

      def initialize(object_pool)
        @object_pool = object_pool.gitaly_object_pool
        @storage = object_pool.storage
      end

      def create(repository)
        request = Gitaly::CreateObjectPoolRequest.new(
          object_pool: object_pool,
          origin: repository.gitaly_repository)

        GitalyClient.with_feature_flag_actors(**gitaly_feature_flag_actors(repository)) do
          GitalyClient.call(storage, :object_pool_service, :create_object_pool,
                            request, timeout: GitalyClient.medium_timeout)
        end
      end

      def delete
        request = Gitaly::DeleteObjectPoolRequest.new(object_pool: object_pool)

        GitalyClient.call(storage, :object_pool_service, :delete_object_pool,
                          request, timeout: GitalyClient.long_timeout)
      end

      def link_repository(repository)
        request = Gitaly::LinkRepositoryToObjectPoolRequest.new(
          object_pool: object_pool,
          repository: repository.gitaly_repository
        )

        GitalyClient.with_feature_flag_actors(**gitaly_feature_flag_actors(repository)) do
          GitalyClient.call(storage, :object_pool_service, :link_repository_to_object_pool,
                            request, timeout: GitalyClient.fast_timeout)
        end
      end

      def fetch(repository)
        request = Gitaly::FetchIntoObjectPoolRequest.new(
          object_pool: object_pool,
          origin: repository.gitaly_repository
        )

        GitalyClient.with_feature_flag_actors(**gitaly_feature_flag_actors(repository)) do
          GitalyClient.call(storage, :object_pool_service, :fetch_into_object_pool,
                            request, timeout: GitalyClient.long_timeout)
        end
      end
    end
  end
end