autoscaler 0.4.0 → 0.4.1

data/lib/autoscaler/delayed_shutdown.rb

@@ -0,0 +1,44 @@
+module Autoscaler
+  # This is a sort of middleware that keeps the last worker up for a minimum amount of time
+  class DelayedShutdown
+    # @param [ScalingStrategy] strategy object that makes most decisions
+    # @param [Numeric] timeout time to stay up after base strategy says zero
+    def initialize(strategy, timeout)
+      @strategy = strategy
+      @timeout = timeout
+      active_now!
+    end
+
+    # @param [QueueSystem] system interface to the queuing system
+    # @param [Numeric] event_idle_time number of seconds since a job related event
+    # @return [Integer] target number of workers
+    def call(system, event_idle_time)
+      target_workers = strategy.call(system, event_idle_time)
+      if target_workers > 0
+        active_now!
+        target_workers
+      elsif time_left?(event_idle_time)
+        1
+      else
+        0
+      end
+    end
+
+    private
+
+    attr_reader :strategy
+    attr_reader :timeout
+
+    def active_now!
+      @activity = Time.now
+    end
+
+    def level_idle_time
+      Time.now - @activity
+    end
+
+    def time_left?(event_idle_time)
+      [event_idle_time, level_idle_time].min < timeout
+    end
+  end
+end

data/lib/autoscaler/version.rb

@@ -1,4 +1,4 @@
 module Autoscaler
   # version number
-  VERSION = "0.4.0"
+  VERSION = "0.4.1"
 end

data/spec/autoscaler/delayed_shutdown_spec.rb

@@ -0,0 +1,23 @@
+require 'spec_helper'
+require 'test_system'
+require 'autoscaler/delayed_shutdown'
+
+describe Autoscaler::DelayedShutdown do
+  let(:cut) {Autoscaler::DelayedShutdown}
+
+  it "returns normal values" do
+    strategy = cut.new(lambda{|s,t| 2}, 0)
+    strategy.call(nil, 1).should == 2
+  end
+
+  it "delays zeros" do
+    strategy = cut.new(lambda{|s,t| 0}, 60)
+    strategy.call(nil, 1).should == 1
+  end
+
+  it "eventually returns zero" do
+    strategy = cut.new(lambda{|s,t| 0}, 60)
+    strategy.stub(:level_idle_time).and_return(61)
+    strategy.call(nil, 61).should == 0
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: autoscaler
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.4.1
   prerelease: 
 platform: ruby
 authors:
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-06-21 00:00:00.000000000 Z
+date: 2013-06-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sidekiq
@@ -133,6 +133,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/autoscaler/binary_scaling_strategy.rb
+- lib/autoscaler/delayed_shutdown.rb
 - lib/autoscaler/heroku_scaler.rb
 - lib/autoscaler/sidekiq/activity.rb
 - lib/autoscaler/sidekiq/celluloid_monitor.rb
@@ -152,6 +153,7 @@ files:
 - examples/simple.rb
 - Guardfile
 - spec/autoscaler/binary_scaling_strategy_spec.rb
+- spec/autoscaler/delayed_shutdown_spec.rb
 - spec/autoscaler/heroku_scaler_spec.rb
 - spec/autoscaler/sidekiq/activity_spec.rb
 - spec/autoscaler/sidekiq/celluloid_monitor_spec.rb
@@ -190,6 +192,7 @@ summary: Start/stop Sidekiq workers on Heroku
 test_files:
 - Guardfile
 - spec/autoscaler/binary_scaling_strategy_spec.rb
+- spec/autoscaler/delayed_shutdown_spec.rb
 - spec/autoscaler/heroku_scaler_spec.rb
 - spec/autoscaler/sidekiq/activity_spec.rb
 - spec/autoscaler/sidekiq/celluloid_monitor_spec.rb