heroku_dj_auto_scale 0.0.1

Sign up to get free protection for your applications and to get access to all the features.
data/.gitignore ADDED
@@ -0,0 +1,3 @@
1
+ pkg/*
2
+ *.gem
3
+ .bundle
data/Gemfile ADDED
@@ -0,0 +1,7 @@
1
+ source :gemcutter
2
+
3
+ # Specify your gem's dependencies in heroku_dj_auto_scale.gemspec
4
+ gemspec
5
+
6
+ gem 'delayed_job', :git => 'git://github.com/joelbasson/delayed_job.git'
7
+ gem 'heroku'
data/Rakefile ADDED
@@ -0,0 +1,2 @@
1
+ require 'bundler'
2
+ Bundler::GemHelper.install_tasks
@@ -0,0 +1,22 @@
1
+ # -*- encoding: utf-8 -*-
2
+ require File.expand_path("../lib/heroku_dj_auto_scale/version", __FILE__)
3
+
4
+ Gem::Specification.new do |s|
5
+ s.name = "heroku_dj_auto_scale"
6
+ s.version = HerokuDjAutoScale::VERSION
7
+ s.platform = Gem::Platform::RUBY
8
+ s.authors = ["Joel Basson"]
9
+ s.email = []
10
+ s.homepage = "http://rubygems.org/gems/heroku_dj_auto_scale"
11
+ s.summary = %q{Allows auto-scaling of delayed job workers in Heroku}
12
+ s.description = %q{Allows auto-scaling of delayed job workers in Heroku}
13
+
14
+ s.required_rubygems_version = ">= 1.3.6"
15
+ s.rubyforge_project = "heroku_dj_auto_scale"
16
+
17
+ s.add_development_dependency "bundler", ">= 1.0.0.rc.6"
18
+
19
+ s.files = `git ls-files`.split("\n")
20
+ s.executables = `git ls-files`.split("\n").map{|f| f =~ /^bin\/(.*)/ ? $1 : nil}.compact
21
+ s.require_path = 'lib'
22
+ end
@@ -0,0 +1,3 @@
1
+ module HerokuDjAutoScale
2
+ VERSION = "0.0.1"
3
+ end
@@ -0,0 +1,59 @@
1
+ require 'heroku'
2
+
3
+ module HerokuDjAutoScale
4
+ module Scaler
5
+ class << self
6
+ @@heroku = Heroku::Client.new(ENV['HEROKU_USER'], ENV['HEROKU_PASS']) if ENV['USE_HEROKU_SCALING'] == 'true'
7
+
8
+ def workers
9
+ ENV['USE_HEROKU_SCALING'] == 'true' ? @@heroku.info(ENV['HEROKU_APP'])[:workers].to_i : 1
10
+ end
11
+
12
+ def workers=(qty)
13
+ @@heroku.set_workers(ENV['HEROKU_APP'], qty) if ENV['USE_HEROKU_SCALING'] == 'true'
14
+ end
15
+
16
+ def job_count
17
+ Delayed::Job.all.count.to_i
18
+ end
19
+
20
+ def job_count_unlocked
21
+ Delayed::Job.where(:locked_by => nil).count.to_i
22
+ end
23
+ end
24
+ end
25
+
26
+ def completed(*args)
27
+ # Nothing fancy, just shut everything down if we have no jobs
28
+ Scaler.workers = 0 if Scaler.job_count.zero?
29
+ end
30
+
31
+ def enqueue(*args)
32
+ [
33
+ {
34
+ :workers => 1, # This many workers
35
+ :job_count => 1 # For this many jobs or more, until the next level
36
+ },
37
+ {
38
+ :workers => 1,
39
+ :job_count => 400
40
+ },
41
+ {
42
+ :workers => 1,
43
+ :job_count => 2500
44
+ }
45
+ ].reverse_each do |scale_info|
46
+ # Run backwards so it gets set to the highest value first
47
+ # Otherwise if there were 70 jobs, it would get set to 1, then 2, then 3, etc
48
+
49
+ # If we have a job count greater than or equal to the job limit for this scale info
50
+ if Scaler.job_count >= scale_info[:job_count]
51
+ # Set the number of workers unless they are already set to a level we want. Don't scale down here!
52
+ if Scaler.workers <= scale_info[:workers]
53
+ Scaler.workers = scale_info[:workers]
54
+ end
55
+ break # We've set or ensured that the worker count is high enough
56
+ end
57
+ end
58
+ end
59
+ end
metadata ADDED
@@ -0,0 +1,92 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: heroku_dj_auto_scale
3
+ version: !ruby/object:Gem::Version
4
+ hash: 29
5
+ prerelease: false
6
+ segments:
7
+ - 0
8
+ - 0
9
+ - 1
10
+ version: 0.0.1
11
+ platform: ruby
12
+ authors:
13
+ - Joel Basson
14
+ autorequire:
15
+ bindir: bin
16
+ cert_chain: []
17
+
18
+ date: 2011-03-14 00:00:00 +02:00
19
+ default_executable:
20
+ dependencies:
21
+ - !ruby/object:Gem::Dependency
22
+ name: bundler
23
+ prerelease: false
24
+ requirement: &id001 !ruby/object:Gem::Requirement
25
+ none: false
26
+ requirements:
27
+ - - ">="
28
+ - !ruby/object:Gem::Version
29
+ hash: 15424057
30
+ segments:
31
+ - 1
32
+ - 0
33
+ - 0
34
+ - rc
35
+ - 6
36
+ version: 1.0.0.rc.6
37
+ type: :development
38
+ version_requirements: *id001
39
+ description: Allows auto-scaling of delayed job workers in Heroku
40
+ email: []
41
+
42
+ executables: []
43
+
44
+ extensions: []
45
+
46
+ extra_rdoc_files: []
47
+
48
+ files:
49
+ - .gitignore
50
+ - Gemfile
51
+ - Rakefile
52
+ - heroku_dj_auto_scale.gemspec
53
+ - lib/heroku_dj_auto_scale.rb
54
+ - lib/heroku_dj_auto_scale/version.rb
55
+ has_rdoc: true
56
+ homepage: http://rubygems.org/gems/heroku_dj_auto_scale
57
+ licenses: []
58
+
59
+ post_install_message:
60
+ rdoc_options: []
61
+
62
+ require_paths:
63
+ - lib
64
+ required_ruby_version: !ruby/object:Gem::Requirement
65
+ none: false
66
+ requirements:
67
+ - - ">="
68
+ - !ruby/object:Gem::Version
69
+ hash: 3
70
+ segments:
71
+ - 0
72
+ version: "0"
73
+ required_rubygems_version: !ruby/object:Gem::Requirement
74
+ none: false
75
+ requirements:
76
+ - - ">="
77
+ - !ruby/object:Gem::Version
78
+ hash: 23
79
+ segments:
80
+ - 1
81
+ - 3
82
+ - 6
83
+ version: 1.3.6
84
+ requirements: []
85
+
86
+ rubyforge_project: heroku_dj_auto_scale
87
+ rubygems_version: 1.3.7
88
+ signing_key:
89
+ specification_version: 3
90
+ summary: Allows auto-scaling of delayed job workers in Heroku
91
+ test_files: []
92
+