circuit_switch 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d036ceb2748b3432749f5a03d83d324b59d1acb340bb7d64cbe3ef86f5531c4f
-  data.tar.gz: 32b8e3dbe02c1eaf8e25b40ab26f97036ecf33402bf974e6b71cb989db8c3178
+  metadata.gz: 9281021c31b4e885f3e8c8ca7fbd1fd83e713202950776c910f9b0951891a5c6
+  data.tar.gz: 8f4969657c64c2016724fabadd8d2cbddf063e6ffc3dd0298fb2e9e0b06e688a
 SHA512:
-  metadata.gz: fa10c7a2196b37619dc94605efca82c8b3ce5b3182063004911f8aa28bef44f711e945adc73f38857e85a40a6f005b1139a7fac990e76e39a45080e481a93c3f
-  data.tar.gz: 3619df1cd293ba93f271ca536bcb55f4152a20810e52fb7aba16cd44e11b6bd081645bc1804f4f01c47a49c056a1793e840de5715f4c54126b4bdda8887f1d0c
+  metadata.gz: 94cc63d4de19ffdc4cea5a5e331af21ab458f26f5333a6ceb4b84c7b3ba13edb44ebee7c0379c555d1ea7f390e8a31cb159c35c8d181cd8dec4a16cb049507ab
+  data.tar.gz: 2f490f1e01f4e05351cbde8ee6fe328e200c020dcbb0d83a1f627983a47c192f4c4ab8f5c3b3878bd58e7a026003d0bd79a5995044987b275b59b6d9902f478b

data/CHANGELOG.md

@@ -1,3 +1,21 @@
+## 0.2.1
+
+### New features
+
+* Add `initially_closed` option to `run` and `open?`.
+* Add `key` argument for easy handling for human more than caller.  
+To migrate, run next.
+
+```
+rails generate circuit_switch:migration add_key
+rails db:migrate
+```
+
+### Changes
+
+* Modify log level from warn to info when default value for `close_if_reach_limit` is used.
+* Suppress warning that ivar is not initialized.
+
 ## 0.2.0
 
 ### Breaking Changes
@@ -6,7 +24,7 @@
 
 ## 0.1.2
 
-* Modify `CircuitSwitch.open?` recieves same arguments as `CircuitSwtich.run`
+* Modify `CircuitSwitch.open?` receives same arguments as `CircuitSwitch.run`
 
 ## 0.1.1
 

data/README.md

@@ -37,7 +37,7 @@ rails generate circuit_switch:install
 ```
 
 Generate a migration for ActiveRecord.  
-This table saves circuit caller, called count, limit count and so on.
+This table saves named key, circuit caller, called count, limit count and so on.
 
 ```
 rails generate circuit_switch:migration circuit_switch
@@ -61,11 +61,14 @@ To switch circuit opening and closing, some conditions can be set. By default, t
 You can also set `limit_count` to close circuit when reached the specified count. Default limit_count is 10. To change this default value, modify `circuit_switches.run_limit_count` default value in the migration file.  
 `run` receives optional arguments.
 
+- `key`: [String] Named key to find switch instead of caller  
+  If no key passed, use caller.
 - `if`: [Boolean, Proc] Call proc when `if` is truthy (default: true)
 - `close_if`: [Boolean, Proc] Call proc when `close_if` is falsy (default: false)
 - `close_if_reach_limit`: [Boolean] Stop calling proc when run count reaches limit (default: false)
 - `limit_count`: [Integer] Limit count. Use `run_limit_count` default value if it's nil (default: nil)  
   Can't be set 0 when `close_if_reach_limit` is true
+- `initially_closed`: [Boolean] Create switch with terminated mode (default: false)
 
 To close the circuit at specific date or when called 1000 times, code goes like:
 
@@ -106,6 +109,8 @@ CircuitSwitch.report(if: some_condition)
 Same as `run`, some conditions can be set. By default, reporting is stopped when reached the specified count. The default count is 10. To change this default value, modify `circuit_switches.report_limit_count` default value in the migration file.  
 `report` receives optional arguments.
 
