54 lines
1.3 KiB
Ruby
54 lines
1.3 KiB
Ruby
|
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
|
||
|
|
repository_storages.flat_map do |storage_name|
|
||
|
|
result, elapsed = with_timing { check(storage_name) }
|
||
|
|
labels = { shard: storage_name }
|
||
|
|
|
||
|
|
[
|
||
|
|
metric("#{metric_prefix}_success", successful?(result) ? 1 : 0, **labels),
|
||
|
|
metric("#{metric_prefix}_latency_seconds", elapsed, **labels)
|
||
|
|
].flatten
|
||
|
|
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 successful?(result)
|
||
|
|
result[:success]
|
||
|
|
end
|
||
|
|
|
||
|
|
def repository_storages
|
||
|
|
storages.keys
|
||
|
|
end
|
||
|
|
|
||
|
|
def storages
|
||
|
|
Gitlab.config.repositories.storages
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|