debian-mirror-gitlab/lib/gitlab/health_checks/probes/collection.rb

60 lines
1.3 KiB
Ruby
Raw Normal View History

2019-12-21 20:55:43 +05:30
# frozen_string_literal: true
module Gitlab
module HealthChecks
module Probes
class Collection
attr_reader :checks
# This accepts an array of objects implementing `:readiness`
# that returns `::Gitlab::HealthChecks::Result`
def initialize(*checks)
@checks = checks
end
def execute
readiness = probe_readiness
success = all_succeeded?(readiness)
Probes::Status.new(
success ? 200 : 503,
status(success).merge(payload(readiness))
)
2021-06-08 01:23:25 +05:30
rescue StandardError => e
2020-07-28 23:09:34 +05:30
exception_payload = { message: "#{e.class} : #{e.message}" }
Probes::Status.new(
500,
status(false).merge(exception_payload))
2019-12-21 20:55:43 +05:30
end
private
def all_succeeded?(readiness)
readiness.all? do |name, probes|
probes.any?(&:success)
end
end
def status(success)
{ status: success ? 'ok' : 'failed' }
end
def payload(readiness)
readiness.transform_values do |probes|
probes.map(&:payload)
end
end
def probe_readiness
checks
2021-03-11 19:13:27 +05:30
.select(&:available?)
2019-12-21 20:55:43 +05:30
.flat_map(&:readiness)
.compact
.group_by(&:name)
end
end
end
end
end