+- `key`: [String] Named key to find switch instead of caller  
+  If no key passed, use caller.
 - `if`: [Boolean, Proc] Report when `if` is truthy (default: true)
 - `stop_report_if`: [Boolean, Proc] Report when `close_if` is falsy (default: false)
 - `stop_report_if_reach_limit`: [Boolean] Stop reporting when reported count reaches limit (default: true)
@@ -131,10 +136,11 @@ called_path: /app/services/greetings_service:21 block in validate
 
 ## Task
 
-When find a problem and you want to terminate running or reporting right now, execute a task with it's caller.
+When find a problem and you want to terminate running or reporting right now, execute a task with it's caller.  
+You can specify either key or caller.
 
 ```
-rake circuit_switch:terminate_to_run[/app/services/greetings_service:21 block in validate]
+rake circuit_switch:terminate_to_run[your_key]
 rake circuit_switch:terminate_to_report[/app/services/greetings_service:21 block in validate]
 ```
 

data/lib/circuit_switch/builder.rb

@@ -0,0 +1,66 @@
+require_relative 'core'
+
+module CircuitSwitch
+  class Builder < Core
+    def initialize
+      super
+      @run = false
+      @reported = false
+    end
+
+    def assign_runner(
+      key: nil,
+      if: true,
+      close_if: false,
+      close_if_reach_limit: nil,
+      limit_count: nil,
+      initially_closed: false
+    )
+      @key = key
+      @run_if = binding.local_variable_get(:if)
+      @close_if = close_if
+      @close_if_reach_limit = close_if_reach_limit
+      @run_limit_count = limit_count
+      @initially_closed = initially_closed
+    end
+
+    def assign_reporter(
+      key: nil,
+      if: true,
+      stop_report_if: false,
+      stop_report_if_reach_limit: true,
+      limit_count: nil
+    )
+      @key = key
+      @report_if = binding.local_variable_get(:if)
+      @stop_report_if = stop_report_if
+      @stop_report_if_reach_limit = stop_report_if_reach_limit
+      @report_limit_count = limit_count
+    end
+
+    def run(key: nil, if: nil, close_if: nil, close_if_reach_limit: nil, limit_count: nil, initially_closed: nil, &block)
+      arguments = {
+        key: key,
+        if: binding.local_variable_get(:if),
+        close_if: close_if,
+        close_if_reach_limit: close_if_reach_limit,
+        limit_count: limit_count,
+        initially_closed: initially_closed,
+      }.select { |_, v| v }
+      assign_runner(**arguments)
+      execute_run(&block)
+    end
+
+    def report(key: nil, if: nil, stop_report_if: nil, stop_report_if_reach_limit: nil, limit_count: nil)
+      arguments = {
+        key: key,
+        if: binding.local_variable_get(:if),
+        stop_report_if: stop_report_if,
+        stop_report_if_reach_limit: stop_report_if_reach_limit,
+        limit_count: limit_count
+      }.select { |_, v| v }
+      assign_reporter(**arguments)
+      execute_report
+    end
+  end
+end

data/lib/circuit_switch/core.rb

@@ -1,54 +1,55 @@
+require_relative 'notification'
+require_relative 'workers/reporter'
+require_relative 'workers/run_count_updater'
+
 module CircuitSwitch
   class Core
     delegate :config, to: ::CircuitSwitch
+    attr_reader :key, :run_if, :close_if, :close_if_reach_limit, :run_limit_count, :initially_closed,
+      :report_if, :stop_report_if, :stop_report_if_reach_limit, :report_limit_count
 
-    def run(
-      if: true,
-      close_if: false,
-      close_if_reach_limit: nil,
-      limit_count: nil,
-      &block
-    )
-      if close_if_reach_limit && limit_count == 0
+    def execute_run(&block)
+      if close_if_reach_limit && run_limit_count == 0
         raise CircuitSwitchError.new('Can\'t set limit_count to 0 when close_if_reach_limit is true')
       end
       if close_if_reach_limit.nil?
-        Logger.new($stdout).warn('Default value for close_if_reach_limit is modified from true to false at ver 0.2.0.')
-        close_if_reach_limit = false
+        Logger.new($stdout).info('Default value for close_if_reach_limit is modified from true to false at ver 0.2.0.')
+        @close_if_reach_limit = false
       end
-      return self if evaluate(close_if) || !evaluate(binding.local_variable_get(:if))
-      return self if close_if_reach_limit && switch.reached_run_limit?(limit_count)
+
+      return self if evaluate(close_if) || !evaluate(run_if)
+      return self if close_if_reach_limit && switch.reached_run_limit?(run_limit_count)
       return self if switch.run_is_terminated?
 
-      yield
+      unless switch.new_record? && initially_closed
+        yield
+        @run = true
+      end
       RunCountUpdater.perform_later(
-        limit_count: limit_count,
+        key: key,
+        limit_count: run_limit_count,
         called_path: called_path,
-        reported: reported?
+        reported: reported?,
+        initially_closed: initially_closed
       )
-      @run = true
       self
     end
 
-    def report(
-      if: true,
-      stop_report_if: false,
-      stop_report_if_reach_limit: true,
-      limit_count: nil
-    )
+    def execute_report
       if config.reporter.nil?
         raise CircuitSwitchError.new('Set config.reporter.')
       end
-      if stop_report_if_reach_limit && limit_count == 0
+      if stop_report_if_reach_limit && report_limit_count == 0
         raise CircuitSwitchError.new('Can\'t set limit_count to 0 when stop_report_if_reach_limit is true')
       end
       return self unless config.enable_report?
-      return self if evaluate(stop_report_if) || !evaluate(binding.local_variable_get(:if))
+      return self if evaluate(stop_report_if) || !evaluate(report_if)
       return self if switch.report_is_terminated?
-      return self if stop_report_if_reach_limit && switch.reached_report_limit?(limit_count)
+      return self if stop_report_if_reach_limit && switch.reached_report_limit?(report_limit_count)
 
       Reporter.perform_later(
-        limit_count: limit_count,
+        key: key,
+        limit_count: report_limit_count,
         called_path: called_path,
         run: run?
       )
@@ -58,18 +59,24 @@ module CircuitSwitch
 
     # @return [Boolean]
     def run?
-      !!@run
+      @run
     end
 
     # @return [Boolean]
     def reported?
-      !!@reported
+      @reported
     end
 
     private
 
     def switch
-      @switch ||= CircuitSwitch.find_or_initialize_by(caller: called_path)
+      return @switch if defined? @switch
+
+      if key
+        @switch = CircuitSwitch.find_or_initialize_by(key: key)
+      else
+        @switch = CircuitSwitch.find_or_initialize_by(caller: called_path)
+      end
     end
 
     def called_path

data/lib/circuit_switch/notification.rb

@@ -1,4 +1,4 @@
-require 'circuit_switch/stacktrace_modifier'
+require_relative 'stacktrace_modifier'
 
 module CircuitSwitch
   class CircuitSwitchError < RuntimeError

data/lib/circuit_switch/orm/active_record/circuit_switch.rb

@@ -1,5 +1,11 @@
+require 'active_record'
+
 module CircuitSwitch
   class CircuitSwitch < ::ActiveRecord::Base
+    after_initialize do |switch|
+      switch.key ||= switch.caller
+    end
+
     def assign(run_limit_count: nil, report_limit_count: nil)
       self.run_limit_count = run_limit_count if run_limit_count
       self.report_limit_count = report_limit_count if report_limit_count
@@ -31,7 +37,8 @@ module CircuitSwitch
     end
 
     def message
-      "Watching process is called for #{report_count}th. Report until for #{report_limit_count}th."
+      process = key == caller ? 'Watching process' : "Process for '#{key}'"
+      "#{process} is called for #{report_count}th. Report until for #{report_limit_count}th."
     end
 
     private

