sr71 0.0.2

data/.gitignore

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

data/.rspec

@@ -0,0 +1 @@
+--color --format documentation

data/.rvmrc

@@ -0,0 +1 @@
+rvm use 1.9.3-p194@sr71 --create

data/Gemfile

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

data/LICENSE

@@ -0,0 +1,19 @@
+Copyright (c) 2012 Michael Pilat
+
+Permission is hereby granted, free of charge, to any person obtaining a copy of
+this software and associated documentation files (the "Software"), to deal in
+the Software without restriction, including without limitation the rights to
+use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
+of the Software, and to permit persons to whom the Software is furnished to do
+so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/Rakefile

@@ -0,0 +1,7 @@
+require "bundler/gem_tasks"
+require "rspec/core/rake_task"
+
+RSpec::Core::RakeTask.new(:spec)
+task :default => :spec
+
+task :build => :spec

data/bin/sr71

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+
+require 'sr71'
+require 'sr71/cli'
+
+Sr71::CLI.new.start

data/file

@@ -0,0 +1,26 @@
+require 'spec_helper'
+
+describe Sr71::Service do
+  describe "needs_check?" do
+    it "is true when the check interval has elapsed" do
+      check = Sr71::Service.new("url" => "http://www.example.com", "interval" => 30)
+      check.needs_check?.should be_true
+
+      Time.stub(:now).and_return(15)
+      check.needs_check?.should be_false
+    end
+  end
+
+  describe "start_check" do
+    it "makes an HTTP request to the url" do
+      server = nil
+      EM.run do
+        server = StubServer.new(http_response("200 OK")) { EM.stop }
+
+        check = Sr71::Service.new("url" => "http://127.0.0.1:8081", "interval" => 30)
+        check.start_check
+      end
+      server.request_count.should == 1
+    end
+  end
+end

data/lib/sr71.rb

@@ -0,0 +1,31 @@
+require "eventmachine"
+require "em-http-request"
+
+require "sr71/configuration"
+require "sr71/http_check"
+require "sr71/logger"
+require "sr71/service"
+require "sr71/ticker"
+require "sr71/version"
+
+module Sr71
+  class << self
+    attr_accessor :services
+  end
+
+  self.services = []
+
+  def self.configure(options)
+    @config_file = options[:sr71_config]
+    @config = Configuration.from_file(@config_file)
+    @services = @config.map do | (name, attributes) |
+      Sr71::Service.new(attributes)
+    end
+  end
+
+  def self.start!
+    EM.run do
+      Ticker.start!
+    end
+  end
+end

data/lib/sr71/cli.rb

@@ -0,0 +1,38 @@
+require 'optparse'
+
+module Sr71
+  class CLI
+    class Options
+      def parse!(args)
+        args, options = args.dup, {}
+
+        options[:sr71_config] = '/etc/sr71.yml'
+
+        opt_parser = OptionParser.new do |opts|
+          opts.banner = "Usage: sr71 [options]"
+          opts.on("-c", "--config=file", String, "SR71 configuration file") { |v| options[:sr71_config] = v }
+
+          opts.separator ""
+
+          opts.on("-h", "--help", "Show this help message.") { puts opts; exit }
+        end
+
+        opt_parser.parse! args
+
+        options
+      end
+    end
+
+    def start
+      options = Options.new.parse!(ARGV)
+
+      unless File.exists?(options[:sr71_config])
+        puts "Could not find #{options[:sr71_config]}. Specify correct location with -c file"
+        exit 1
+      end
+
+      Sr71.configure(options)
+      Sr71.start!
+    end
+  end
+end

data/lib/sr71/configuration.rb

@@ -0,0 +1,9 @@
+require 'yaml'
+
+module Sr71
+  class Configuration
+    def self.from_file(file)
+      YAML.load_file(file)
+    end
+  end
+end

data/lib/sr71/http_check.rb

@@ -0,0 +1,58 @@
+module Sr71
+  class HttpCheck
+    class ConnectionTimer
+      attr_reader :connect_time
+      def initialize(&block)
+        @callback = block
+      end
+
+      def request(client, head, body)
+        @callback.call
+        [head, body]
+      end
+
+      def response(resp)
+      end
+    end
+
+    def self.check(url)
+      request = EventMachine::HttpRequest.new(url, {
+        :connect_timeout    => 5,
+        :inactivity_timeout => 30
+      })
+
+      connect_time = nil
+      request.use ConnectionTimer do
+        connect_time = Time.now
+      end
+
+      start_time = Time.now
+      http = request.get
+
+      http.callback do
+        end_time = Time.now
+        Sr71::Logger.log({
+          :type => "http_check",
+          :start_time => start_time,
+          :connect_time => connect_time - start_time,
+          :elapsed_time => end_time - start_time,
+          :end_time => end_time.to_f,
+          :url => url,
+          :status_code => http.response_header.status
+        })
+      end
+
+      http.errback do
+        end_time = Time.now
+        Sr71::Logger.log({
+          :type => "http_check",
+          :start_time => start_time,
+          :elapsed_time => end_time - start_time,
+          :end_time => end_time.to_f,
+          :url => url,
+          :error => http.error
+        })
+      end
+    end
+  end
+end

