50 lines
1.2 KiB
Ruby
50 lines
1.2 KiB
Ruby
require 'gitaly'
|
|
|
|
module Gitlab
|
|
module GitalyClient
|
|
SERVER_VERSION_FILE = 'GITALY_SERVER_VERSION'.freeze
|
|
|
|
def self.gitaly_address
|
|
if Gitlab.config.gitaly.socket_path
|
|
"unix://#{Gitlab.config.gitaly.socket_path}"
|
|
end
|
|
end
|
|
|
|
def self.channel
|
|
return @channel if defined?(@channel)
|
|
|
|
@channel =
|
|
if enabled?
|
|
# NOTE: Gitaly currently runs on a Unix socket, so permissions are
|
|
# handled using the file system and no additional authentication is
|
|
# required (therefore the :this_channel_is_insecure flag)
|
|
GRPC::Core::Channel.new(gitaly_address, {}, :this_channel_is_insecure)
|
|
else
|
|
nil
|
|
end
|
|
end
|
|
|
|
def self.enabled?
|
|
gitaly_address.present?
|
|
end
|
|
|
|
def self.feature_enabled?(feature)
|
|
enabled? && ENV["GITALY_#{feature.upcase}"] == '1'
|
|
end
|
|
|
|
def self.migrate(feature)
|
|
is_enabled = feature_enabled?(feature)
|
|
metric_name = feature.to_s
|
|
metric_name += "_gitaly" if is_enabled
|
|
|
|
Gitlab::Metrics.measure(metric_name) do
|
|
yield is_enabled
|
|
end
|
|
end
|
|
|
|
def self.expected_server_version
|
|
path = Rails.root.join(SERVER_VERSION_FILE)
|
|
path.read.chomp
|
|
end
|
|
end
|
|
end
|