sensu-plugins-kafka 0.7.0 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 796bfc7ffe11da8fce4d84d3819ef91b228dada6
-  data.tar.gz: 80257076434b73e0089246be7a4ab2b5eb349db7
+  metadata.gz: 436d7165e1ea6b4ac7efe7ee7c4456619cb0d7bd
+  data.tar.gz: 1b33367a5fb5e418031ab76b3f1f571c660d4e69
 SHA512:
-  metadata.gz: 624ce35280ff74093e0d0c917eb2db44c12d0f5e1b59011b1ec9fd79abe76938e2cb5bbcdff41c896afe3ab0eb30a3d5335087305d46fef4e854b9a2ae8c74d2
-  data.tar.gz: 4ef0d7001c151f80a77fe68296e1419181ba7739366c49ad2f3c63f3f7b93183840ac406bf295ab8789a28ae84a2884ce526b784e5d8c1991fae665104a9f36b
+  metadata.gz: e0386a39236c73b7b192bd287e1cd00c7a02da9ef8af2e79f26315d49a3dc005e87ddc77f66e19bbc47fb0f96b519963001fa574e496758ef3df92a35ab8825c
+  data.tar.gz: b500208baf3951334ef8a055feb277cb27aa261f7f1ea0b21f4618ba86ba5ad0287abad5b462f00ad61c963c22defb08d7278ab15e2d28c78b2519b21c6e5ea6

data/CHANGELOG.md

