lib/workless/scalers/heroku.rb in workless-1.0.2 vs lib/workless/scalers/heroku.rb in workless-1.1.0

@@ -1,24 +1,24 @@
-require 'heroku'
+require 'heroku-api'
 
 module Delayed
   module Workless
     module Scaler
 
       class Heroku < Base
 
         extend Delayed::Workless::Scaler::HerokuClient
 
         def self.up
-          client.set_workers(ENV['APP_NAME'], 1) if self.workers == 0
+          client.put_workers(ENV['APP_NAME'], 1) if self.workers == 0
         end
 
         def self.down
-          client.set_workers(ENV['APP_NAME'], 0) unless self.workers == 0 or self.jobs.count > 0
+          client.put_workers(ENV['APP_NAME'], 0) unless self.workers == 0 or self.jobs.count > 0
         end
 
         def self.workers
-          client.info(ENV['APP_NAME'])[:workers].to_i
+          client.get_ps(ENV['APP_NAME']).body.count { |p| p["process"] =~ /worker\.\d?/ }
         end
 
       end
 
     end