nagios-splunk 0.0.1 → 0.0.2

data/lib/nagios/splunk.rb

@@ -11,6 +11,9 @@ module Nagios
     autoload :Check, "nagios/splunk/check"
     autoload :RestClient, "nagios/splunk/rest_client"
     autoload :CLI, "nagios/splunk/cli"
+    autoload :Exception, "nagios/splunk/exception"
+    autoload :NoPoolsFound, "nagios/splunk/exception"
+    autoload :NoLicensesFound, "nagios/splunk/exception"
 
   end
 end

data/lib/nagios/splunk/check.rb

@@ -15,43 +15,117 @@ module Nagios
         @rest_client = client
       end
 
+      # pool usage check
+      # @param [String] name of the pool
+      # @param [Integer] warn license usage threshhold
+      # @param [Integer] crit license usage threshhold
+      # @param [String] stack_id which pool is assigned to
+      # @return [Array<Integer, String>] exit code and message
+      def pool_usage(name, warn, crit)
+        quota = pool_quota(name)
+        used_quota = find_pool_by_name(name)["used_bytes"].to_i
+
+        code = check_threshold(used_quota, quota, warn, crit)
+
+        message = "License pool '#{name}' #{STATUS[code]}: #{used_quota * 100 / quota}% of license pool capacity is used"
+        message << " | quota: #{quota} B; used: #{used_quota} B"
+        return [code, message]
+      end
+
       # license usage check
       # @param [Integer] warn license usage threshhold
       # @param [Integer] crit license usage threshhold
+      # @param [String] stack_id
       # @return [Array<Integer, String>] exit code and message
-      def license_usage(warn, crit)
-        quota = licenses.
-          select {|k,v| v["type"] == "enterprise" && v["status"] == "VALID" }.
-          reduce(0) { |r,l| r+=l[1]["quota"].to_i }
-        used_quota = pools.reduce(0) {|r,p| r+=p[1]["used_bytes"].to_i }
-        case true
-        when used_quota > quota * crit.to_i / 100:
-            code = 2
-        when used_quota > quota * warn.to_i / 100:
-            code = 1
-        else
-          code = 0
-        end
+      def license_usage(warn, crit, stack_id = "enterprise")
+        quota = license_quota(stack_id)
+        used_quota = pools_in_stack_usage(stack_id)
+
+        code = check_threshold(used_quota, quota, warn, crit)
+
         message = "License #{STATUS[code]}: #{used_quota * 100 / quota}% of license capacity is used"
         message << " | quota: #{quota} B; used: #{used_quota} B"
         return [code, message]
       end
 
+      private
+
+      # Compare usage and quota with threshold
+      # @param [Integer] usage
+      # @param [Integer] quota
+      # @param [Integer] warn
+      # @param [Integer] crit
+      # @return [Integer] exit code
+      def check_threshold(usage, quota, warn, crit)
+        case true
+        when usage > quota * crit.to_i / 100:
+          return 2
+        when usage > quota * warn.to_i / 100:
+          return 1
+        else
+          return 0
+        end
+      end
+
+      # Find out pool quota
+      # @param [String] name of the pool
+      # @return [Integer] quota number of bytes
+      def pool_quota(name)
+        pool = find_pool_by_name(name)
+        # get license quota if it is not limited
+        pool["quota"] == "MAX" ? license_quota(pool["stack_id"]) : pool["quota"].to_i
+      end
+
+      # Find out license quota
+      # @param [String] stack_id licesne stack
+      # @return [Integer] license stack quota in bytes
+      def license_quota(stack_id)
+        find_licenses(stack_id).reduce(0) { |r,l| r+=l[1]["quota"].to_i }
+      end
+
+      # Find out pool(s) license usage
+      # @param [String] stack_id which pools are assigned to
+      # @retunr [Integer] number of bytes used by pool(s)
+      def pools_in_stack_usage(stack_id)
+        pools = find_pools(stack_id)
+        pools.reduce(0) {|r,p| r+=p[1]["used_bytes"].to_i }
+      end
+
       # list of avialable licenses
+      # @param [String] stack_id
       # @return [Array<Hash>]
-      def licenses
+      def find_licenses(stack_id)
         response = rest_client.get(LICENSE_LIST_URL)