data/lib/sr71/logger.rb

@@ -0,0 +1,9 @@
+require 'json'
+
+module Sr71
+  class Logger
+    def self.log(hash)
+      puts JSON.generate(hash, :space => ' ', :object_nl => ' ')
+    end
+  end
+end

data/lib/sr71/service.rb

@@ -0,0 +1,18 @@
+module Sr71
+  class Service
+    def initialize(attrs = {})
+      @url        = attrs["url"]
+      @interval   = attrs["interval"]
+      @last_check = 0
+    end
+
+    def needs_check?
+      (@last_check + @interval) <= Time.now.to_f
+    end
+
+    def start_check
+      @last_check = Time.now.to_f
+      HttpCheck.check(@url)
+    end
+  end
+end

data/lib/sr71/ticker.rb

@@ -0,0 +1,13 @@
+module Sr71
+  class Ticker
+    def self.start!
+      @timer = EM::PeriodicTimer.new(0.1, method(:tick))
+    end
+
+    def self.tick
+      Sr71.services.each do |service|
+        service.start_check if service.needs_check?
+      end
+    end
+  end
+end

data/lib/sr71/version.rb

@@ -0,0 +1,3 @@
+module Sr71
+  VERSION = "0.0.2"
+end

data/spec/spec_helper.rb

@@ -0,0 +1,35 @@
+require 'sr71'
+require 'rspec'
+
+Dir.glob("#{File.expand_path('support', File.dirname(__FILE__))}/**/*.rb").each { |f| require f }
+
+TEST_CONFIG = File.expand_path('support/test_config.yml', File.dirname(__FILE__))
+
+def http_response(string)
+  return <<-RESPONSE
+HTTP/1.1 #{string}
+Connection: close
+
+OK
+RESPONSE
+end
+
+def run_in_reactor(timeout = 1)
+  Timeout::timeout(timeout) do
+    EM.epoll
+    EM.run do
+      yield
+    end
+  end
+rescue Timeout::Error
+  fail 'Eventmachine was not stopped before the timeout expired'
+end
+
+def capturing_stdout(&block)
+  old_stdout = $stdout
+  $stdout = StringIO.new
+  yield
+  result = $stdout.string
+  $stdout = old_stdout
+  result
+end

data/spec/sr71/configuration_spec.rb

@@ -0,0 +1,17 @@
+require 'spec_helper'
+
+describe Sr71::Configuration do
+  describe "from_file" do
+    it "reads configuration from a file" do
+      config = Sr71::Configuration.from_file(TEST_CONFIG)
+
+      config["google"].should_not be_nil
+      config["google"]["url"].should == "http://www.google.com"
+      config["google"]["interval"].should == 30
+
+      config["example"].should_not be_nil
+      config["example"]["url"].should == "http://www.example.com"
+      config["example"]["interval"].should == 60
+    end
+  end
+end

data/spec/sr71/http_check_spec.rb

@@ -0,0 +1,12 @@
+require 'spec_helper'
+
+describe Sr71::HttpCheck do
+  describe "check" do
+    it "makes a get request to the URL" do
+      run_in_reactor(5) do
+        server = StubServer.new(http_response("200 OK"), 0.25) { EM.stop }
+        Sr71::HttpCheck.check("http://127.0.0.1:8081")
+      end
+    end
+  end
+end

data/spec/sr71/logger_spec.rb

@@ -0,0 +1,17 @@
+require 'spec_helper'
+
+describe Sr71::Logger do
+  describe "log" do
+    it "logs a request as JSON" do
+      output = capturing_stdout do
+        Sr71::Logger.log({
+          :response => "200 OK",
+          :foo => "bar",
+          :baz => "blah"
+        })
+      end
+
+      output.should == "{ \"response\": \"200 OK\", \"foo\": \"bar\", \"baz\": \"blah\" }\n"
+    end
+  end
+end

data/spec/sr71/service_spec.rb

@@ -0,0 +1,21 @@
+require 'spec_helper'
+
+describe Sr71::Service do
+  describe "needs_check?" do
+    it "is true when the check interval has elapsed" do
+      check = Sr71::Service.new("url" => "http://www.example.com", "interval" => 30)
+      check.needs_check?.should be_true
+
+      Time.stub(:now).and_return(15)
+      check.needs_check?.should be_false
+    end
+  end
+
+  describe "start_check" do
+    it "makes an HTTP request to the url" do
+      service = Sr71::Service.new("url" => "http://127.0.0.1:8081", "interval" => 30)
+      Sr71::HttpCheck.should_receive(:check).with("http://127.0.0.1:8081")
+      service.start_check
+    end
+  end
+end

