ready-for-migration 0.1.1 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '0594c87355e8056fb15067e0aab39c234d56f603e10377f86fde0b65d48e3fdf'
-  data.tar.gz: c140de4c6b325e2655bffb64abdc3fc3b6eb95cc85ae49053679275baad3e78c
+  metadata.gz: f855a777fcbf390d9e7df10b5a2a8b0002623c9b9985a254f7d79f2d25f93990
+  data.tar.gz: '052758f3910e9864972b96ec3d7d46ea637416a57ca4fad205f91efda59c7709'
 SHA512:
-  metadata.gz: affa9b3b0aa6def18e31f46876232f96b6497b7c48528fa13f9059950a1a063606ba053d6996da3beec1e2cf51743ae72cdcbb7ee571787655a740229a9ae6fb
-  data.tar.gz: 3fee0628bb2f9192fb508986dfcb346c519e3ab08b989cbca07442fc0285366c8ba4d69f8b7ede426f404fe45de6d9330f527ba0a13f70a1dca8d9b854a16ecd
+  metadata.gz: 11036a04db24b448d2d06efda25638cbd605e764be39cd12245717e3914ff12c10f92524cc6bde17d6ec5a33f1a9be30c47d3c57a249395f93283fccd983be27
+  data.tar.gz: 5bf2c1b2beae4df61c6e0f59b007035c9af90fa35ebd2648d6e2c0d5f8eb41e7252e38771df396e2f350dce3c419315ad330a5cef64f594ca2f9c7e9521456c4

data/README.md

@@ -29,5 +29,23 @@ Or install it yourself as:
 $ gem install ready-for-migration
 ```
 
+## Simulate Health Check Response
+
+You can simulate health check response by adding specific query parameters to your health check requests.
+
+Here's the pattern we're providing. (values are just an example)
+
+|pattern|behavior|
+|--|--|
+|No query parameters|health check as usual.|
+|`?status=503`|respond with specified status code.|
+|`?status=503&after=60`|respond with specified status code once specified seconds have passed since the first access with `after`.|
+|`?status=503&sleep=60`|respond with specified status code after sleeping specified seconds.|
+|`?status=503&random=10` |randomly respond with specified status code, once in a specified number as `random`.|
+|`?sleep=60` |health check after sleeping specified seconds.|
+
+NOTE: Other patterns than listed in the above example will be ignored.  
+NOTE: "sleep" parameter is disabled in default for avoiding DoS attack. To enabling this, you set environment variables "RFM_ENABLE_SLEEP" as "1".
+
 ## License
 The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/app/controllers/concerns/ready/for/migration/health_action_inspectable.rb

@@ -0,0 +1,112 @@
+require 'active_support/cache/file_store'
+
+module Ready
+  module For
+    module Migration
+      module HealthActionInspectable
+
+        def self.cache
+          @cache ||= ActiveSupport::Cache::MemoryStore.new
+        end
+
+        def self.sleep_enabled?
+          return false if ENV['RFM_ENABLE_SLEEP'].nil?
+          b = [ false, 0,
+                "0", :"0",
+                "f", :f,
+                "F", :F,
+                "false", :false,
+                "FALSE", :FALSE,
+                "off", :off,
+                "OFF", :OFF,
+              ].include?(ENV['RFM_ENABLE_SLEEP'])
+          return !b
+        end
+
+        def readiness
+          if params.has_key?(:status)
+            case
+            when params.has_key?(:after)
+              perform_inspection_with_status_and_after
+            when params.has_key?(:sleep)
+              perform_inspection_with_status_and_sleep
+            when params.has_key?(:random)
+              perform_inspection_with_status_and_random
+            else
+              perform_inspection_with_status
+            end
+          elsif params.has_key?(:sleep)
+            perform_inspection_with_sleep
+          end
+
+          super unless performed?
+        end
+
+        private
+
+          def get_status_value
+            raw = params.dig(:status)
+            status = raw.to_i if raw && /^\d{3}$/.match?(raw)
+            status
+          end
+
+          def get_after_value
+            raw = params.dig(:after)
+            value = raw.to_i if raw && /^\d+$/.match?(raw)
+            value
+          end
+
+          def get_random_value
+            raw = params.dig(:random)
+            value = raw.to_i if raw && /^\d+$/.match?(raw)
+            value
+          end
+
+          def get_sleep_value
+            raw = params.dig(:sleep)
+            value = raw.to_i if raw && /^\d+$/.match?(raw)
+            value
+          end
+
+          def perform_inspection_with_status_and_after
+            return unless status = get_status_value
+            return unless value = get_after_value
+
+            first_access_time = Ready::For::Migration::HealthActionInspectable.cache.fetch(:first_access_time) { Time.now }
+            if first_access_time + value < Time.now
+              head status
+            end
+          end
+
+          def perform_inspection_with_status_and_sleep
+            return unless status = get_status_value
+            return unless value = get_sleep_value
+
+            sleep(value) if Ready::For::Migration::HealthActionInspectable.sleep_enabled?
+            head status
+          end
+
+          def perform_inspection_with_status_and_random
+            return unless status = get_status_value
+            return unless value = get_random_value
+
+            if value == 1 || rand(value - 1) == 0
+              head status
+            end
+          end
+
+          def perform_inspection_with_status
+            return unless status = get_status_value
+            head status
+          end
+
+          def perform_inspection_with_sleep
+            return unless value = get_sleep_value
+
+            sleep(value) if Ready::For::Migration::HealthActionInspectable.sleep_enabled?
+          end
+
+      end
+    end
+  end
+end

data/app/controllers/ready/for/migration/health_controller.rb

@@ -2,6 +2,7 @@ module Ready
   module For
     module Migration
       class HealthController < ::ActionController::Base
+        prepend HealthActionInspectable
 
         def readiness
           if ApplicationRecord.connection.migration_context.needs_migration?

data/lib/ready/for/migration/engine.rb

@@ -2,6 +2,11 @@ module Ready
   module For
     module Migration
       class Engine < ::Rails::Engine
+
+        initializer 'ready-for-migration.initialize_cache' do
+          require config.root.join('app/controllers/concerns/ready/for/migration/health_action_inspectable')
+          Ready::For::Migration::HealthActionInspectable.cache.write(:first_access_time, Time.now)
+        end
       end
     end
   end

data/lib/ready/for/migration/version.rb

@@ -1,7 +1,7 @@
 module Ready
   module For
     module Migration
-      VERSION = '0.1.1'
+      VERSION = '0.1.4'
     end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ready-for-migration
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.4
 platform: ruby
 authors:
 - bizside-developers
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-07-28 00:00:00.000000000 Z
+date: 2023-08-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rails
@@ -60,6 +60,7 @@ files:
 - MIT-LICENSE
 - README.md
 - Rakefile
+- app/controllers/concerns/ready/for/migration/health_action_inspectable.rb
 - app/controllers/ready/for/migration/health_controller.rb
 - config/routes.rb
 - lib/ready/for/migration.rb
@@ -84,8 +85,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.7.6.2
+rubygems_version: 3.1.6
 signing_key: 
 specification_version: 4
 summary: ready for migration