Commit 6d46e510 authored by Alejandro Rodríguez's avatar Alejandro Rodríguez

Use gitaly_migrate helper on all current migrations in repository

parent e7e13252
...@@ -114,7 +114,7 @@ module Gitlab ...@@ -114,7 +114,7 @@ module Gitlab
# Returns the number of valid branches # Returns the number of valid branches
def branch_count def branch_count
Gitlab::GitalyClient.migrate(:branch_names) do |is_enabled| gitaly_migrate(:branch_names) do |is_enabled|
if is_enabled if is_enabled
gitaly_ref_client.count_branch_names gitaly_ref_client.count_branch_names
else else
...@@ -133,7 +133,7 @@ module Gitlab ...@@ -133,7 +133,7 @@ module Gitlab
# Returns the number of valid tags # Returns the number of valid tags
def tag_count def tag_count
Gitlab::GitalyClient.migrate(:tag_names) do |is_enabled| gitaly_migrate(:tag_names) do |is_enabled|
if is_enabled if is_enabled
gitaly_ref_client.count_tag_names gitaly_ref_client.count_tag_names
else else
...@@ -471,7 +471,7 @@ module Gitlab ...@@ -471,7 +471,7 @@ module Gitlab
def ref_name_for_sha(ref_path, sha) def ref_name_for_sha(ref_path, sha)
# NOTE: This feature is intentionally disabled until # NOTE: This feature is intentionally disabled until
# https://gitlab.com/gitlab-org/gitaly/issues/180 is resolved # https://gitlab.com/gitlab-org/gitaly/issues/180 is resolved
# Gitlab::GitalyClient.migrate(:find_ref_name) do |is_enabled| # gitaly_migrate(:find_ref_name) do |is_enabled|
# if is_enabled # if is_enabled
# gitaly_ref_client.find_ref_name(sha, ref_path) # gitaly_ref_client.find_ref_name(sha, ref_path)
# else # else
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment