Module: RocketJob::Batch::Throttle::ClassMethods

Defined in:
lib/rocket_job/batch/throttle.rb

Instance Method Summary collapse

Instance Method Details

#batch_throttle?(method_name) ⇒ Boolean

Has a throttle been defined?

Returns:

  • (Boolean) 


66
67
68
 
# File 'lib/rocket_job/batch/throttle.rb', line 66

def batch_throttle?(method_name)
  rocket_job_batch_throttles.exist?(method_name)
end

#define_batch_throttle(method_name, filter: :throttle_filter_class) ⇒ Object

Add a new throttle.

Parameters:

method_name: [Symbol]
  Name of method to call to evaluate whether a throttle has been exceeded.
  Note: Must return true or false.
filter: [Symbol|Proc]
  Name of method to call to return the filter when the throttle has been exceeded.
  Or, a block that will return the filter.
  Default: :throttle_filter_class (Throttle all jobs of this class)

Note: Throttles are executed in the order they are defined.



49
50
51
52
53
54
 
# File 'lib/rocket_job/batch/throttle.rb', line 49

def define_batch_throttle(method_name, filter: :throttle_filter_class)
  # Duplicate to prevent modifying parent class throttles
  definitions = rocket_job_batch_throttles ? rocket_job_batch_throttles.dup : ThrottleDefinitions.new
  definitions.add(method_name, filter)
  self.rocket_job_batch_throttles = definitions
end

#undefine_batch_throttle(method_name) ⇒ Object

Undefine a previously defined throttle



57
58
59
60
61
62
63
 
# File 'lib/rocket_job/batch/throttle.rb', line 57

def undefine_batch_throttle(method_name)
  return unless rocket_job_batch_throttles

  definitions = rocket_job_batch_throttles.dup
  definitions.remove(method_name)
  self.rocket_job_batch_throttles = definitions
end