data/lib/circuit_switch/stacktrace_modifier.rb

@@ -1,4 +1,4 @@
-require 'circuit_switch/configuration.rb'
+require 'active_support/core_ext/module/delegation'
 
 module CircuitSwitch
   class StacktraceModifier

data/lib/circuit_switch/tasks/circuit_switch.rake

@@ -1,40 +1,40 @@
 namespace :circuit_switch do
   desc 'Update run_is_terminated to true to close circuit'
-  task :terminate_to_run, ['caller'] => :environment do |_, arg|
-    called_path = arg[:caller]
-    puts "Start to update run_is_terminated of circuit_switch for '#{called_path}' to true."
+  task :terminate_to_run, ['key_or_caller'] => :environment do |_, arg|
+    key_or_caller = arg[:key_or_caller]
+    puts "Start to update run_is_terminated of circuit_switch for '#{key_or_caller}' to true."
     sleep(3)
 
-    switch = CircuitSwitch::CircuitSwitch.find_by!(caller: called_path)
+    switch = CircuitSwitch::CircuitSwitch.find_by(key: key_or_caller) || CircuitSwitch::CircuitSwitch.find_by!(caller: key_or_caller)
     puts "circuit_switch is found. id: #{switch.id}."
 
     switch.update(run_is_terminated: true)
-    puts "Updated run_is_terminated of circuit_switch for '#{called_path}' to true."
+    puts "Updated run_is_terminated of circuit_switch for '#{key_or_caller}' to true."
   end
 
   desc 'Update report_is_terminated to true to stop reporting'
-  task :terminate_to_report, ['caller'] => :environment do |_, arg|
-    called_path = arg[:caller]
-    puts "Start to update report_is_terminated of circuit_switch for '#{called_path}' to true."
+  task :terminate_to_report, ['key_or_caller'] => :environment do |_, arg|
+    key_or_caller = arg[:key_or_caller]
+    puts "Start to update report_is_terminated of circuit_switch for '#{key_or_caller}' to true."
     sleep(3)
 
-    switch = CircuitSwitch::CircuitSwitch.find_by!(caller: called_path)
+    switch = CircuitSwitch::CircuitSwitch.find_by(key: key_or_caller) || CircuitSwitch::CircuitSwitch.find_by!(caller: key_or_caller)
     puts "circuit_switch is found. id: #{switch.id}."
 
     switch.update(report_is_terminated: true)
-    puts "Updated report_is_terminated of circuit_switch for '#{called_path}' to true."
+    puts "Updated report_is_terminated of circuit_switch for '#{key_or_caller}' to true."
   end
 
   desc 'Delete switch'
-  task :delete_switch, ['caller'] => :environment do |_, arg|
-    called_path = arg[:caller]
-    puts "Start to delete circuit_switch for '#{called_path}'."
+  task :delete_switch, ['key_or_caller'] => :environment do |_, arg|
+    key_or_caller = arg[:key_or_caller]
+    puts "Start to delete circuit_switch for '#{key_or_caller}'."
     sleep(3)
 
-    switch = CircuitSwitch::CircuitSwitch.find_by!(caller: called_path)
+    switch = CircuitSwitch::CircuitSwitch.find_by(key: key_or_caller) || CircuitSwitch::CircuitSwitch.find_by!(caller: key_or_caller)
     puts "circuit_switch is found. id: #{switch.id}."
 
     switch.destroy!
-    puts "Successfully deleted circuit_switch for '#{called_path}'."
+    puts "Successfully deleted circuit_switch for '#{key_or_caller}'."
   end
 end

data/lib/circuit_switch/version.rb

@@ -1,3 +1,3 @@
 module CircuitSwitch
-  VERSION = '0.2.0'
+  VERSION = '0.2.1'
 end

data/lib/circuit_switch/workers/due_date_notifier.rb

