nugget 0.0.10 → 0.0.11

data/README.md

@@ -18,10 +18,10 @@ Nugget also includes a very basic web service daemon that simply reads the curre
 
         $ nugget -w
 
-Additionally, nugget includes support for sending results of the tests to [backstop](https://github.com/obfuscurity/backstop).
+Additionally, nugget includes support for sending results of the tests to [backstop](https://github.com/obfuscurity/backstop) and [statsd](https://github.com/reinh/statsd).
 
 ![](https://dl.dropboxusercontent.com/s/1xhl7fnuw0y0934/2014-01-13%20at%207.47%20PM.png)
 
 #### License
 
-nugget is open source software available under the MIT License
+nugget is open source software available under the MIT License

data/bin/check_nugget

@@ -23,9 +23,25 @@ end
 
 failed_tests = {}
 
-results.each do |test, info|
-  if info[:result] == "FAIL"
-    failed_tests.store(test.to_s, info[:response][:failed])
+if ARGV[0] == "-s"
+  if ARGV[1]
+    if results[ARGV[1].to_sym]
+      if results[ARGV[1].to_sym][:result] == "FAIL"
+        failed_tests.store(ARGV[1], results[ARGV[1].to_sym][:response][:failed])
+      end
+    else
+      puts "no check by that name"
+      exit 1
+    end
+  else
+    puts "-s requires an argument in the form of a check name"
+    exit 1
+  end
+else
+  results.each do |test, info|
+    if info[:result] == "FAIL"
+      failed_tests.store(test.to_s, info[:response][:failed])
+    end
   end
 end
 

data/lib/nugget.rb

@@ -6,8 +6,10 @@ require 'mixlib/config'
 require 'mixlib/log'
 require 'yajl/json_gem'
 require 'thin'
+require 'statsd'
 require 'open-uri'
 require 'iconv' if RUBY_VERSION =~ /1.8/
+require 'timeout'
 
 __DIR__ = File.dirname(__FILE__)
 
@@ -21,8 +23,11 @@ require 'nugget/cli'
 require 'nugget/service'
 require 'nugget/web'
 require 'nugget/backstop'
+require 'nugget/statsd'
 require 'nugget/version'
 
+TIMEOUT = 180
+
 module  Nugget
   class << self
 

data/lib/nugget/backstop.rb

@@ -10,6 +10,8 @@ module Nugget
 
       if result == "FAIL"
         backstop_requst("#{name}.result", 1)
+      else
+        backstop_requst("#{name}.result", 0)
       end
 
     end
@@ -51,4 +53,4 @@ module Nugget
     end
 
   end
-end
+end

data/lib/nugget/cli.rb

@@ -55,10 +55,25 @@ module Nugget
       :long => "--backstop URL",
       :description => "URL for backstop metrics thing."
 
+    option :statsd_namespace,
+      :short => "-s NAMESPACE",
+      :long => "--statsdnamespace NAMESPACE",
+      :description => "statsd namespace."
+
+    option :statsd_host,
+      :short => "-h HOST",
+      :long => "--statsdhost HOST",
+      :description => "statsd host."
+
+    option :statsd_port,
+      :short => "-a PORT",
+      :long => "--statsdport PORT",
+      :description => "statsd port."
+
     option :resultsfile,
       :short => "-r FILE",
       :long => "--results FILE",
-      :default => "/tmp/nugget_results.json",
+      :default => false,
       :description => "Path to where results file is written/read."
 
     option :help,

data/lib/nugget/service.rb

@@ -5,9 +5,11 @@ module Nugget
       Nugget::Log.info("Starting up Nugget in daemon mode ...")
 
       loop do
+        Nugget::Log.debug("Running tests ...")
         run(test_name)
 
         # chill
+        Nugget::Log.debug("Sleeping for #{Nugget::Config.interval.to_i} ...")
         sleep(Nugget::Config.interval.to_i)
       end
     end
@@ -23,6 +25,7 @@ module Nugget
       config = parser.parse(config_file)
 
       results = Hash.new
+      threadlist = Array.new
 
       if test_name
         if definition = config[test_name.to_s.to_sym]
@@ -32,13 +35,15 @@ module Nugget
         end
       else
         config.each do |test, definition|
-          run_test(results, test, definition)
+          threadlist << Thread.new { run_test(results, test, definition) }
         end
       end
 
-      if Nugget::Config.daemon
-        Nugget::Service.write_results(results)
-      end
+      threadlist.each { |x|
+          x.join
+      }
+
+      Nugget::Service.write_results(results)
     end
 
     def self.run_test(results, test, definition)
@@ -49,8 +54,17 @@ module Nugget
         request_definition = config_converter(definition)
         response_definition = definition[:response]
 
-        response = Turd.run(request_definition, response_definition)
+        status = Timeout::timeout(TIMEOUT) {
+          Nugget::Log.debug("Asserting turd definitions ...")
+          response = Turd.run(request_definition, response_definition)
+        }
         result = "PASS"
+      rescue Timeout::Error => e
+        Nugget::Log.error("#{definition[:type]} test #{test} took too long to run (#{TIMEOUT}s)!")
+        Nugget::Log.error(e)
+
+        result = "FAIL"
+        response = "timeout"
       rescue Exception => e
         Nugget::Log.error("#{definition[:type]} test #{test} failed due to #{e.response[:failed]}!")
         Nugget::Log.error(e)
@@ -68,9 +82,7 @@ module Nugget
         :timestamp => Time.now.to_i
       })
 
-      if Nugget::Config.backstop_url
-        Nugget::Backstop.send_metrics(test, result, response)
-      end
+      send_metrics(test, result, response)
     end
 
     def self.config_converter(definition)
@@ -95,14 +107,27 @@ module Nugget
 
     def self.write_results(results)
       begin
-        file = File.open(Nugget::Config.resultsfile, "w")
-        file.puts(results.to_json)
-        file.close
+        if Nugget::Config.resultsfile
+          Nugget::Log.debug("Writing results to #{Nugget::Config.resultsfile} ...")
+          file = File.open(Nugget::Config.resultsfile, "w")
+          file.puts(results.to_json)
+          file.close
+        end
       rescue Exception => e
         Nugget::Log.error("Something went wrong with writing out the results file!")
         Nugget::Log.error(e)
       end
     end
 
+    def self.send_metrics(test, result, response)
+      if Nugget::Config.backstop_url
+        Nugget::Backstop.send_metrics(test, result, response)
+      end
+
+      if Nugget::Config.statsd_host
+        Nugget::NStatsd.send_metrics(test, result, response)
+      end
+    end
+
   end
 end

data/lib/nugget/statsd.rb

@@ -0,0 +1,40 @@
+module Nugget
+  class NStatsd # NStatsd since Statsd collides
+
+    def self.stats
+      @stats ||= Statsd.new(Nugget::Config.statsd_host, Nugget::Config.statsd_port).tap do |statsd|
+        statsd.namespace =  Nugget::Config.statsd_namespace
+      end
+    end
+
+    def self.send_metrics(name, result, response)
+      statsd = Nugget::NStatsd.stats
+      send_test_result(statsd, name, result)
+      send_test_timings(statsd, name, response)
+    end
+
+    def self.send_test_result(statsd, name, result)
+
+      if result == "FAIL"
+        statsd.gauge("#{name}.failures.count", 1)
+        Nugget::Log.debug("Sending the following to statsd: #{name}_failure_count: 1")
+      else
+        statsd.gauge("#{name}.failures.count", 0)
+        Nugget::Log.debug("Sending the following to statsd: #{name}_failure_count: 0")
+      end
+
+    end
+
+    def self.send_test_timings(statsd, name, response)
+
+      response.each do |key, value|
+        if key.to_s.include?("_time")
+          Nugget::Log.debug("Sending the following to statsd: #{key}: #{value}")
+          statsd.timing("#{name}.#{key}", value)
+        end
+      end
+
+    end
+
+  end
+end

data/lib/nugget/version.rb

@@ -1,3 +1,3 @@
 module Nugget
-  VERSION = "0.0.10"
+  VERSION = "0.0.11"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: nugget
 version: !ruby/object:Gem::Version
-  version: 0.0.10
+  version: 0.0.11
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-05-06 00:00:00.000000000 Z
+date: 2014-10-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mixlib-config
@@ -107,6 +107,22 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: statsd-ruby
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -155,6 +171,7 @@ files:
 - lib/nugget/config.rb
 - lib/nugget/log.rb
 - lib/nugget/service.rb
+- lib/nugget/statsd.rb
 - lib/nugget/version.rb
 - lib/nugget/web.rb
 - lib/nugget.rb