health-monitor-rails 12.5.0 → 12.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c16cdc04971eb81463bb19da409b3e1d41cf83683840819882ad33a6d8cdf4b0
-  data.tar.gz: 8091013b1770de119a9d9448696f97c3ec7baadafac9ce4173d6c279b65bc313
+  metadata.gz: 9e05b368505661b4b4a0849035c3449309b6905f92651e2e41701f6651bd36e4
+  data.tar.gz: 57fab8c81135598a106a2f07e35956db9bcf3eb2c7bcba91d826a7af5f20b167
 SHA512:
-  metadata.gz: 2d6f3818a0dabb60c514382d32815832e94f516ccd3d76751acf90889b61e90edfcfd62b4df44e34669909ed475e2c25585b4c08b5ebc0cfebea33f8f8c965a4
-  data.tar.gz: 7f519417d7e0adee2f367f776c0d193bc44b0cb8dd48661d5e0067171618efced5e0834ba26a66b51e6b0335a9c8ee81f4d483e1d01ae51a3951a4f872200d36
+  metadata.gz: b0f62bf00240926b34f102b96670513caed11686d57a475c0fca7d149da3752bb3331d5a621eccbeea8d233758ec9cc25a359ad537e7399b5a921b1411311529
+  data.tar.gz: 148f73504ac45c215629a1761a844b29eef80cac3d72e212bd66caba59a6fddd4d1be6c81061c7e0bbbc22447b1ecaefe01896107cbf91f9c1fef88c27c48242

data/README.md

@@ -202,6 +202,21 @@ HealthMonitor.configure do |config|
 end
 ```
 
+When you have multiple databases and and want to check each configuration separately you can use following method:
+```ruby
+HealthMonitor.configure do |config|
+  config.no_database # Disable default all databases check
+  config.database.configure do |provider_config|
+    provider_config.config_name = 'primary'
+  end
+  config.database.configure do |provider_config|
+    provider_config.name = 'Secondary'
+    provider_config.config_name = 'secondary'
+    provider_config.critical = false
+  end
+end
+```
+
 ### Provider Configuration
 
 All providers accept a general set of baseline configuration:
@@ -322,7 +337,7 @@ Please note that `url` or `connection` can't be used at the same time.
 
 This check allows you to create a file on your server when you would like to force the check to fail. For example if you are utilizing the `health.json` as you health check page for your load balancer and would like to force a machine offline.
 
-* `filename`: the file relative to the rails root that must remain absent for the health check to remain passing. For example: `public/remove-from-nginx`
+* `filename`: the file relative to the rails root that must remain absent for the health check to remain passing. For example: `public/remove-from-nginx` (Can also be a full path `/opt/app/remove-from-nginx`)
 
 ### Adding a Custom Provider
 

data/lib/health_monitor/providers/database.rb

@@ -7,23 +7,58 @@ module HealthMonitor
     class DatabaseException < StandardError; end
 
     class Database < Base
+      class Configuration < Base::Configuration
+        DEFAULT_CONFIG_NAME = nil
+
+        attr_reader :config_name
+
+        def initialize(provider)
+          super
+
+          @config_name = DEFAULT_CONFIG_NAME
+        end
+
+        def config_name=(value)
+          @config_name = value.presence&.to_s
+        end
+      end
+
       def check!
+        checked = false
         failed_databases = []
 
         ActiveRecord::Base.connection_handler.connection_pool_list(:all).each do |cp|
-          if cp.respond_to? :lease_connection
-            cp.lease_connection.execute('SELECT 1')
-          else
-            cp.connection.execute('SELECT 1')
-          end
+          next unless check_connection_pool?(cp)
+
+          checked = true
+          check_connection(cp)
         rescue Exception
           failed_databases << cp.db_config.name
         end
 
         raise "unable to connect to: #{failed_databases.uniq.join(',')}" unless failed_databases.empty?
+        raise "no connections checked with name: #{configuration.config_name}" if configuration.config_name && !checked
       rescue Exception => e
         raise DatabaseException.new(e.message)
       end
+
+      private
+
+      def configuration_class
+        ::HealthMonitor::Providers::Database::Configuration
+      end
+
+      def check_connection_pool?(connection_pool)
+        configuration.config_name.nil? || configuration.config_name == connection_pool.db_config.name
+      end
+
+      def check_connection(connection_pool)
+        if connection_pool.respond_to?(:lease_connection)
+          connection_pool.lease_connection.execute('SELECT 1')
+        else
+          connection_pool.connection.execute('SELECT 1')
+        end
+      end
     end
   end
 end

data/lib/health_monitor/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module HealthMonitor
-  VERSION = '12.5.0'
+  VERSION = '12.6.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: health-monitor-rails
 version: !ruby/object:Gem::Version
-  version: 12.5.0
+  version: 12.6.0
 platform: ruby
 authors:
 - Leonid Beder
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-12-15 00:00:00.000000000 Z
+date: 2025-03-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: railties