sensu-plugins-edgelab 1.16.0 → 1.17.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a5ae185b0178726421cb6424a2520cbda7097bb27b95032cbf6ab936effd88bf
-  data.tar.gz: e7c6d24454546237899cd3fa42dc02364132a78c82bf1f01cfbc312478523501
+  metadata.gz: 02abb4c1626d01a1c9434ab4ad55dba2b1be0c225f7f7a63b8e06d549bbeb17c
+  data.tar.gz: 6c2593babae5d55aebe767a562fc187e618d72b49b57860f09f89bb84a3db8de
 SHA512:
-  metadata.gz: 584fde208f1a8d1c5da5838098970d3a7b035469518245b067d904a8dcff10fec6847fca1f40e08a94891b278ad7d34437babd8fa51c8b3c52a62877fc134f58
-  data.tar.gz: ea4dd7edef755088d26b4d1db51891ae3c3d647c68acebf29812f72a294b0a360efa213aa4fc73fd11bf2540f324dc431ea9396fd79eea4604c4b36e8ace4c11
+  metadata.gz: 45f18be0e7706239fc5444e6c8d55c6fc91252e2eadf080f3e61668a8a13b07a53820025341a39a7cdac528055253b9846b4e4624892c5c6633d6bc38b0bbeae
+  data.tar.gz: b8cf6b8a55bad75d16fd975926ee3daec82611111ff07b5be9e87b6d44f0b186e35618b6583568817f004ad60165acd54b07c24e80f770b612f852c5b02b6c95

data/bin/check-elasticsearch-cluster-health.rb

@@ -0,0 +1,142 @@
+#! /usr/bin/env ruby
+#
+#   check-elasticsearch-cluster-health
+#
+# DESCRIPTION:
+#   This plugin checks the Elasticsearch cluster health from its status and number of nodes present.
+#
+# OUTPUT:
+#   plain text
+#
+# PLATFORMS:
+#   Linux
+#
+# DEPENDENCIES:
+#   gem: sensu-plugin
+#   gem: elasticsearch
+#
+# USAGE:
+#   Checks against the Elasticsearch api for cluster health using the
+#   Elasticsearch gem
+#
+# NOTES:
+#   Adapted from sensu-plugins-elasticsearch `check-es-cluster-health.rb`
+#
+# LICENSE:
+#   Jonathan Ballet <jballet@edgelab.ch>
+#   Brendan Gibat <brendan.gibat@gmail.com>
+#   Released under the same terms as Sensu (the MIT license); see LICENSE
+#   for details.
+
+require 'sensu-plugin/check/cli'
+require 'elasticsearch'
+
+class ESClusterHealth < Sensu::Plugin::Check::CLI
+  option :host,
+         description: 'Elasticsearch host',
+         short: '-h HOST',
+         long: '--host HOST',
+         default: 'localhost'
+
+  option :port,
+         description: 'Elasticsearch port',
+         short: '-p PORT',
+         long: '--port PORT',
+         proc: proc(&:to_i),
+         default: 9200
+
+  option :scheme,
+         description: 'Elasticsearch connection scheme, defaults to https for authenticated connections',
+         short: '-s SCHEME',
+         long: '--scheme SCHEME'
+
+  option :password,
+         description: 'Elasticsearch connection password',
+         short: '-P PASSWORD',
+         long: '--password PASSWORD'
+
+  option :user,
+         description: 'Elasticsearch connection user',
+         short: '-u USER',
+         long: '--user USER'
+
+  option :timeout,
+         description: 'Elasticsearch query timeout in seconds',
+         short: '-t TIMEOUT',
+         long: '--timeout TIMEOUT',
+         proc: proc(&:to_i),
+         default: 30
+
+  option :min_nodes,
+         description: 'Minimum of nodes that should be present in the cluster',
+         short: '-n NB_NODES',
+         long: '--minimum-nodes NB_NODES',
+         proc: proc(&:to_i),
+         default: 0
+
+  # From sensu-plugins-elasticsearch's ElasticsearchCommon
+  def client
+    transport_class = nil
+
+    host = {
+      host:               config[:host],
+      port:               config[:port],
+      request_timeout:    config[:timeout],
+      scheme:             config[:scheme]
+    }
+
+    if !config[:user].nil? && !config[:password].nil?
+      host[:user] = config[:user]
+      host[:password] = config[:password]
+      host[:scheme] = 'https' unless config[:scheme]
+    end
+
+    transport_options = {}
+
+    if config[:headers]
+
+      headers = {}
+
+      config[:headers].split(',').each do |header|
+        h, v = header.split(':', 2)
+        headers[h.strip] = v.strip
+      end
+
+      transport_options[:headers] = headers
+
+    end
+
+    @client ||= Elasticsearch::Client.new(transport_class: transport_class, hosts: [host], transport_options: transport_options)
+  end
+
+  def run
+    options = {}
+    health = client.cluster.health options
+
+    message = ''
+
+    case health['status']
+    when 'yellow'
+      cb = method(:warning)
+      message += 'Cluster state is Yellow'
+    when 'red'
+      cb = method(:critical)
+      critical 'Cluster state is Red'
+    when 'green'
+      cb = method(:ok)
+      message = 'Cluster state is Green'
+    else
+      cb = method(:unknown)
+      message = "Cluster state is in an unknown health: #{health['status']}"
+    end
+
+    message += " (#{health['active_shards_percent_as_number'].round(2)} % of active shards)"
+
+    if health['number_of_nodes'] < config[:min_nodes]
+      cb = method(:critical)
+      message = "Not enough nodes: #{health['number_of_nodes']} < #{config[:min_nodes]} - " + message
+    end
+
+    cb.call(message)
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sensu-plugins-edgelab
 version: !ruby/object:Gem::Version
-  version: 1.16.0
+  version: 1.17.0
 platform: ruby
 authors:
 - Edgelab
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-06-28 00:00:00.000000000 Z
+date: 2018-07-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: cassandra-driver
@@ -94,6 +94,20 @@ dependencies:
     - - '='
       - !ruby/object:Gem::Version
         version: 1.8.0
+- !ruby/object:Gem::Dependency
+  name: elasticsearch
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.14
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.0.14
 - !ruby/object:Gem::Dependency
   name: sensu-plugin
   requirement: !ruby/object:Gem::Requirement
@@ -184,6 +198,7 @@ executables:
 - check-apt-expired-keys.rb
 - check-cassandra-nodes.rb
 - check-consul-services.rb
+- check-elasticsearch-cluster-health.rb
 - check-logstash-pipeline.rb
 - check-nomad-jobs.rb
 - check-nomad-leader.rb
@@ -200,6 +215,7 @@ files:
 - bin/check-apt-expired-keys.rb
 - bin/check-cassandra-nodes.rb
 - bin/check-consul-services.rb
+- bin/check-elasticsearch-cluster-health.rb
 - bin/check-logstash-pipeline.rb
 - bin/check-nomad-jobs.rb
 - bin/check-nomad-leader.rb