lib/workless/scalers/heroku_cedar.rb in rcarvalho-workless-1.0.2.9 vs lib/workless/scalers/heroku_cedar.rb in rcarvalho-workless-1.0.2.10

@@ -7,35 +7,32 @@
       class HerokuCedar < Base
 
         extend Delayed::Workless::Scaler::HerokuClient
 
         def self.up
-          @@current_num_workers ||= 0
           nw = self.calculate_num_workers
           if nw
-            if @@current_num_workers != nw
-              @@current_num_workers = nw
+            if self.num_workers_cache != nw
+              self.num_workers_cache = nw
               client.ps_scale(ENV['APP_NAME'], :type => 'worker', :qty => @@current_num_workers) 
             end
           end
         rescue
         end
 
         def self.down
-          @@current_num_workers ||= 0
           nw = self.calculate_num_workers
           if nw
-            if @@current_num_workers != nw
-              @@current_num_workers = nw
+            if self.num_workers_cache != nw
+              self.num_workers_cache = nw
               client.ps_scale(ENV['APP_NAME'], :type => 'worker', :qty => nw)
             end
           end          
         rescue
         end
 
         def self.workers
-          @@current_num_workers ||= 0
-          @@current_num_workers || client.ps(ENV['APP_NAME']).count { |p| p["process"] =~ /worker\.\d?/ }
+          self.num_workers_cache
         end
 
       end
 
     end