what 0.0.1

data/.gitignore

@@ -0,0 +1,4 @@
+*.gem
+.bundle
+Gemfile.lock
+pkg/*

data/Gemfile

@@ -0,0 +1,4 @@
+source "http://rubygems.org"
+
+# Specify your gem's dependencies in what.gemspec
+gemspec

data/Rakefile

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

data/bin/what

@@ -0,0 +1,28 @@
+require 'rubygems'
+require 'optparse'
+require 'what'
+
+opts = OptionParser.new do |opts|
+  opts.banner = "Usage: what [options]"
+  opts.separator ""
+
+  opts.on('-c', '--config FILE', 'Specify the location of the YAML config file.') do |fn|
+    What::Config.load(fn)
+  end
+
+  opts.on_tail('-h', '--help', 'Show this message.') do
+    puts opts
+    exit
+  end
+end
+opts.parse!(ARGV)
+
+unless What::Config.loaded?
+  puts opts
+  exit
+end
+
+Thin::Server.start('0.0.0.0', 8164) do
+  run What::Server.new
+end
+

data/example/what.yml

@@ -0,0 +1,27 @@
+# Config file for the What monitoring system.
+
+### Set the polling interval for all services.
+interval: 10
+
+### Specify paths containing custom modules to load, relative to this file.
+# module_paths:
+# - modules
+
+### List all modules to be monitored.
+modules:
+- unicorn
+# - ram
+# - redis
+# - dir
+# - whatever
+
+### Pass parameters into modules.
+# module_config:
+#   ram:
+#     min: nil
+#   dir:
+#     path: '/mnt/datastore'
+
+### Specify any other config files to load.
+# configs:
+# - more_config/stuff.yml

data/lib/what/config.rb

@@ -0,0 +1,43 @@
+class What::Config
+  @config = {}
+
+  def self.load(fn)
+    load_primary(fn)
+    load_secondary(@config['configs'])
+  end
+
+  def self.load_primary(fn)
+    @config = YAML.load_file(fn)
+    @config['base'] ||= File.expand_path(File.dirname(fn))
+    @loaded = true
+  end
+
+  def self.load_secondary(fns)
+    return if !fns
+
+    fns.each do |fn|
+      path = if fn.match(/^\//)
+               fn
+             else
+               File.join(@config['base'], fn)
+             end
+      @config.merge!(YAML.load_file(path))
+    end
+  end
+
+  def self.loaded?
+    @loaded
+  end
+
+  def self.[](attr)
+    @config[attr]
+  end
+
+  def self.[]=(attr, val)
+    @config[attr] = val
+  end
+
+  def self.all
+    @config
+  end
+end

data/lib/what/helpers.rb

@@ -0,0 +1,20 @@
+module What::Helpers
+  # Stolen from Rails (http://api.rubyonrails.org/classes/ActiveSupport/Inflector.html)
+  def self.camelize(lower_case_and_underscored_word, first_letter_in_uppercase = true)
+    if first_letter_in_uppercase
+      lower_case_and_underscored_word.to_s.gsub(/\/(.?)/) { "::#{$1.upcase}" }.gsub(/(?:^|_)(.)/) { $1.upcase }
+    else
+      lower_case_and_underscored_word.to_s[0].chr.downcase + camelize(lower_case_and_underscored_word)[1..-1]
+    end
+  end
+
+  def self.underscore(camel_cased_word)
+    word = camel_cased_word.to_s.dup
+    word.gsub!(/::/, '/')
+    word.gsub!(/([A-Z]+)([A-Z][a-z])/,'\1_\2')
+    word.gsub!(/([a-z\d])([A-Z])/,'\1_\2')
+    word.tr!("-", "_")
+    word.downcase!
+    word
+  end
+end

data/lib/what/modules/base.rb

@@ -0,0 +1,26 @@
+class What::Modules::Base
+  def initialize
+    @config = What::Config[self.name]
+  end
+
+  def name
+    What::Helpers.underscore(self.class.name.split('::').last)
+  end
+
+  def check!
+    raise "Module #{self.class.name} doesn't override 'check!'"
+  end
+
+  def status
+    { :health => health }.merge(details)
+  end
+
+  def health
+    raise "Module #{self.class.name} doesn't override 'health'"
+  end
+
+  def details
+    {}
+  end
+end
+

data/lib/what/modules/unicorn.rb

@@ -0,0 +1,22 @@
+class What::Modules::Unicorn < What::Modules::Base
+  @unicorns = []
+
+  def check!
+    @unicorns = `ps aux`.grep(/unicorn_rails worker/).map do |ln|
+                  ln =~ /^\w+\s+(\d+).*(\d+:\d\d(?:\.\d\d)?) unicorn/
+                  {:pid => $1, :cpu_time => $2}
+                end
+  end
+
+  def health
+    if @unicorns.count > 0
+      :ok
+    else
+      :alert
+    end
+  end
+
+  def details
+    {:workers => @unicorns.count, :details => @unicorns}
+  end
+end

data/lib/what/modules.rb

@@ -0,0 +1,19 @@
+module What::Modules
+  # load all modules defined in what/modules, in addition to any paths
+  # specified in the config file.
+  def self.load_all
+    globs = [File.join(File.dirname(__FILE__), 'modules', '*.rb')]
+
+    if What::Config['module_paths']
+      What::Config['module_paths'].each do |module_path|
+        globs << File.join(What::Config['base'], module_path, '*.rb')
+      end
+    end
+
+    globs.each do |glob|
+      Dir[glob].each do |fn|
+        require fn
+      end
+    end
+  end
+end

data/lib/what/monitor.rb

@@ -0,0 +1,30 @@
+class What::Monitor
+  def self.go!
+    @modules = What::Config['modules'].map do |m|
+      name = What::Helpers.camelize(m)
+      What::Modules.const_get(name).new
+    end
+
+    Thread.abort_on_exception = true
+    @thread = Thread.new(@modules) { |modules| self.do_it(modules) }
+  end
+
+  def self.do_it(modules)
+    loop do
+      overall = :ok
+      modules.each do |mod|
+        mod.check!
+        case mod.health
+          when :ok
+          when :warning
+            overall = :warning if overall != :alert
+          else
+            overall = :alert
+        end
+        What::Status[mod.name] = mod.status
+      end
+      What::Status[:health] = overall
+      sleep What::Config['interval']
+    end
+  end
+end

data/lib/what/server.rb

@@ -0,0 +1,14 @@
+class What::Server
+  def initialize
+    What::Modules.load_all
+    What::Monitor.go!
+  end
+
+  def call(_)
+    [
+      What::Status[:health] != :alert ? 200 : 503,
+      {'Content-Type' => 'application/json'},
+      JSON.unparse(What::Status.all) + "\n"
+    ]
+  end
+end

data/lib/what/status.rb

@@ -0,0 +1,15 @@
+class What::Status
+  @status = {}
+
+  def self.[](attr)
+    @status[attr]
+  end
+
+  def self.[]=(attr, val)
+    @status[attr] = val
+  end
+
+  def self.all
+    @status
+  end
+end

data/lib/what/version.rb

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

data/lib/what.rb

@@ -0,0 +1,14 @@
+module What
+end
+
+require 'json'
+require 'yaml'
+require 'thin'
+
+require 'what/version'
+require 'what/helpers'
+require 'what/config'
+require 'what/status'
+require 'what/modules'
+require 'what/monitor'
+require 'what/server'

data/what.gemspec

@@ -0,0 +1,26 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "what/version"
+
+Gem::Specification.new do |s|
+  s.name        = "what"
+  s.version     = What::VERSION
+  s.platform    = Gem::Platform::RUBY
+  s.authors     = ["Ryan Lower", "Ryan Fitzgerald"]
+  s.email       = ["team@academia.edu"]
+  s.homepage    = "http://academia.edu/"
+  s.summary     = %q{Modular server monitoring with JSON endpoint}
+  s.description = %q{What runs a Thin server which returns a JSON object representing
+the state of services running on a machine. It currently only includes a
+module for monitoring Unicorn workers, but it's easy to add custom modules.}
+
+  s.rubyforge_project = "what"
+
+  s.files         = `git ls-files`.split("\n")
+  s.test_files    = `git ls-files -- {test,spec,features}/*`.split("\n")
+  s.executables   = `git ls-files -- bin/*`.split("\n").map{ |f| File.basename(f) }
+  s.require_paths = ["lib"]
+
+  s.add_dependency("thin")
+  s.add_dependency("json")
+end

metadata

@@ -0,0 +1,114 @@
+--- !ruby/object:Gem::Specification 
+name: what
+version: !ruby/object:Gem::Version 
+  hash: 29
+  prerelease: 
+  segments: 
+  - 0
+  - 0
+  - 1
+  version: 0.0.1
+platform: ruby
+authors: 
+- Ryan Lower
+- Ryan Fitzgerald
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2011-05-10 00:00:00 -07:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thin
+  prerelease: false
+  requirement: &id001 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id001
+- !ruby/object:Gem::Dependency 
+  name: json
+  prerelease: false
+  requirement: &id002 !ruby/object:Gem::Requirement 
+    none: false
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        hash: 3
+        segments: 
+        - 0
+        version: "0"
+  type: :runtime
+  version_requirements: *id002
+description: |-
+  What runs a Thin server which returns a JSON object representing
+  the state of services running on a machine. It currently only includes a
+  module for monitoring Unicorn workers, but it's easy to add custom modules.
+email: 
+- team@academia.edu
+executables: 
+- what
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- .gitignore
+- Gemfile
+- Rakefile
+- bin/what
+- example/what.yml
+- lib/what.rb
+- lib/what/config.rb
+- lib/what/helpers.rb
+- lib/what/modules.rb
+- lib/what/modules/base.rb
+- lib/what/modules/unicorn.rb
+- lib/what/monitor.rb
+- lib/what/server.rb
+- lib/what/status.rb
+- lib/what/version.rb
+- what.gemspec
+has_rdoc: true
+homepage: http://academia.edu/
+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: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: what
+rubygems_version: 1.5.1
+signing_key: 
+specification_version: 3
+summary: Modular server monitoring with JSON endpoint
+test_files: []
+