heroku_dj_auto_scale 0.0.1

data/.gitignore

@@ -0,0 +1,3 @@
+pkg/*
+*.gem
+.bundle

data/Gemfile

@@ -0,0 +1,7 @@
+source :gemcutter
+
+# Specify your gem's dependencies in heroku_dj_auto_scale.gemspec
+gemspec
+
+gem 'delayed_job', :git => 'git://github.com/joelbasson/delayed_job.git'
+gem 'heroku'

data/Rakefile

@@ -0,0 +1,2 @@
+require 'bundler'
+Bundler::GemHelper.install_tasks

data/heroku_dj_auto_scale.gemspec

@@ -0,0 +1,22 @@
+# -*- encoding: utf-8 -*-
+require File.expand_path("../lib/heroku_dj_auto_scale/version", __FILE__)
+
+Gem::Specification.new do |s|
+  s.name        = "heroku_dj_auto_scale"
+  s.version     = HerokuDjAutoScale::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Joel Basson"]
+  s.email       = []
+  s.homepage    = "http://rubygems.org/gems/heroku_dj_auto_scale"
+  s.summary     = %q{Allows auto-scaling of delayed job workers in Heroku}
+  s.description = %q{Allows auto-scaling of delayed job workers in Heroku}
+
+  s.required_rubygems_version = ">= 1.3.6"
+  s.rubyforge_project         = "heroku_dj_auto_scale"
+  
+  s.add_development_dependency "bundler", ">= 1.0.0.rc.6"
+  
+  s.files        = `git ls-files`.split("\n")
+  s.executables  = `git ls-files`.split("\n").map{|f| f =~ /^bin\/(.*)/ ? $1 : nil}.compact
+  s.require_path = 'lib'
+end

data/lib/heroku_dj_auto_scale/version.rb

@@ -0,0 +1,3 @@
+module HerokuDjAutoScale
+  VERSION = "0.0.1"
+end

data/lib/heroku_dj_auto_scale.rb

@@ -0,0 +1,59 @@
+require 'heroku'
+
+module HerokuDjAutoScale
+  module Scaler
+    class << self
+      @@heroku = Heroku::Client.new(ENV['HEROKU_USER'], ENV['HEROKU_PASS']) if ENV['USE_HEROKU_SCALING'] == 'true'
+
+      def workers
+        ENV['USE_HEROKU_SCALING'] == 'true' ? @@heroku.info(ENV['HEROKU_APP'])[:workers].to_i : 1
+      end
+
+      def workers=(qty)
+        @@heroku.set_workers(ENV['HEROKU_APP'], qty) if ENV['USE_HEROKU_SCALING'] == 'true'
+      end
+
+      def job_count
+        Delayed::Job.all.count.to_i
+      end
+      
+      def job_count_unlocked
+        Delayed::Job.where(:locked_by => nil).count.to_i
+      end
+    end
+  end
+
+  def completed(*args)
+    # Nothing fancy, just shut everything down if we have no jobs
+    Scaler.workers = 0 if Scaler.job_count.zero? 
+  end
+
+  def enqueue(*args)
+    [
+      {
+        :workers => 1, # This many workers
+        :job_count => 1 # For this many jobs or more, until the next level
+      },
+      {
+        :workers => 1,
+        :job_count => 400
+      },
+      {
+        :workers => 1,
+        :job_count => 2500
+      }
+    ].reverse_each do |scale_info|
+      # Run backwards so it gets set to the highest value first
+      # Otherwise if there were 70 jobs, it would get set to 1, then 2, then 3, etc
+
+      # If we have a job count greater than or equal to the job limit for this scale info
+      if Scaler.job_count >= scale_info[:job_count]
+        # Set the number of workers unless they are already set to a level we want. Don't scale down here!
+        if Scaler.workers <= scale_info[:workers]
+          Scaler.workers = scale_info[:workers]
+        end
+        break # We've set or ensured that the worker count is high enough
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,92 @@
+--- !ruby/object:Gem::Specification 
+name: heroku_dj_auto_scale
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: false
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Joel Basson
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-03-14 00:00:00 +02:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: bundler
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 15424057
+        segments: 
+        - 1
+        - 0
+        - 0
+        - rc
+        - 6
+        version: 1.0.0.rc.6
+  type: :development
+  version_requirements: *id001
+description: Allows auto-scaling of delayed job workers in Heroku
+email: []
+
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- Rakefile
+- heroku_dj_auto_scale.gemspec
+- lib/heroku_dj_auto_scale.rb
+- lib/heroku_dj_auto_scale/version.rb
+has_rdoc: true
+homepage: http://rubygems.org/gems/heroku_dj_auto_scale
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 23
+      segments: 
+      - 1
+      - 3
+      - 6
+      version: 1.3.6
+requirements: []
+
+rubyforge_project: heroku_dj_auto_scale
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: Allows auto-scaling of delayed job workers in Heroku
+test_files: []
+