@@ -10,7 +10,7 @@ module CircuitSwitch
       circuit_switches = CircuitSwitch.where('due_date <= ?', Date.today).order(id: :asc)
       if circuit_switches.present?
         message = "Due date has come! Let's consider about removing switches and cleaning up code! :)\n" +
-          circuit_switches.map { |switch|  "id: #{switch.id}, caller: '#{switch.caller}' , created_at: #{switch.created_at}" }.join("\n")
+          circuit_switches.map { |switch|  "id: #{switch.id}, key: '#{switch.key}', caller: '#{switch.caller}', created_at: #{switch.created_at}" }.join("\n")
         config.due_date_notifier.call(message)
       else
         switches_count = CircuitSwitch.all.size

data/lib/circuit_switch/workers/reporter.rb

@@ -1,16 +1,20 @@
+require 'active_job'
+require 'active_support/core_ext/module/delegation'
+
 module CircuitSwitch
   class Reporter < ::ActiveJob::Base
     delegate :config, to: ::CircuitSwitch
 
-    def perform(limit_count:, called_path:, run:)
-      circuit_switch =
-      if run
-        # Wait for RunCountUpdater saves circuit_switch
-        sleep(3)
-        CircuitSwitch.find_by!(caller: called_path)
-      else
-        CircuitSwitch.find_or_initialize_by(caller: called_path)
+    def perform(key:, limit_count:, called_path:, run:)
+      # Wait for RunCountUpdater saves circuit_switch
+      sleep(3) if run
+
+      circuit_switch = key ? CircuitSwitch.find_by(key: key) : CircuitSwitch.find_by(caller: called_path)
+      if run && circuit_switch.nil?
+        raise ActiveRecord::RecordNotFound.new('Couldn\'t find CircuitSwitch::CircuitSwitch')
       end
+
+      circuit_switch ||= CircuitSwitch.new(key: key, caller: called_path)
       circuit_switch.due_date ||= config.due_date
       circuit_switch.assign(report_limit_count: limit_count).increment_report_count
       raise CalledNotification.new(circuit_switch.message)

data/lib/circuit_switch/workers/run_count_updater.rb

@@ -1,16 +1,20 @@
+require 'active_job'
+require 'active_support/core_ext/module/delegation'
+
 module CircuitSwitch
   class RunCountUpdater < ::ActiveJob::Base
     delegate :config, to: ::CircuitSwitch
 
-    def perform(limit_count:, called_path:, reported:)
-      circuit_switch =
-        if reported
-          # Wait for Reporter saves circuit_switch
-          sleep(3)
-          CircuitSwitch.find_by!(caller: called_path)
-        else
-          CircuitSwitch.find_or_initialize_by(caller: called_path)
-        end
+    def perform(key:, limit_count:, called_path:, reported:, initially_closed:)
+      # Wait for Reporter saves circuit_switch
+      sleep(3) if reported
+
+      circuit_switch = key ? CircuitSwitch.find_by(key: key) : CircuitSwitch.find_by(caller: called_path)
+      if reported && circuit_switch.nil?
+        raise ActiveRecord::RecordNotFound.new('Couldn\'t find CircuitSwitch::CircuitSwitch')
+      end
+
+      circuit_switch ||= CircuitSwitch.new(key: key, caller: called_path, run_is_terminated: initially_closed)
       circuit_switch.due_date ||= config.due_date
       circuit_switch.assign(run_limit_count: limit_count).increment_run_count
     end

data/lib/circuit_switch.rb

@@ -1,12 +1,9 @@
-require "circuit_switch/configuration.rb"
-require "circuit_switch/core"
-require "circuit_switch/notification"
-require "circuit_switch/orm/active_record/circuit_switch"
-require "circuit_switch/railtie" if defined?(Rails::Railtie)
-require "circuit_switch/version"
-require "circuit_switch/workers/due_date_notifier"
-require "circuit_switch/workers/reporter"
-require "circuit_switch/workers/run_count_updater"
+require_relative 'circuit_switch/configuration'
+require_relative 'circuit_switch/builder'
+require_relative 'circuit_switch/orm/active_record/circuit_switch'
+require_relative 'circuit_switch/railtie' if defined?(Rails::Railtie)
+require_relative 'circuit_switch/version'
+require_relative 'circuit_switch/workers/due_date_notifier'
 
 module CircuitSwitch
   class << self
