lib/gitlab/health_checks/puma_check.rb


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

# frozen_string_literal: true

module Gitlab
  module HealthChecks
    # This check can only be run on Puma `master` process
    class PumaCheck
      extend SimpleAbstractCheck

      class << self
        private

        def metric_prefix
          'puma_check'
        end

        def successful?(result)
          result > 0
        end

        def check
          return unless Gitlab::Runtime.puma?

          stats = Puma.stats
          stats = Gitlab::Json.parse(stats)

          # If `workers` is missing this means that
          # Puma server is running in single mode
          stats.fetch('workers', 1)
        rescue NoMethodError
          # server is not ready
          0
        end
      end
    end
  end
end