kubernetes-health 3.0.5 → 3.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 175ef925d5c1959ffa37122e4f2f978e137bface0ff4113c286a480560deb3b0
-  data.tar.gz: '02854ee2fffaa4c0384198e2414e3c6124a8de05b6915e383ea980f17875b4f0'
+  metadata.gz: 59bbdf0e963539df25aa7a994a4fcc54a653506d76df2fa5aa6bbae70994ee98
+  data.tar.gz: b783c8b8df00a8304b5f853b0c13ce9fe09bf8972b55c906d15f42a23c23c5b0
 SHA512:
-  metadata.gz: f61536178ae2d69cb5f55ce4ce8931dd516141b3293203888863f2dd7333f28a9c60c769ce546d0cec2d5cf6c6e7d59ca128fed8ace987cc9a94b8de280d8779
-  data.tar.gz: c7fa2329ae05e4e5a94050529e6aeaabf47653c744522b41297d1e6aa7703c5b80984f290911d61d0dc8ed1a77ee90b9756ec1a9a14cb57a12958d2d9d88a8b1
+  metadata.gz: 103a26a9a2224f13e246fe1dcc2544990f39a1944f2c989626e0cf44a903a84b2a2fc88b293e1a6ce4269498ccf9abd988a55c4a2ebe7282cbd209371bebd5a2
+  data.tar.gz: 3cd7707bd895b10acb41c68f9d574320599fe5ac1e01fb29f30e9afe2c2a26b382443e2f1db9f0ef4b9dabe54a8641f32920b38075a8151e0405a67ed815e310

data/README.md

@@ -1,11 +1,11 @@
 # Kubernetes::Health
 
-This gem allows kubernetes monitoring your app while it is running migrates and after it started.
+This gem allows kubernetes monitoring your app while it is running migrates and after it starts.
 
 # Features
 - add routes `/_readiness`, `/_liveness` on rails stack.
 - add routes `/_readiness`, `/_liveness` and `/_metrics` as a puma plugin.
-- metrics are prometheus compatible (code copied from `puma-metrics` gem).
+- metrics are prometheus compatible (code copied from `puma-metrics` gem) or json.
 - allow custom checks for `/_readiness` and `/_liveness`.
 - add routes `/_readiness` and `/_liveness` while `rake db:migrate` runs. (optional)
 - add support to avoid parallel running of `rake db:migrate` while keep kubernetes waiting. (optional)
@@ -15,7 +15,7 @@ This gem allows kubernetes monitoring your app while it is running migrates and
 Add this line to your application's Gemfile:
 
 ```ruby
-gem 'kubernetes-health', '~> 3.0'
+gem 'kubernetes-health', '~> 3.3'
 ```
 
 ## Enabling puma plugin
@@ -48,18 +48,6 @@ In Kubernetes you need to configure your deployment `readinessProbe` and `livene
 ```
 
 Setting `failureThreshold` is import to avoid problems when app finish migrates and is starting the web process.
-
-## Enabling prometheus metrics
-
-```
-  template:
-    metadata:
-      annotations:
-        prometheus.io/path: '/_metrics'
-        prometheus.io/port: '9393'
-        prometheus.io/scrape: 'true'
-```
-
 ## Enabling monitoring while `rake db:migrate` runs
 
 Your Dockerfile's entry script needs to run migrates before start your web app.