-        response.code == "200" ? parse_data(response.body) : {}
+        result = response.code == "200" ? parse_data(response.body) : {}
+        # find all VALID licenses by stack_id
+        result.reject! {|k,v| v["stack_id"] != stack_id || v["status"] != "VALID" }
+        raise NoLicensesFound.new("No licenses are found in stack '#{stack_id}'") if result.empty?
+        result
       end
 
       # list of available pools
+      # @param [String] stack_id
       # @return [Array<Hash>]
-      def pools
+      def find_pools(stack_id = nil)
         response = rest_client.get(POOL_LIST_URL)
-        response.code == "200" ? parse_data(response.body) : {}
+        result = response.code == "200" ? parse_data(response.body) : {}
+        # find all license pools by stack_id
+        result.reject! {|k,v| v["stack_id"] != stack_id } unless stack_id.nil?
+        raise NoPoolsFound.new("No pools are found in stack '#{stack_id}'") if result.empty?
+        result
       end
 
-      private
+      # Find pool by name
+      # param [String] name of the pool
+      # param [String] stack_id which pool is assigned to
+      # @return [Hash] pool
+      def find_pool_by_name(name)
+        pools = find_pools
+        result = pools.find { |k,v| k == name } # => ["pool-name", { "param1" => "value1" }]
+        raise NoPoolsFound.new("No pool named '#{name}' is found") if result.nil?
+        # return hash of parameters for the pool
+        result[1]
+      end
 
       def parse_data(data)
         doc = Nokogiri::Slop(data)

data/lib/nagios/splunk/cli.rb

@@ -7,8 +7,8 @@ module Nagios
       include Mixlib::CLI
 
       option(:server_url,
-             :short => "-s URL",
-             :long  => "--server URL",
+             :short => "-u URL",
+             :long  => "--server-url URL",
              :required => true,
              :description => "Splunk server url",
              :on => :head)
@@ -25,13 +25,33 @@ module Nagios
              :default => '90',
              :description => "Critical % of license capacity usage (default: 90)")
 
+      option(:pool,
+             :short => "-p POOL",
+             :long  => "--pool-name POOL",
+             :description => "Pool name wich usage is being checked.")
+
+      option(:stack_id,
+             :short => "-s STACK_ID",
+             :long  => "--stack-id STACK_ID",
+             :default => 'enterprise',
+             :description => "License Stack ID.")
+
       def run(argv = ARGV)
         parse_options(argv)
 
         client = RestClient.new(config[:server_url])
         splunk = Check.new(client)
 
-        status, message = splunk.license_usage(config[:warn], config[:crit])
+        begin
+          if config[:pool]
+            status, message = splunk.pool_usage(config[:pool], config[:warn], config[:crit])
+          else
+            status, message = splunk.license_usage(config[:warn], config[:crit], config[:stack_id])
+          end
+        rescue Nagios::Splunk::Exception => e
+          message = e.message
+          status = 3
+        end
 
         puts message
         status

data/lib/nagios/splunk/exception.rb

@@ -0,0 +1,12 @@
+module Nagios
+  module Splunk
+    class Exception < ::Exception
+    end
+
+    class NoLicensesFound < Exception
+    end
+
+    class NoPoolsFound < Exception
+    end
+  end
+end

data/nagios-splunk.gemspec

@@ -2,7 +2,7 @@ $:.push File.expand_path("../lib", __FILE__)
 
 Gem::Specification.new do |s|
   s.name        = "nagios-splunk"
-  s.version     = "0.0.1"
+  s.version     = "0.0.2"
   s.authors     = 'Max Horbul'
   s.email       = ["max@gorbul.net"]
   s.homepage    = "https://github.com/mhorbul/nagios-splunk"

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: nagios-splunk
 version: !ruby/object:Gem::Version 
-  hash: 29
+  hash: 27
   prerelease: 
   segments: 
   - 0
   - 0
-  - 1
-  version: 0.0.1
+  - 2
+  version: 0.0.2
 platform: ruby
 authors: 
 - Max Horbul
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2012-04-12 00:00:00 Z
+date: 2012-07-19 00:00:00 Z
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: nokogiri
@@ -86,6 +86,7 @@ files:
 - bin/check_splunk
 - lib/nagios/splunk/check.rb
 - lib/nagios/splunk/cli.rb
+- lib/nagios/splunk/exception.rb
 - lib/nagios/splunk/rest_client.rb
 - lib/nagios/splunk.rb
 - test/test_helper.rb