@@ -18,73 +15,69 @@ module CircuitSwitch
       @config ||= Configuration.new
     end
 
+    # @param key [String] Named key to find switch instead of caller
     # @param if [Boolean, Proc] Call proc when `if` is truthy (default: true)
     # @param close_if [Boolean, Proc] Call proc when `close_if` is falsy (default: false)
     # @param close_if_reach_limit [Boolean] Stop calling proc when run count reaches limit (default: false)
     # @param limit_count [Integer] Limit count. Use `run_limit_count` default value if it's nil
     #   Can't be set 0 when `close_if_reach_limit` is true (default: nil)
+    # @param initially_closed [Boolean] Create switch with terminated mode (default: false)
     # @param [Proc] block
-    def run(
-      if: true,
-      close_if: false,
-      close_if_reach_limit: nil,
-      limit_count: nil,
-      &block
-    )
-      Core.new.run(
+    def run(key: nil, if: nil, close_if: nil, close_if_reach_limit: nil, limit_count: nil, initially_closed: nil, &block)
+      arguments = {
+        key: key,
         if: binding.local_variable_get(:if),
         close_if: close_if,
         close_if_reach_limit: close_if_reach_limit,
         limit_count: limit_count,
-        &block
-      )
+        initially_closed: initially_closed,
+      }.select { |_, v| v }
+      Builder.new.run(**arguments, &block)
     end
 
+    # @param key [String] Named key to find switch instead of caller
     # @param if [Boolean, Proc] Report when `if` is truthy (default: true)
     # @param stop_report_if [Boolean, Proc] Report when `close_if` is falsy (default: false)
     # @param stop_report_if_reach_limit [Boolean] Stop reporting when reported count reaches limit (default: true)
     # @param limit_count [Integer] Limit count. Use `report_limit_count` default value if it's nil
     #   Can't be set 0 when `stop_report_if_reach_limit` is true (default: nil)
-    def report(
-      if: true,
-      stop_report_if: false,
-      stop_report_if_reach_limit: true,
-      limit_count: nil
-    )
+    def report(key: nil, if: nil, stop_report_if: nil, stop_report_if_reach_limit: nil, limit_count: nil)
       if block_given?
         raise ArgumentError.new('CircuitSwitch.report doesn\'t receive block. Use CircuitSwitch.run if you want to pass block.')
       end
 
-      Core.new.report(
+      arguments = {
+        key: key,
         if: binding.local_variable_get(:if),
         stop_report_if: stop_report_if,
         stop_report_if_reach_limit: stop_report_if_reach_limit,
         limit_count: limit_count
-      )
+      }.select { |_, v| v }
+      Builder.new.report(**arguments)
     end
 
     # Syntax sugar for `CircuitSwitch.run`
+    # @param key [String] Named key to find switch instead of caller
     # @param if [Boolean, Proc] `CircuitSwitch.run` is runnable when `if` is truthy (default: true)
     # @param close_if [Boolean, Proc] `CircuitSwitch.run` is runnable when `close_if` is falsy (default: false)
     # @param close_if_reach_limit [Boolean] `CircuitSwitch.run` is NOT runnable when run count reaches limit (default: true)
     # @param limit_count [Integer] Limit count. Use `run_limit_count` default value if it's nil. Can't be set 0 (default: nil)
+    # @param initially_closed [Boolean] Create switch with terminated mode (default: false)
     # @return [Boolean]
-    def open?(
-      if: true,
-      close_if: false,
-      close_if_reach_limit: true,
-      limit_count: nil
-    )
+    def open?(key: nil, if: nil, close_if: nil, close_if_reach_limit: nil, limit_count: nil, initially_closed: nil)
       if block_given?
         raise ArgumentError.new('CircuitSwitch.open doesn\'t receive block. Use CircuitSwitch.run if you want to pass block.')
       end
 
