sensu-run-check 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 4c955968a850e224f8c594fa95232dbce3753c7a
-  data.tar.gz: 6edb3a30525201462fccde5ead01242d8fc9bacd
+  metadata.gz: 124fa69d27a4dc7394117818fc7ae64f01a349f4
+  data.tar.gz: c8c8073b37ca4ccce499be14337a9dab581232f7
 SHA512:
-  metadata.gz: 830cb80b44157a7212ff4b10ac928a0f7169fc52f82bc9bf57450c871758a14dc03ff0b655c2d02336e0b8f50dde568c89a50e2a1d35e92fa13b37666e3e06d7
-  data.tar.gz: ea68e6f66aaf102e246197aebb41dbfc62c3da8db7e19aa6452ac8ed9433ecb2602c4aad8654f23d17af2ddca526b596d2d75d22f3d869aaf51be7f9cef3ea56
+  metadata.gz: 46b7d5a9c7cd60eae203612e285364b47eaa0f0d06951f5ce8b8bdb4a5cf97a0d18415865a8b64749529aec958e6def984586397b3a93981dcae0cfddadd13ca
+  data.tar.gz: 6b679510da951ea62c99d9036ed409f6f29b36f0295385c43c51e77f6d2c1a7665d2f1b5cb69c449c1be1499ce78bcab55fc081fa6ba91a40a9cb0ba2f9a5620

data/README.md

@@ -1,6 +1,6 @@
 # Sensu Run Check
 
-Often after just doing a change on servers you want to just be sure that they’re all going to pass a certain or all Sensu checks. This gem exposes Sensus checks to be executed local on the command line.
+Often after just doing a change on servers you want to just be sure that they’re all going to pass a certain or all (Sensu)[http://sensuapp.org] checks. This gem exposes Sensu checks to be executed local on the command line.
 
 _WARNING_ This is very much a hack and will break with other versions* of Sensu!
 
@@ -44,4 +44,4 @@ Run all checks defined on this host.
 
 (c) 2015 - Rickard von Essen
 
-Released under the MIT license, see LICENSE.txt
+Released under the MIT license, see LICENSE.txt

data/lib/sensu-run-check/runner.rb

@@ -1,5 +1,6 @@
 require "rubygems"
 
+require "open3"
 require "sensu-run-check/version"
 require "sensu/client/process"
 require "sensu/settings"
@@ -34,6 +35,7 @@ module SensuRunCheck
         statuses = []
         s.get_all_checks.collect{ |check| check[:name] }.sort.each do |checkname|
           stdout, status = s.run_check(checkname)
+          status ||= 3
           statuses << status
           puts "#{checkname} #{status} #{stdout}"
         end
@@ -41,6 +43,7 @@ module SensuRunCheck
       else
         stdout, status = s.run_check(options[:run_check])
         puts stdout
+        status ||= 3
         exit(status)
       end
     end
@@ -57,6 +60,47 @@ module SensuRunCheck
       @logger = SensuRunCheck::NilLog.new
     end
 
+    # Execute a check command, capturing its output (STDOUT/ERR),
+    # exit status code, execution duration, timestamp, and publish
+    # the result. This method guards against multiple executions for
+    # the same check. Check command tokens are substituted with the
+    # associated client attribute values. If there are unmatched
+    # check command tokens, the check command will not be executed,
+    # instead a check result will be published reporting the
+    # unmatched tokens.
+    #
+    # @param check [Hash]
+    def execute_check_command(check)
+      @logger.debug("attempting to execute check command", :check => check)
+      unless @checks_in_progress.include?(check[:name])
+        @checks_in_progress << check[:name]
+        command, unmatched_tokens = substitute_check_command_tokens(check)
+        if unmatched_tokens.empty?
+          check[:executed] = Time.now.to_i
+          started = Time.now.to_f
+
+          output, status = Open3.capture2(command)
+          check[:duration] = ("%.3f" % (Time.now.to_f - started)).to_f
+          check[:output] = output
+          check[:status] = status.exitstatus
+          @checks_in_progress.delete(check[:name])
+          publish_check_result(check)
+        else
+          check[:output] = "Unmatched command tokens: " + unmatched_tokens.join(", ")
+          check[:status] = 3
+          check[:handle] = false
+          @checks_in_progress.delete(check[:name])
+          publish_check_result(check)
+        end
+      else
+        @logger.warn("previous check command execution in progress", :check => check)
+      end
+    end
+
+    def publish_check_result(check)
+      return check[:output], check[:status]
+    end
+
     # Find a Sensu check by name.
     #
     # @param [String] name of the check
@@ -84,16 +128,20 @@ module SensuRunCheck
         execute_check_command(check)
       end
     end
-
   end
 
   # Swallow logs
   class NilLog
-    def debug(*ignore)
-      # Ignore
+    def initialize
+      self.class.create_level_methods
     end
-    def warn(*ignore)
-      # Ignore
+
+    def self.create_level_methods
+      Sensu::Logger::LEVELS.each do |level|
+        define_method(level) do |*args|
+          # Do nothing
+        end
+      end
     end
   end
 end

data/lib/sensu-run-check/version.rb

@@ -1,3 +1,3 @@
 module SensuRunCheck
-  VERSION = '0.1.0'
+  VERSION = '0.2.0'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sensu-run-check
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Rickard von Essen
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2015-04-15 00:00:00.000000000 Z
+date: 2015-04-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sensu