rails_health_checks 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 726057067f5f90f64123fe8aa2772bd94149b111f9bda7b51914fed99233f219
-  data.tar.gz: a7baa24da9b0a1b04aa446612a73b062fe3e4ef20ff9b1e8efe7c10e74cab065
+  metadata.gz: d0f595984abf8a7db84183766d61918bdf684642b6ea67b2075afbda8f4cbae6
+  data.tar.gz: ab35a0deb32d298eae8dadc04c39dbf8ca34b8e77018625ebad8ba81e1901f34
 SHA512:
-  metadata.gz: deb2906855bafc33635d8aeb3432fcced690f41cee05e8f3d7b288eec4908de97458291610dbb4b3c554e03b36ec85519e656fb38d67410517c0d6f056b0e81b
-  data.tar.gz: f0e14a473c6f6ad646733a170f0c83232393a776a0e4b5dee124330b428e9402ed780644d97fb800611405636b7c9306599fd7b3ebcc17e8c0a6a86eec23c5aa
+  metadata.gz: 2f2db831794c15dbf59c6763ed65a092e9bc56db1dc87e35d4565cccd63303cfe9555a7b36f738e6a861086c0362f8eb98459e99ea46f5838bfeb3862130ebf5
+  data.tar.gz: 3d5b057c32506961cf0416d37ea45c70f5218767961ddeed208b37887abd7781187fc1f480aff2e5fb522bfad8400e355fa745a04ec876a064e1aeb938e91c78

data/README.md

@@ -131,6 +131,9 @@ The block receives the `ActionDispatch::Request` object and must return a truthy
 | `:solid_queue` | Solid Queue DB connectivity; optional `config.solid_queue_job_count` threshold for pending jobs |
 | `:good_job` | GoodJob queue latency; optional `config.good_job_latency` (seconds) threshold for oldest pending job |
 | `:resque` | Resque Redis connectivity; optional `config.resque_queue_size` threshold for total queue depth |
+| `:disk` | Free disk bytes via `df`; optional `config.disk_warn_threshold` / `config.disk_critical_threshold` (bytes) and `config.disk_path` (default: `/`) |
+| `:memory` | Process RSS via `ps`; optional `config.memory_threshold` (bytes) reports `degraded` when exceeded |
+| `:http` | HTTP GET to `config.http_url`; reports `critical` if response code differs from `config.http_expected_status` (default: `200`) or a network error occurs |
 
 [↑ Back to top](#table-of-contents)
 

data/lib/rails_health_checks/check_registry.rb

@@ -10,7 +10,17 @@ module RailsHealthChecks
       sidekiq:     -> { Checks::SidekiqCheck.new(queue_size: RailsHealthChecks.configuration.sidekiq_queue_size) },
       solid_queue: -> { Checks::SolidQueueCheck.new(job_count: RailsHealthChecks.configuration.solid_queue_job_count) },
       good_job:    -> { Checks::GoodJobCheck.new(latency: RailsHealthChecks.configuration.good_job_latency) },
-      resque:      -> { Checks::ResqueCheck.new(queue_size: RailsHealthChecks.configuration.resque_queue_size) }
+      resque:      -> { Checks::ResqueCheck.new(queue_size: RailsHealthChecks.configuration.resque_queue_size) },
+      memory:      -> { Checks::MemoryCheck.new(threshold: RailsHealthChecks.configuration.memory_threshold) },
+      http:        -> { Checks::HttpCheck.new(
+        url:             RailsHealthChecks.configuration.http_url,
+        expected_status: RailsHealthChecks.configuration.http_expected_status
+      ) },
+      disk:        -> { Checks::DiskCheck.new(
+        warn_threshold:     RailsHealthChecks.configuration.disk_warn_threshold,
+        critical_threshold: RailsHealthChecks.configuration.disk_critical_threshold,
+        path:               RailsHealthChecks.configuration.disk_path
+      ) }
     }.freeze
 
     def self.build(check_names)

data/lib/rails_health_checks/checks/disk_check.rb