@@ -10,6 +10,11 @@ This CHANGELOG follows the format listed at [Keep A Changelog](http://keepachang
 ### Fixed
 ### Changed
 
+## [Unreleased]
+
+### Breaking Changes
+- Refactoring check-consumer-lag to remove shell/jvm execution
+
 ## [0.7.0]
 
 ### Added

data/bin/check-consumer-lag.rb

@@ -27,6 +27,17 @@
 
 require 'sensu-plugin/check/cli'
 
+require 'celluloid-io'
+require 'json'
+require 'poseidon'
+require 'zookeeper'
+
+module Poseidon
+  class Connection
+    TCPSocket = Celluloid::IO::TCPSocket
+  end
+end
+
 class ConsumerLagCheck < Sensu::Plugin::Check::CLI
   option :group,
          description: 'Consumer group',
@@ -34,25 +45,6 @@ class ConsumerLagCheck < Sensu::Plugin::Check::CLI
          long:        '--group NAME',
          required:    true
 
-  option :kafka_home,
-         description: 'Kafka home',
-         short:       '-k NAME',
-         long:        '--kafka-home NAME',
-         default:     '/opt/kafka'
-
-  option :topic_excludes,
-         description: 'Excludes consumer topics',
-         short:       '-e NAME',
-         long:        '--topic-excludes NAME',
-         proc:        proc { |a| a.split(',') }
-
-  option :autolist,
-         description: 'Auto list topics',
-         short:       '-a VALUE',
-         long:        '--auto-list VALUE',
-         boolean: true,
-         default: true
-
   option :zookeeper,
          description: 'ZooKeeper connect string',
          short:       '-z NAME',
@@ -60,85 +52,85 @@ class ConsumerLagCheck < Sensu::Plugin::Check::CLI
          default:     'localhost:2181'
 
   option :warning_over,
-         description: 'Warning if metric statistics is over specified value.',
+         description: 'Warning if lag is over specified value.',
          short:       '-W N',
          long:        '--warning-over N'
 
   option :critical_over,
-         description: 'Critical if metric statistics is over specified value.',
+         description: 'Critical if lag is over specified value.',
          short:       '-C N',
          long:        '--critical-over N'
 
   option :warning_under,
-         description: 'Warning if metric statistics is under specified value.',
+         description: 'Warning if lag is under specified value.',
          short:       '-w N',
          long:        '--warning-under N'
 
   option :critical_under,
-         description: 'Critical if metric statistics is under specified value.',
+         description: 'Critical if lag is under specified value.',
          short:       '-c N',
          long:        '--critical-under N'
 
-  # read the output of a command
-  # @param cmd [String] the command to read the output from
-  def read_lines(cmd)
-    IO.popen(cmd + ' 2>&1') do |child|
-      child.read.split("\n")
-    end
+  def kafka_topics(zk, group)
+    zk.get_children(path: "/consumers/#{group}/owners")[:children].sort
   end
 
-  # create a hash from the output of each line of a command
-  # @param line [String]
-  # @param cols
-  def line_to_hash(line, *cols)
-    Hash[cols.zip(line.strip.split(/\s+/, cols.size))]
+  def topics_partitions(zk, topic)
+    JSON.parse(zk.get(path: "/brokers/topics/#{topic}")[:data])['partitions'].keys.map(&:to_i).sort
   end
 
-  # run command and return a hash from the output
-  # @param cmd [String]
-  def run_offset(cmd)
-    read_lines(cmd).drop(1).map do |line|
-      line_to_hash(line, :group, :topic, :pid, :offset, :logsize, :lag, :owner)
-    end
+  def leader_broker(zk, topic, partition)
+    state = zk.get(path: "/brokers/topics/#{topic}/partitions/#{partition}/state")
+    leader = JSON.parse(state[:data])['leader']
+    JSON.parse(zk.get(path: "/brokers/ids/#{leader}")[:data])
   end
 
-  # run command and return a hash from the output
-  # @param cmd [String]
-  def run_topics(cmd)
-    topics = []
-    read_lines(cmd).map do |line|
-      if !line.include?('__consumer_offsets') && !line.include?('marked for deletion')
-        topics.push(line)
-      end
-    end
-    topics
+  def consumer_offset(zk, group, topic, partition)
+    zk.get(path: "/consumers/#{group}/offsets/#{topic}/#{partition}")[:data].to_i
+  end
+
+  def partition_owner(zk, group, partition)
+    zk.get(path: "/consumers/#{group}/offsets/#{owners}/#{partition}")[:data]
   end
 
   def run
-    kafka_run_class = "#{config[:kafka_home]}/bin/kafka-run-class.sh"
-    unknown "Can not find #{kafka_run_class}" unless File.exist?(kafka_run_class)
-
-    topics_to_read = []
-    if config[:autolist].to_s == 'true'
-      cmd_topics = "#{kafka_run_class} kafka.admin.TopicCommand --zookeeper #{config[:zookeeper]} --list"
-      topics_to_read = run_topics(cmd_topics)
-      topics_to_read.delete_if { |x| config[:topic_excludes].include?(x) } if config[:topic_excludes]
-    end
+    z = Zookeeper.new(config[:zookeeper])
+
+    group = config[:group]
+    topics = kafka_topics(z, group)
+
+    critical 'Could not found topics' if topics.empty?
+
+    consumers = {}
+    topics.each do |topic|
+      consumers[topic] = {}
 
-    cmd_offset = "#{kafka_run_class} kafka.tools.ConsumerOffsetChecker --group #{config[:group]} --zookeeper #{config[:zookeeper]}"
-    cmd_offset += " --topic #{topics_to_read.join(',')}" unless topics_to_read.empty?
+      topics_partitions(z, topic).each do |partition|
+        owner = partition_owner(z, group, partition)
 
-    topics = run_offset(cmd_offset).group_by { |h| h[:topic] }
+        leader = leader_broker(z, topic, partition)
+        consumer = Poseidon::PartitionConsumer.new('CheckConsumerLag', leader['host'], leader['port'], topic, partition, :latest_offset)
+        logsize = consumer.next_offset
 
-    critical 'Could not found topics/partitions' if topics.empty?
+        offset = consumer_offset(z, group, topic, partition)
+
+        lag = logsize - offset
+
+        consumers[topic][:partition] = partition
+        consumers[topic][:logsize] = logsize
+        consumers[topic][:offset] = offset
+        consumers[topic][:lag] = lag
+        consumers[topic][:owner] = owner
+      end
+    end
 
     [:offset, :logsize, :lag].each do |field|
-      topics.map do |k, v|
+      consumers.map do |k, v|
         critical "Topic #{k} has partitions with #{field} < 0" unless v.select { |w| w[field].to_i < 0 }.empty?
       end
     end
 
-    topics.map do |k, v|
+    consumers.map do |k, v|
       critical "Topic #{k} has partitions with no owner" unless v.select { |w| w[:owner] == 'none' }.empty?
     end
 
@@ -165,7 +157,7 @@ class ConsumerLagCheck < Sensu::Plugin::Check::CLI
           end
         when :under
           if min_lag < threshold.to_i
-            msg =  "Topics `#{min_topics}` for the group `#{config[:group]}` lag: #{min_lag} (<= #{threshold})"
+            msg = "Topics `#{min_topics}` for the group `#{config[:group]}` lag: #{min_lag} (<= #{threshold})"
             send severity, msg
           end
         end

data/lib/sensu-plugins-kafka/version.rb

@@ -5,7 +5,7 @@ module SensuPluginsKafka
   # This defines the version of the gem
   module Version
     MAJOR = 0
-    MINOR = 7
+    MINOR = 8
     PATCH = 0
 
     VER_STRING = [MAJOR, MINOR, PATCH].compact.join('.')

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sensu-plugins-kafka
 version: !ruby/object:Gem::Version
-  version: 0.7.0
+  version: 0.8.0
 platform: ruby
 authors:
 - Sensu-Plugins and contributors
@@ -9,8 +9,36 @@ autorequire:
 bindir: bin
 cert_chain:
 - certs/sensu-plugins.pem
-date: 2017-02-10 00:00:00.000000000 Z
+date: 2017-02-15 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: celluloid-io
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.17.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.17.3
+- !ruby/object:Gem::Dependency
+  name: poseidon
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.0.5
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - '='
+      - !ruby/object:Gem::Version
+        version: 0.0.5
 - !ruby/object:Gem::Dependency
   name: sensu-plugin
   requirement: !ruby/object:Gem::Requirement