hirefireapp 0.0.1

data/.gitignore

@@ -0,0 +1,2 @@
+.DS_Store
+*.gem

data/README.md

@@ -0,0 +1,5 @@
+# HireFireApp.com - The Heroku Worker Monitor
+
+See the official website for more information!
+
+[http://hirefireapp.com/](http://hirefireapp.com/)

data/bin/hirefireapp

@@ -0,0 +1,39 @@
+#! /usr/bin/env ruby
+
+require 'open-uri'
+
+def usage
+  puts
+  puts "Usage:"
+  puts
+  puts "  hirefireapp http://mydomain.com/"
+  puts
+  puts "Or locally:"
+  puts
+  puts "  gem install thin"
+  puts "  thin start -p 3000"
+  puts "  hirefireapp http://127.0.0.1:3000/"
+  puts
+  puts "SSL Enabled URLs:"
+  puts
+  puts "  hirefireapp https://mydomain.com/"
+end
+
+if (url = ARGV[0]).nil?
+  usage
+else
+  begin
+    response = open(File.join(url, 'hirefireapp', 'test')).read
+  rescue
+    puts
+    puts "Could not connect to: #{url}"
+    usage
+    exit 1
+  end
+
+  if response =~ /\[HireFireApp\:\s/
+    puts response
+  else
+    puts "Could not find HireFireApp at #{url}."
+  end
+end

data/hirefireapp.gemspec

@@ -0,0 +1,20 @@
+# encoding: utf-8
+
+Gem::Specification.new do |gem|
+
+  # General configuration / information
+  gem.name        = 'hirefireapp'
+  gem.version     = '0.0.1'
+  gem.platform    = Gem::Platform::RUBY
+  gem.authors     = 'Michael van Rooijen'
+  gem.email       = 'meskyanichi@gmail.com'
+  gem.homepage    = 'http://hirefireapp.com/'
+  gem.summary     = %|HireFireApp.com - The Heroku Worker Monitor - Save money and scale at the same time!|
+  gem.description = %|HireFireApp.com - The Heroku Worker Monitor - Save money and scale at the same time! We monitor your applications by the minute!|
+
+  # Files and folder that need to be compiled in to the Ruby Gem
+  gem.files         = %x[git ls-files].split("\n")
+  gem.executables   = ['hirefireapp']
+  gem.require_path  = 'lib'
+
+end

data/lib/hirefireapp/middleware.rb

@@ -0,0 +1,165 @@
+# encoding: utf-8
+
+module HireFireApp
+  class Middleware
+
+    ##
+    # Initialize the Rack Middleware by setting the app instance
+    # as well as allowing HireFire to request information via the token uri.
+    def initialize(app)
+      @app   = app
+      @token = ENV['HIREFIREAPP_TOKEN']
+    end
+
+    ##
+    # If we are currently in the Heroku environment, and the path to the HireFire info uri
+    # has been requested (and the token belongs to the app) then we'll calculate the amount of
+    # jobs that are currently pending (either Delayed Job with Active Record / Mongoid or Resque with Redis).
+    #
+    # Once the job_count has been determined, we build a simple JSON string object and
+    # create a rack-based json response with 200 status. This will be returned to the HireFire service
+    # in order to determine what actions to take in terms of scaling up or down.
+    def call(env)
+      @env = env
+
+      if test?
+        [ 200, {'Content-Type' => 'text/html'}, self ]
+      elsif info?
+        [ 200, {'Content-Type' => 'application/json'}, self ]
+      else
+        @app.call(env)
+      end
+    end
+
+    ##
+    # Response body
+    def each(&block)
+      if test?
+        block.call "[HireFireApp: #{ok}] Worker: #{worker} - Mapper: #{mapper}"
+      elsif info?
+        block.call %|{"job_count":#{job_count || 'null'}}|
+      end
+    end
+
+    private
+
+    ##
+    # Counts the amount of jobs that are currently queued
+    # and show be processed as soon as possible (aka the ones that are pending)
+    #
+    # @returns [Fixnum, nil] job_count returns nil if something went wrong
+    def job_count
+      begin
+        if defined?(Delayed::Worker)
+          count_delayed_job
+        elsif defined?(Resque)
+          count_resque
+        end
+      rescue => error
+        nil
+      end
+    end
+
+    ##
+    # Makes Delayed::Job count the amount of currently pending jobs.
+    # It'll use the ActiveRecord ORM, or the Mongoid ODM depending on
+    # which of them is defined.
+    #
+    # If ActiveRecord 2 (or earlier) is being used, ActiveRecord::Relation doesn't
+    # exist, and we'll have to use the old :conditions hash notation.
+    #
+    # @returns [Fixnum] delayed_job_count the amount of jobs currently pending
+    def count_delayed_job
+      if defined?(ActiveRecord) and Delayed::Worker.backend.to_s =~ /ActiveRecord/
+        if defined?(ActiveRecord::Relation)
+          Delayed::Job.
+          where(:failed_at => nil).
+          where('run_at <= ?', Time.now).count
+        else
+          Delayed::Job.all(
+            :conditions => [
+              'failed_at IS NULL and run_at <= ?', Time.now.utc
+            ]
+          ).count
+        end
+      elsif defined?(Mongoid) and Delayed::Worker.backend.to_s =~ /Mongoid/
+        Delayed::Job.where(
+          :failed_at  => nil,
+          :run_at.lte => Time.now
+        ).count
+      end
+    end
+
+    ##
+    # Makes Resque count the amount of currently pending jobs.
+    #
+    # @returns [Fixnum] resque_job_count
+    #  the number of jobs pending + the amount of workers currently working
+    def count_resque
+      Resque.info[:pending].to_i + Resque.info[:working].to_i
+    end
+
+    ##
+    # Returns the name of the mapper, or "Not Found" if not found
+    #
+    # @returns [String]
+    def mapper
+      if defined?(Redis) and defined?(Resque)
+        "Redis"
+      elsif defined?(Delayed::Worker)
+        if defined?(ActiveRecord) and Delayed::Worker.backend.to_s =~ /ActiveRecord/
+          "Active Record"
+        elsif defined?(Mongoid) and Delayed::Worker.backend.to_s =~ /Mongoid/
+          "Mongoid"
+        else
+          "Not Found"
+        end
+      else
+        "Not Found"
+      end
+    end
+
+    ##
+    # Returns the name of the worker type, or "Not Found" if not found
+    #
+    # @returns [String]
+    def worker
+      if defined?(Delayed::Job)
+        "Delayed Job"
+      elsif defined?(Resque)
+        "Resque"
+      else
+        "Not Found"
+      end
+    end
+
+    ##
+    # Returns "OK" if both the mapper and worker were found
+    #
+    # @returns [String]
+    def ok
+      if mapper =~ /Not Found/ or worker =~ /Not Found/
+        "Incomplete"
+      else
+        "OK"
+      end
+    end
+
+    ##
+    # Returns true if the REQUEST_PATH matches the test url
+    #
+    # @returns [String]
+    def test?
+      @env['REQUEST_PATH'] == "/hirefireapp/test"
+    end
+
+    ##
+    # Returns true if the REQUEST_PATH matches the info url
+    #
+    # @returns [String]
+    def info?
+      @env['REQUEST_PATH'] == "/hirefireapp/#{@token}/info"
+    end
+
+  end
+end

data/lib/hirefireapp/railtie.rb

@@ -0,0 +1,9 @@
+# encoding: utf-8
+
+module HireFireApp
+  class Railtie < ::Rails::Railtie
+    initializer 'hirefireapp.add_middleware' do |app|
+      app.config.middleware.use 'HireFireApp::Middleware'
+    end
+  end
+end

data/lib/hirefireapp.rb

@@ -0,0 +1,12 @@
+# encoding: utf-8
+
+HIREFIRE_APP_PATH = File.expand_path('../hirefireapp', __FILE__)
+
+# Load the HireFireApp middleware
+require File.join(HIREFIRE_APP_PATH, 'middleware')
+
+# If Rails::Railtie exists, then hook up HireFireApp automatically
+if defined?(Rails::Railtie)
+  require File.join(HIREFIRE_APP_PATH, 'railtie')
+end
+

metadata

@@ -0,0 +1,62 @@
+--- !ruby/object:Gem::Specification 
+name: hirefireapp
+version: !ruby/object:Gem::Version 
+  prerelease: 
+  version: 0.0.1
+platform: ruby
+authors: 
+- Michael van Rooijen
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-05-23 00:00:00 +02:00
+default_executable: 
+dependencies: []
+
+description: HireFireApp.com - The Heroku Worker Monitor - Save money and scale at the same time! We monitor your applications by the minute!
+email: meskyanichi@gmail.com
+executables: 
+- hirefireapp
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- README.md
+- bin/hirefireapp
+- hirefireapp.gemspec
+- lib/hirefireapp.rb
+- lib/hirefireapp/middleware.rb
+- lib/hirefireapp/railtie.rb
+has_rdoc: true
+homepage: http://hirefireapp.com/
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.6.2
+signing_key: 
+specification_version: 3
+summary: HireFireApp.com - The Heroku Worker Monitor - Save money and scale at the same time!
+test_files: []
+