@@ -0,0 +1,42 @@
+# frozen_string_literal: true
+
+require "open3"
+
+module RailsHealthChecks
+  module Checks
+    class DiskCheck < Check
+      def initialize(warn_threshold: nil, critical_threshold: nil, path: "/")
+        @warn_threshold = warn_threshold
+        @critical_threshold = critical_threshold
+        @path = path
+      end
+
+      def call
+        free = nil
+        measure { free = free_bytes }
+
+        if @critical_threshold && free < @critical_threshold
+          return fail_with("free disk space #{free} bytes below critical threshold #{@critical_threshold} bytes")
+        end
+
+        if @warn_threshold && free < @warn_threshold
+          return warn_with("free disk space #{free} bytes below warn threshold #{@warn_threshold} bytes")
+        end
+
+        pass
+      rescue StandardError => e
+        fail_with(e.message)
+      end
+
+      private
+
+      def free_bytes
+        out, _, status = Open3.capture3("df", "-Pk", @path.to_s)
+        raise "df command failed with exit status #{status.exitstatus}" unless status.success?
+
+        parts = out.split("\n").last.split
+        parts[3].to_i * 1024
+      end
+    end
+  end
+end

data/lib/rails_health_checks/checks/http_check.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+require "net/http"
+require "uri"
+
+module RailsHealthChecks
+  module Checks
+    class HttpCheck < Check
+      def initialize(url:, expected_status: 200)
+        @url = url
+        @expected_status = expected_status
+      end
+
+      def call
+        measure do
+          response = Net::HTTP.get_response(URI.parse(@url))
+          code = response.code.to_i
+          return fail_with("HTTP GET #{@url} returned #{code}, expected #{@expected_status}") if code != @expected_status
+        end
+        pass
+      rescue StandardError => e
+        fail_with(e.message)
+      end
+    end
+  end
+end

data/lib/rails_health_checks/checks/memory_check.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+require "open3"
+
+module RailsHealthChecks
+  module Checks
+    class MemoryCheck < Check
+      def initialize(threshold: nil)
+        @threshold = threshold
+      end
+
+      def call
+        rss = nil
+        measure { rss = rss_bytes }
+
+        if @threshold && rss > @threshold
+          return warn_with("process RSS #{rss} bytes exceeds threshold #{@threshold} bytes")
+        end
+
+        pass
+      rescue StandardError => e
+        fail_with(e.message)
+      end
+
+      private
+
+      def rss_bytes
+        out, _, status = Open3.capture3("ps", "-o", "rss=", "-p", Process.pid.to_s)
+        raise "ps command failed with exit status #{status.exitstatus}" unless status.success?
+
+        out.strip.to_i * 1024
+      end
+    end
+  end
+end

data/lib/rails_health_checks/configuration.rb

@@ -3,7 +3,8 @@
 module RailsHealthChecks
   class Configuration
     attr_accessor :checks, :timeout, :allowed_ips, :token, :sidekiq_queue_size, :solid_queue_job_count, :good_job_latency,
-                  :resque_queue_size
+                  :resque_queue_size, :disk_warn_threshold, :disk_critical_threshold, :disk_path,
+                  :memory_threshold, :http_url, :http_expected_status
     attr_reader :authenticate_block
 
     def initialize
@@ -16,6 +17,12 @@ module RailsHealthChecks
       @solid_queue_job_count = nil
       @good_job_latency = nil
       @resque_queue_size = nil
+      @disk_warn_threshold = nil
+      @disk_critical_threshold = nil
+      @disk_path = "/"
+      @memory_threshold = nil
+      @http_url = nil
+      @http_expected_status = 200
     end
 
     def authenticate(&block)

data/lib/rails_health_checks/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RailsHealthChecks
-  VERSION = "0.3.0"
+  VERSION = "0.4.0"
 end

data/lib/rails_health_checks.rb

@@ -11,6 +11,9 @@ require "rails_health_checks/checks/sidekiq_check"
 require "rails_health_checks/checks/solid_queue_check"
 require "rails_health_checks/checks/good_job_check"
 require "rails_health_checks/checks/resque_check"
+require "rails_health_checks/checks/disk_check"
+require "rails_health_checks/checks/memory_check"
+require "rails_health_checks/checks/http_check"
 require "rails_health_checks/check_registry"
 require "rails_health_checks/response_builder"
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails_health_checks
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Chuck Smith
@@ -46,7 +46,10 @@ files:
 - lib/rails_health_checks/check_registry.rb
 - lib/rails_health_checks/checks/cache_check.rb
 - lib/rails_health_checks/checks/database_check.rb
+- lib/rails_health_checks/checks/disk_check.rb
 - lib/rails_health_checks/checks/good_job_check.rb
+- lib/rails_health_checks/checks/http_check.rb
+- lib/rails_health_checks/checks/memory_check.rb
 - lib/rails_health_checks/checks/resque_check.rb
 - lib/rails_health_checks/checks/sidekiq_check.rb
 - lib/rails_health_checks/checks/solid_queue_check.rb