-      Core.new.run(
+      arguments = {
+        key: key,
         if: binding.local_variable_get(:if),
         close_if: close_if,
         close_if_reach_limit: close_if_reach_limit,
-        limit_count: limit_count
-      ) {}.run?
+        limit_count: limit_count,
+        initially_closed: initially_closed,
+      }.select { |_, v| v }
+      Builder.new.run(**arguments) {}.run?
     end
   end
 end

data/lib/generators/circuit_switch/migration_generator.rb

@@ -4,9 +4,16 @@ module CircuitSwitch
   class MigrationGenerator < ActiveRecord::Generators::Base
     desc 'Create a migration to manage circuit switches state'
     source_root File.expand_path('templates', __dir__)
+    argument :migration_type, required: false, type: :array, default: ['create'],
+      desc: 'Type of migration to create or add key column. By default to create.',
+      banner: 'create or add_key'
 
     def generate_migration
-      migration_template 'migration.rb.erb', 'db/migrate/create_circuit_switches.rb', migration_version: migration_version
+      if migration_type == ['add_key']
+        migration_template 'add_key.rb.erb', 'db/migrate/add_key_to_circuit_switches.rb', migration_version: migration_version
+      else
+        migration_template 'migration.rb.erb', 'db/migrate/create_circuit_switches.rb', migration_version: migration_version
+      end
     end
 
     def migration_version

data/lib/generators/circuit_switch/templates/add_key.rb.erb

@@ -0,0 +1,13 @@
+class AddKeyToCircuitSwitches < ActiveRecord::Migration<%= migration_version %>
+  def up
+    add_column :circuit_switches, :key, :string, after: :id
+    CircuitSwitch::CircuitSwitch.all.each { |switch| switch.update_column(:key, switch.caller) }
+    change_column_null :circuit_switches, :key, false
+    add_index :circuit_switches, :key
+  end
+
+  def down
+    remove_index :circuit_switches, :key
+    remove_column :circuit_switches, :key
+  end
+end

data/lib/generators/circuit_switch/templates/initializer.rb

@@ -29,7 +29,7 @@ CircuitSwitch.configure do |config|
   # You may be want backtrace when report to plain feed; e.g. Slack or email.
   # config.with_backtrace = false
 
-  # Allowd backtrace paths to report
+  # Allowed backtrace paths to report
   # Specify with `with_backtrace` option.
   # Allowed all paths when set `[]`.
   # config.allowed_backtrace_paths = [Dir.pwd]

data/lib/generators/circuit_switch/templates/migration.rb.erb

@@ -1,6 +1,7 @@
 class CreateCircuitSwitches < ActiveRecord::Migration<%= migration_version %>
   def change
     create_table :circuit_switches do |t|
+      t.string :key, null: false, index: true
       t.string :caller, null: false
       t.integer :run_count, default: 0, null: false
       t.integer :run_limit_count, default: 10, null: false

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: circuit_switch
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - makicamel
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-10-13 00:00:00.000000000 Z
+date: 2021-10-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activejob
@@ -114,6 +114,7 @@ files:
 - bin/setup
 - circuit_switch.gemspec
 - lib/circuit_switch.rb
+- lib/circuit_switch/builder.rb
 - lib/circuit_switch/configuration.rb
 - lib/circuit_switch/core.rb
 - lib/circuit_switch/notification.rb
@@ -128,6 +129,7 @@ files:
 - lib/circuit_switch/workers/run_count_updater.rb
 - lib/generators/circuit_switch/install_generator.rb
 - lib/generators/circuit_switch/migration_generator.rb
+- lib/generators/circuit_switch/templates/add_key.rb.erb
 - lib/generators/circuit_switch/templates/initializer.rb
 - lib/generators/circuit_switch/templates/migration.rb.erb
 homepage: https://github.com/makicamel/circuit_switch