58 lines
1.2 KiB
Ruby
58 lines
1.2 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Gitlab
|
|
module HealthChecks
|
|
class GitalyCheck
|
|
extend BaseAbstractCheck
|
|
|
|
METRIC_PREFIX = 'gitaly_health_check'
|
|
|
|
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(
|
|
name,
|
|
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
|