51 lines
1.2 KiB
Ruby
51 lines
1.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module HealthChecks
|
|
class GitalyCheck
|
|
extend BaseAbstractCheck
|
|
|
|
METRIC_PREFIX = 'gitaly_health_check'.freeze
|
|
|
|
class << self
|
|
def readiness
|
|
repository_storages.map do |storage_name|
|
|
check(storage_name)
|
|
end
|
|
end
|
|
|
|
def metrics
|
|
Gitaly::Server.all.flat_map do |server|
|
|
result, elapsed = with_timing { server.read_writeable? }
|
|
labels = { shard: server.storage }
|
|
|
|
[
|
|
metric("#{metric_prefix}_success", result ? 1 : 0, **labels),
|
|
metric("#{metric_prefix}_latency_seconds", elapsed, **labels)
|
|
]
|
|
end
|
|
end
|
|
|
|
def check(storage_name)
|
|
serv = Gitlab::GitalyClient::HealthCheckService.new(storage_name)
|
|
result = serv.check
|
|
HealthChecks::Result.new(result[:success], result[:message], shard: storage_name)
|
|
end
|
|
|
|
private
|
|
|
|
def metric_prefix
|
|
METRIC_PREFIX
|
|
end
|
|
|
|
def repository_storages
|
|
storages.keys
|
|
end
|
|
|
|
def storages
|
|
Gitlab.config.repositories.storages
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|