BigW Consortium Gitlab

db_check.rb 626 Bytes
Newer Older
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
module Gitlab
  module HealthChecks
    class DbCheck
      extend SimpleAbstractCheck

      class << self
        private

        def metric_prefix
          'db_ping'
        end

        def is_successful?(result)
          result == '1'
        end

        def check
          catch_timeout 10.seconds do
            if Gitlab::Database.postgresql?
              ActiveRecord::Base.connection.execute('SELECT 1 as ping')&.first&.[]('ping')
            else
              ActiveRecord::Base.connection.execute('SELECT 1 as ping')&.first&.first&.to_s
            end
          end
        end
      end
    end
  end
end