gitlab-org--gitlab-foss/lib/gitlab/gitaly_client.rb

53 lines
1.3 KiB
Ruby
Raw Normal View History

require 'gitaly'
module Gitlab
module GitalyClient
SERVER_VERSION_FILE = 'GITALY_SERVER_VERSION'.freeze
2017-03-24 13:22:42 -04:00
def self.configure_channel(storage, address)
@addresses ||= {}
@addresses[storage] = address
@channels ||= {}
@channels[storage] = new_channel(address)
end
def self.new_channel(address)
# 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(address, {}, :this_channel_is_insecure)
end
2017-03-24 13:22:42 -04:00
def self.get_channel(storage)
@channels[storage]
end
def self.get_address(storage)
@addresses[storage]
end
def self.enabled?
Gitlab.config.gitaly.enabled
end
2017-02-24 10:53:44 -05:00
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