sensu-plugins-json-mutator 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 84175624864453c65ec0f511e94cdcfc19c6c360
-  data.tar.gz: 235b149a1a3dbbf7b682cebf366f973ec41556fa
+  metadata.gz: 4bf3678e20709f8edea8527f78613936c67be799
+  data.tar.gz: 854350ddc124f082ca742e5b08979275f960c4db
 SHA512:
-  metadata.gz: a3eb9db8f3889ef8e7e7cd79566a49e9f9550a44617bd242f712c6c8885ca207c3fd5874b027cb703b4acde990a5acb9feeea05d7643d4232615195a84531b52
-  data.tar.gz: 6533c92d23fde011ecf8518d98d780ba801217407d1ee39a90cf6d060ea02778c3649b8cb140496a237ab2e611af08b59479ac63c75d59d95f6aff3faf27de75
+  metadata.gz: 26ca01d37cbd76995846a265508dd841f6a122ff493537b9d2094a4815abf93c8b0e840dd88c0c5eec5f820c7ccba8cd1023ae4bba671743177ebc03b5fc6b0e
+  data.tar.gz: 3498508e951c3655452cdaa9629bb80af2c737998076f4489083d2015ee7c324340a37a433d1a834e973583b46f29ea314ce772c94d2f21599c22741da274cee

data/bin/mutator-json.rb

@@ -16,25 +16,30 @@
 #   gem: sensu-plugin
 
 require 'json'
+require 'sensu-plugin'
+require 'sensu-plugins-json-mutator/mutator'
 
-GRAPHITE = /([\w\.]+)\s(.+).*/  # sample.key 42.13 1462521982
-STATSD   = /([\w\.]+):(.*)\|.*/ # sample.key:42.13|kv
+class JSONMutator < Sensu::Mutator
+  GRAPHITE = /([\w\.]+)\s(.+).*/  # sample.key 42.13 1462521982
+  STATSD   = /([\w\.]+):(.*)\|.*/ # sample.key:42.13|kv
 
-def parse_lines(lines, regexp)
-  kv = {}
+  def mutate
+    lines = @event['check']['output'].lines
 
-  lines.each do |line|
-    result = line.match(regexp)
-    kv[result[1]] = result[2].to_f if result
+    @event['check']['output'] = parse_lines(lines, GRAPHITE) if lines.first =~ GRAPHITE
+    @event['check']['output'] = parse_lines(lines, STATSD) if lines.first =~ STATSD
   end
 
-  kv
-end
+private
 
-event = JSON.parse(STDIN.read)
-lines = event['check']['output'].lines
+  def parse_lines(lines, regexp)
+    kv = {}
 
-event['check']['output'] = parse_lines(lines, GRAPHITE) if lines.first =~ GRAPHITE
-event['check']['output'] = parse_lines(lines, STATSD) if lines.first =~ STATSD
+    lines.each do |line|
+      result = line.match(regexp)
+      kv[result[1]] = result[2].to_f if result
+    end
 
-puts JSON.dump(event)
+    kv
+  end
+end

data/lib/sensu-plugins-json-mutator/mutator.rb

@@ -0,0 +1,49 @@
+require 'json'
+require 'sensu-plugin/utils'
+require 'mixlib/cli'
+
+module Sensu
+  class Mutator
+    include Sensu::Plugin::Utils
+    include Mixlib::CLI
+
+    attr_accessor :argv
+
+    def initialize(argv = ARGV)
+      super()
+      self.argv = parse_options(argv)
+    end
+
+    def mutate
+      ## Override this, be sure any changes are made to @event
+      nil
+    end
+
+    def dump
+      puts JSON.dump(@event)
+    end
+
+    # This works just like Plugin::CLI's autorun.
+    @@autorun = self
+    class << self
+      def method_added(name)
+        if name == :mutate
+          @@autorun = self
+        end
+      end
+    end
+
+    def self.disable_autorun
+      @@autorun = false
+    end
+
+    at_exit do
+      if @@autorun
+        mutator = @@autorun.new
+        mutator.read_event(STDIN)
+        mutator.mutate
+        mutator.dump
+      end
+    end
+  end
+end

data/lib/sensu-plugins-json-mutator/version.rb

@@ -1,3 +1,3 @@
 module SensuPluginsJSONMutator
-  VERSION = '0.1.0'.freeze
+  VERSION = '0.2.0'.freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: sensu-plugins-json-mutator
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Sensu-Plugins and contributors
@@ -92,6 +92,7 @@ files:
 - README.md
 - bin/mutator-json.rb
 - lib/sensu-plugins-json-mutator.rb
+- lib/sensu-plugins-json-mutator/mutator.rb
 - lib/sensu-plugins-json-mutator/version.rb
 homepage: https://github.com/artemeff/sensu-plugins-json-mutator
 licenses: