blob: 2bd8ea711b5188f85fed9e639fc9bdfd4b165a83 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
|
# 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)
storage_healthy = healthy(storage_name)
unless storage_healthy[:success]
return HealthChecks::Result.new(
name,
storage_healthy[:success],
storage_healthy[:message],
shard: storage_name
)
end
storage_ready = ready(storage_name)
HealthChecks::Result.new(
name,
storage_ready[:success],
storage_ready[:message],
shard: storage_name
)
end
def healthy(storage_name)
serv = Gitlab::GitalyClient::HealthCheckService.new(storage_name)
serv.check
end
def ready(storage_name)
serv = Gitlab::GitalyClient::ServerService.new(storage_name)
serv.readiness_check
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
|