@@ -106,18 +94,19 @@ Kubernetes::Health::Config.unlock = lambda {
 
 It only works for routes in rails stack, they are not executed while `rake db:migrate` runs.
 
-I prefer do nothing else on `liveness` to avoid unnecessary `CrashLoopBackOff` status. `params` is optional.
+I prefer do nothing else on `liveness` to avoid unnecessary `CrashLoopBackOff` status. `params` is optional (request params).
 
 ```
-Kubernetes::Health::Config.live_if = lambda {
+Kubernetes::Health::Config.live_if = lambda { |params|
   true
 }
 
 ```
-Ex. Check if PostgreSQL is reachable on `readiness`. `params` is optional.
+Ex. Check if PostgreSQL is reachable on `readiness` indicating that credentials are setup right and keeps cache to avoid doing it a lot. `params` is optional (request params).
 ```
 Kubernetes::Health::Config.ready_if = lambda { |params|
-  ActiveRecord::Base.connection.execute("SELECT 1").cmd_tuples != 1
+  return $kubernetes_health_test_db_connection if $kubernetes_health_test_db_connection
+  $kubernetes_health_test_db_connection = ActiveRecord::Base.connection.execute("SELECT 1").cmd_tuples == 1
 }
 ```
 
@@ -127,3 +116,19 @@ Kubernetes::Health::Config.route_liveness = '/liveness'
 Kubernetes::Health::Config.route_readiness = '/readiness'
 Kubernetes::Health::Config.route_metrics = '/metrics'
 ```
+
+## Response format
+If you are using `https://github.com/zalando-incubator/kube-metrics-adapter` you will want to use `json` format.
+
+Default is `prometheus`.
+```
+Kubernetes::Health::Config.response_format = 'json'
+```
+
+## Customizing requests logs
+
+```
+Kubernetes::Health::Config.request_log_callback = lambda { |req, http_code|
+  Rails.logger.debug "Kubernetes Health: Rack on Migrate - Request: Path: #{req.path_info} / Params: #{req.params} /  HTTP Code: #{http_code}"  rescue nil
+}
+```

data/kubernetes-health.gemspec

@@ -26,11 +26,11 @@ Gem::Specification.new do |spec|
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ["lib"]
 
-  spec.add_development_dependency "bundler", "~> 1.14"
-  spec.add_development_dependency "rake", "~> 10.0"
-  spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_development_dependency "bundler"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec"
   spec.add_dependency "rack"
   spec.add_dependency "rails"
   spec.add_runtime_dependency 'prometheus-client', '~> 0.9.0'
-  spec.add_runtime_dependency 'puma', '~> 3.0'
+  spec.add_runtime_dependency 'puma'
 end

data/lib/kubernetes/health/config.rb

@@ -8,6 +8,12 @@ module Kubernetes
       @@route_liveness = '/_liveness'
       @@route_readiness = '/_readiness'
       @@route_metrics = '/_metrics'
+      @@response_format = 'prometheus'
+
+      @@request_log_callback = lambda { |req, http_code, content|
+        Rails.logger.debug "Kubernetes Health: Rack on Migrate - Request: Path: #{req.path_info} / Params: #{req.params} /  HTTP Code: #{http_code}\n#{content}"  rescue nil
+      }
+
       @@lock_or_wait = lambda { ActiveRecord::Base.connection.execute 'select pg_advisory_lock(123456789123456789);' }
       @@unlock = lambda { ActiveRecord::Base.connection.execute 'select pg_advisory_unlock(123456789123456789);' }
 
@@ -19,6 +25,14 @@ module Kubernetes
         @@lock_or_wait = value
       end
 
+      def self.request_log_callback
+        @@request_log_callback
+      end
+
+      def self.request_log_callback=(value)
+        @@request_log_callback = value
+      end
+
       def self.unlock
         @@unlock
       end
@@ -83,6 +97,13 @@ module Kubernetes
         @@ready_if = value
       end
 
+      def self.response_format
+        @@response_format
+      end
+
+      def self.response_format=(value)
+        @@response_format = value
+      end
     end
   end
 end

data/lib/kubernetes/health/rack_on_migrate.rb

@@ -11,7 +11,7 @@ module Kubernetes
         else
           http_code = 404
         end
-        Rails.logger.debug "Kubernetes Health: Rack on Migrate - Request: Path: #{req.path_info} / Params: #{req.params} /  HTTP Code: #{http_code}"
+        ::Kubernetes::Health::Config.request_log_callback.call(req, http_code)
         [http_code, {}, []]
       end
     end

data/lib/kubernetes/health/version.rb

@@ -1,5 +1,5 @@
 module Kubernetes
   module Health
-    VERSION = "3.0.5"
+    VERSION = "3.3.1"
   end
 end

data/lib/puma/kubernetes/app.rb

@@ -13,26 +13,31 @@ module Puma
       end
 
       def call(_env)
-        req = ::Rack::Request.new(_env)
-        type = {}
-        content = []
-        case req.path_info
-        when ::Kubernetes::Health::Config.route_liveness
-          i_am_live = ::Kubernetes::Health::Config.live_if.arity == 0 ? ::Kubernetes::Health::Config.live_if.call : ::Kubernetes::Health::Config.live_if.call(req.params)
-          http_code = i_am_live ? 200 : 503
-        when ::Kubernetes::Health::Config.route_readiness
-          i_am_ready = ::Kubernetes::Health::Config.ready_if.arity == 0 ? ::Kubernetes::Health::Config.ready_if.call : ::Kubernetes::Health::Config.ready_if.call(req.params)
-          http_code = i_am_ready ? 200 : 503
-        when ::Kubernetes::Health::Config.route_metrics
-          http_code = 200
-          @parser.parse JSON.parse(@launcher.stats)
-          type = { 'Content-Type' => 'text/plain' }
-          content = [Prometheus::Client::Formats::Text.marshal(Prometheus::Client.registry)]
-        else
-          http_code = 404
+        begin
+          req = ::Rack::Request.new(_env)
+          type = {}
+          content = []
+          type = ::Kubernetes::Health::Config.response_format == 'json' ? { 'Content-Type' => 'application/json' } : { 'Content-Type' => 'text/plain' }
+          case req.path_info
+          when ::Kubernetes::Health::Config.route_liveness
+            i_am_live = ::Kubernetes::Health::Config.live_if.arity == 0 ? ::Kubernetes::Health::Config.live_if.call : ::Kubernetes::Health::Config.live_if.call(req.params)
+            http_code = i_am_live ? 200 : 503
+          when ::Kubernetes::Health::Config.route_readiness
+            i_am_ready = ::Kubernetes::Health::Config.ready_if.arity == 0 ? ::Kubernetes::Health::Config.ready_if.call : ::Kubernetes::Health::Config.ready_if.call(req.params)
+            http_code = i_am_ready ? 200 : 503
+          when ::Kubernetes::Health::Config.route_metrics
+            http_code = 200
+            @parser.parse JSON.parse(@launcher.stats)
+            content = ::Kubernetes::Health::Config.response_format == 'json' ? @launcher.stats : Prometheus::Client::Formats::Text.marshal(Prometheus::Client.registry)
+          else
+            http_code = 404
+          end
+        rescue
+          http_code = 500
+          content = []
         end
-        Rails.logger.debug "Kubernetes Health: Puma Plugin - Request: Path: #{req.path_info} / Params: #{req.params} /  HTTP Code: #{http_code}" rescue nil
-        [http_code, type, content]
+        ::Kubernetes::Health::Config.request_log_callback.call(req, http_code, content)
+        [http_code, type, [content]]
       end
     end
   end

metadata

@@ -1,57 +1,57 @@
 --- !ruby/object:Gem::Specification
 name: kubernetes-health
 version: !ruby/object:Gem::Version
-  version: 3.0.5
+  version: 3.3.1
 platform: ruby
 authors:
 - Wagner Caixeta
-autorequire: 
+autorequire:
 bindir: exe
 cert_chain: []
-date: 2019-12-11 00:00:00.000000000 Z
+date: 2021-07-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '1.14'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '1.14'
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '10.0'
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rack
   requirement: !ruby/object:Gem::Requirement
@@ -98,16 +98,16 @@ dependencies:
   name: puma
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: '3.0'
+        version: '0'
 description: "\n    This gem allows kubernetes monitoring your app while it is running
   migrates and after it started.\n    Features:\n    * add routes /_readiness and
   /_liveness on rails stack by default;\n    * allow custom checks for /_readiness
@@ -145,7 +145,7 @@ files:
 homepage: https://github.com/platbr/kubernetes-health
 licenses: []
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -160,8 +160,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
-signing_key: 
+rubygems_version: 3.2.17
+signing_key:
 specification_version: 4
 summary: This gem allows kubernetes monitoring your app while it is running migrates
   and after it started.