Checksum calculation is handled by Gitaly when feature is enabled

1 files changed, 19 insertions, 4 deletions

diff --git a/lib/gitlab/git/checksum.rb b/lib/gitlab/git/checksum.rb
index 3ef0f0a8854..218651906bf 100644
--- a/lib/gitlab/git/checksum.rb
+++ b/lib/gitlab/git/checksum.rb
@@ -7,13 +7,14 @@ module Gitlab
 
       Failure = Class.new(StandardError)
 
-      attr_reader :path, :relative_path, :storage, :storage_path
+      attr_reader :path, :relative_path, :storage, :storage_path, :gl_repository
 
-      def initialize(storage, relative_path)
+      def initialize(storage, relative_path, gl_repository)
         @storage       = storage
         @storage_path  = Gitlab.config.repositories.storages[storage].legacy_disk_path
         @relative_path = "#{relative_path}.git"
         @path          = File.join(storage_path, @relative_path)
+        @gl_repository = gl_repository
       end
 
       def calculate
@@ -21,7 +22,13 @@ module Gitlab
           failure!(Gitlab::Git::Repository::NoRepository, 'No repository for such path')
         end
 
-        calculate_checksum_by_shelling_out
+        raw_repository.gitaly_migrate(:calculate_checksum) do |is_enabled|
+          if is_enabled
+            calculate_checksum_gitaly
+          else
+            calculate_checksum_by_shelling_out
+          end
+        end
       end
 
       private
@@ -30,6 +37,10 @@ module Gitlab
         raw_repository.exists?
       end
 
+      def calculate_checksum_gitaly
+        gitaly_repository_client.calculate_checksum
+      end
+
       def calculate_checksum_by_shelling_out
         args = %W(--git-dir=#{path} show-ref --heads --tags)
         output, status = run_git(args)
@@ -69,7 +80,11 @@ module Gitlab
       end
 
       def raw_repository
-        Gitlab::Git::Repository.new(storage, relative_path, nil)
+        @raw_repository ||= Gitlab::Git::Repository.new(storage, relative_path, gl_repository)
+      end
+
+      def gitaly_repository_client
+        raw_repository.gitaly_repository_client
       end
 
       def run_git(args)