data/spec/sr71/ticker_spec.rb

@@ -0,0 +1,29 @@
+require 'spec_helper'
+
+describe Sr71::Ticker do
+  describe "start!" do
+    it "schedules to run every tenth of a second" do
+      EventMachine::PeriodicTimer.should_receive(:new).with(0.1, kind_of(Method))
+      Sr71::Ticker.start!
+    end
+  end
+
+  describe "tick" do
+    it "calls start_service for each service" do
+      service = Sr71::Service.new("url" => "http://www.example.com", "interval" => 30)
+      Sr71.services = [service]
+
+      service.should_receive(:start_check)
+      Sr71::Ticker.tick
+    end
+
+    it "does not call start_service if a service is not needed" do
+      service = Sr71::Service.new("url" => "http://www.example.com", "interval" => 30)
+      service.stub(:needs_check?).and_return(false)
+      Sr71.services = [service]
+
+      service.should_receive(:start_check).never
+      Sr71::Ticker.tick
+    end
+  end
+end

data/spec/support/stub_server.rb

@@ -0,0 +1,28 @@
+class StubServer
+  attr_reader :request_count
+
+  module Server
+    attr_accessor :response, :delay, :callback
+
+    def receive_data(data)
+      EM.add_timer(@delay) do
+        send_data @response
+        close_connection_after_writing
+        callback.call
+      end
+    end
+  end
+
+  def initialize (response, delay = 0, port = 8081, host = "127.0.0.1", &block)
+    @sig = EventMachine::start_server(host, port, Server) do |s|
+      s.response = response
+      s.delay = delay
+      s.callback = block
+    end
+  end
+
+  def stop
+    EventMachine.stop_server @sig
+    EM.stop
+  end
+end

data/spec/support/test_config.yml

@@ -0,0 +1,7 @@
+google:
+  url: http://www.google.com
+  interval: 30
+
+example:
+  url: http://www.example.com
+  interval: 60

data/sr71.gemspec

@@ -0,0 +1,25 @@
+# -*- encoding: utf-8 -*-
+$:.push File.expand_path("../lib", __FILE__)
+require "sr71/version"
+
+Gem::Specification.new do |s|
+  s.name        = "sr71"
+  s.version     = Sr71::VERSION
+  s.authors     = ["Michael Pilat"]
+  s.email       = ["mike@mikepilat.com"]
+  s.homepage    = "http://mikepilat.com/"
+  s.summary     = %q{A remote service monitor}
+  s.description = %q{A remote service monitor}
+
+  s.rubyforge_project = "sr71"
+
+  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 "eventmachine",      "~> 1.0.0"
+  s.add_dependency "em-http-request",   "~> 1.0"
+
+  s.add_development_dependency "rspec", "~> 2.11.0"
+end

metadata

@@ -0,0 +1,112 @@
+--- !ruby/object:Gem::Specification
+name: sr71
+version: !ruby/object:Gem::Version
+  version: 0.0.2
+  prerelease: 
+platform: ruby
+authors:
+- Michael Pilat
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-10-26 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: &70343863968620 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 1.0.0
+  type: :runtime
+  prerelease: false
+  version_requirements: *70343863968620
+- !ruby/object:Gem::Dependency
+  name: em-http-request
+  requirement: &70343863968120 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: *70343863968120
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: &70343863967620 !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ~>
+      - !ruby/object:Gem::Version
+        version: 2.11.0
+  type: :development
+  prerelease: false
+  version_requirements: *70343863967620
+description: A remote service monitor
+email:
+- mike@mikepilat.com
+executables:
+- sr71
+extensions: []
+extra_rdoc_files: []
+files:
+- .gitignore
+- .rspec
+- .rvmrc
+- Gemfile
+- LICENSE
+- Rakefile
+- bin/sr71
+- file
+- lib/sr71.rb
+- lib/sr71/cli.rb
+- lib/sr71/configuration.rb
+- lib/sr71/http_check.rb
+- lib/sr71/logger.rb
+- lib/sr71/service.rb
+- lib/sr71/ticker.rb
+- lib/sr71/version.rb
+- spec/spec_helper.rb
+- spec/sr71/configuration_spec.rb
+- spec/sr71/http_check_spec.rb
+- spec/sr71/logger_spec.rb
+- spec/sr71/service_spec.rb
+- spec/sr71/ticker_spec.rb
+- spec/support/stub_server.rb
+- spec/support/test_config.yml
+- sr71.gemspec
+homepage: http://mikepilat.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: sr71
+rubygems_version: 1.8.15
+signing_key: 
+specification_version: 3
+summary: A remote service monitor
+test_files:
+- spec/spec_helper.rb
+- spec/sr71/configuration_spec.rb
+- spec/sr71/http_check_spec.rb
+- spec/sr71/logger_spec.rb
+- spec/sr71/service_spec.rb
+- spec/sr71/ticker_spec.rb
+- spec/support/stub_server.rb
+- spec/support/test_config.yml