lib/workless/scaler.rb in workless_revived-1.2.4.2 vs lib/workless/scaler.rb in workless_revived-1.2.5

- old
+ new

@@ -1,22 +1,21 @@ module Delayed module Workless module Scaler + autoload :Heroku, 'workless/scalers/heroku' + autoload :HerokuCedar, 'workless/scalers/heroku_cedar' + autoload :Local, 'workless/scalers/local' + autoload :Null, 'workless/scalers/null' - autoload :Heroku, "workless/scalers/heroku" - autoload :HerokuCedar, "workless/scalers/heroku_cedar" - autoload :Local, "workless/scalers/local" - autoload :Null, "workless/scalers/null" - def self.included(base) base.send :extend, ClassMethods if base.to_s =~ /ActiveRecord/ base.class_eval do - after_commit "self.class.scaler.down", :on => :update, :if => Proc.new {|r| !r.failed_at.nil? } - after_commit "self.class.scaler.down", :on => :destroy, :if => Proc.new {|r| r.destroyed? or !r.failed_at.nil? } - after_commit "self.class.scaler.up", :on => :create - end + after_commit 'self.class.scaler.down'.to_sym, on: :update, if: proc { |r| !r.failed_at.nil? } + after_commit 'self.class.scaler.down'.to_sym, on: :destroy, if: proc { |r| r.destroyed? || !r.failed_at.nil? } + after_commit 'self.class.scaler.up'.to_sym, on: :create + end elsif base.to_s =~ /Sequel/ base.send(:define_method, 'after_destroy') do super self.class.scaler.down end @@ -28,30 +27,28 @@ 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.new {|r| r.failed_at.nil? } + 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 module ClassMethods def scaler - @scaler ||= if ENV.include?("HEROKU_API_KEY") - Scaler::HerokuCedar - else - Scaler::Local + @scaler ||= if ENV.include?('HEROKU_API_KEY') + Scaler::HerokuCedar + else + Scaler::Local end end def scaler=(scaler) @scaler = "Delayed::Workless::Scaler::#{scaler.to_s.camelize}".constantize end end - end - end end