Module: Delayed::Workless::Scaler

Defined in:
lib/workless/scaler.rb,
lib/workless/scalers/base.rb,
lib/workless/scalers/null.rb,
lib/workless/scalers/local.rb,
lib/workless/scalers/heroku.rb

Defined Under Namespace

Modules: ClassMethods, HerokuClient Classes: Base, Heroku, Local, Null

Class Method Summary collapse

Class Method Details

.included(base) ⇒ Object 



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
 
# File 'lib/workless/scaler.rb', line 8

def self.included(base)
  base.send :extend, ClassMethods
  if base.to_s =~ /ActiveRecord/
    base.class_eval do
      after_commit(on: :update, if: proc { |r| !r.failed_at.nil? }) do 
        self.class.scaler.down
      end
      after_commit(on: :destroy, if: proc { |r| r.destroyed? || !r.failed_at.nil? }) do
        self.class.scaler.down
      end
      after_commit(on: :create) do 
        self.class.scaler.up
      end
    end
  elsif base.to_s =~ /Sequel/
    base.send(:define_method, 'after_destroy') do
      super
      self.class.scaler.down
    end
    base.send(:define_method, 'after_create') do
      super
      self.class.scaler.up
    end
    base.send(:define_method, 'after_update') do
      super
      self.class.scaler.down
    end
  else
    base.class_eval do
      after_destroy 'self.class.scaler.down'
      after_create 'self.class.scaler.up'
      after_update 'self.class.scaler.down', unless: proc { |r| r.failed_at.nil? }
    end
  end
end