sensu-extensions-check-deps 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5809597894d95fd153b57025b12ef9a0d9a02e80
-  data.tar.gz: cb13407e0ae4c6553892a43a0201545df833c398
+  metadata.gz: 6e2ba3613eeaea04f758239bfe48ee60bdedd0c2
+  data.tar.gz: fe25b9a5d0333443de861370f662e9929cc98402
 SHA512:
-  metadata.gz: 679709902130df38fa85c9c7a6f61ace6c5b208b278ce306c0b89f5a16dd03ea0f30dcfd9f1a302e82c1b509e263df2f2e846c4c24f1e41ac9cf179ead0bc4db
-  data.tar.gz: ad1e4593451fb35d396c9bdbc9175b091c6f00de8eeeb09446f1b2f1b30f930f0e31f83a2b355978378c656aa14712cbccce3872194a6403678bcddf129ca298
+  metadata.gz: 948ae5fe2c3d3612c1a11af48b8a38d3f7d8e8b3d8ed713594153ccdc53792d4246ca53abedd730381feb7c8994d4a2d4ad8ace8d1325eb386fcf050ec876286
+  data.tar.gz: c66748f87fa1fc9b3f179d50d5b53503496cb68e4b1ade77f5eeb9769c5e55bb8654b187ee8478e25a5296742f2f3902fb73ba5f449f24dc41c8b8e59597e97e

data/lib/sensu/extensions/check-deps-any.rb

@@ -0,0 +1,132 @@
+require "sensu/extension"
+require "timeout"
+require "net/http"
+require 'json'
+
+module Sensu
+  module Extension
+    class CheckDeps < Filter
+      def name
+        "check_deps"
+      end
+
+      def description
+        "filter events when an event exists and send to RabbitMQ event json"
+      end
+
+      # Will post Slack message to channel sensu
+      def writeToSlack(message, err=false)
+            webhookUrl = "https://hooks.slack.com/services/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"
+            channel = "#sensu"
+            emoji = ":no_entry:"
+            begin
+            payload = {
+                :channel  => channel,
+                :text     => message,
+                :icon_url => emoji
+            }.to_json
+              cmd = "curl -X POST --data-urlencode 'payload=#{payload}' #{webhookUrl}"
+              system(cmd)
+            rescue
+                    puts "failed to notify slack, proceeding..."
+            end
+      end
+      #writeToSlack("Event was blocked: " + @event['check']['name'] + "becasuse some of dependencies was already triggered. Deps list: "+ @event['check']['dependencies'])
+
+      # Make an HTTP GET request to the Sensu API, using the URI
+      # path provided. Uses Sensu settings to determine how to
+      # connect to the API.
+      #
+      # @param path [String]
+      # @return [Object] http response object.
+
+      def sensu_api_get_request(path)
+        api = @settings[:api] || {}
+        request = Net::HTTP::Get.new(path)
+        if api[:user]
+          request.basic_auth(api[:user], api[:password])
+        end
+        Net::HTTP.new(api[:host] || '127.0.0.1', api[:port] || 4567).start do |http|
+          http.request(request)
+        end
+      end
+
+      # Check to see if an event exists for a client/check pair. This
+      # method is looking for a HTTP response code of `200`.
+      #
+      # @param client_name [String]
+      # @param check_name [String]
+      # @return [Boolean]
+      def client_event_exists?(client_name, check_name)
+        path = "/events/#{client_name}/#{check_name}"
+        response = sensu_api_get_request(path)
+        response.code.to_i == 200
+      end
+
+      # Check to see if an event exists for a subscription/check pair. This
+      # method is looking for a HTTP response code of `200`.
+      #
+      # @param subscription_name [String]
+      # @param check_name [String]
+      # @return [Boolean]
+      def subscription_event_exists?(subscription_name, check_name)
+        path = "/events"
+        response = sensu_api_get_request(path)
+        events = JSON.load(response.body)
+        !events.select { |event| event[:client][:subscriptions].include?(subscription_name) && event[:check][:name] == check_name}.empty?
+      end
+
+      # Determine if an event exists for any of the check
+      # dependencies declared in the event data, specified in array,
+      # check `dependencies`. A check dependency can be a check
+      # executed by the same Sensu client (eg. `check_app`), a
+      # client/check pair (eg.`i-424242/check_mysql`), or a
+      # subscription/check pair (eg. `subscription:mysql/check_mysql`).
+      #
+      # @param event [Hash]
+      # @return [Boolean]
+
+      def dependency_events_exist?(event)
+        if event[:check][:dependencies].is_a?(Array)
+          event[:check][:dependencies].any? do |dependency|
+            begin
+              check_name, entity = dependency.split("/").reverse
+              if entity =~ /^subscription:.*$/
+                subscription_name = entity.split(":")[1]
+                subscription_event_exists?(subscription_name, check_name)
+              else
+                client_name = entity
+                client_name ||= event[:client][:name]
+                client_event_exists?(client_name, check_name)
+              end
+            rescue => error
+              @logger.error("failed to query api for a check dependency event", :error => error)
+              false
+            end
+          end
+        else
+          false
+        end
+      end
+
+      def run(event, &callback)
+        filter = Proc.new do
+          begin
+            Timeout::timeout(10) do
+              if dependency_events_exist?(event)
+                ["event exists for check dependency", 1]
+                writeToSlack("Event: " + event[check][name] + " will be blocked."  "Some of dependencies were already triggered. Deps list: "+ event[check][dependencies])
+              else
+                ["no current events for check dependencies", 1]
+              end
+            end
+          rescue => error
+            @logger.error("check dependencies filter error", :error => error.to_s)
+            ["check dependencies filter error: #{error}", 1]
+          end
+        end
+        EM.defer(filter, callback)
+      end
+    end
+  end
+end

data/lib/sensu/extensions/check-deps.rb

@@ -1,6 +1,7 @@
 require "sensu/extension"
 require "timeout"
 require "net/http"
+require 'json'
 
 module Sensu
   module Extension
@@ -13,12 +14,32 @@ module Sensu
         "filter events when an event exists and send to RabbitMQ event json"
       end
 
+      # Will post Slack message to channel sensu
+      def writeToSlack(message, err=false)
+            webhookUrl = "https://hooks.slack.com/services/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"
+            channel = "#sensu"
+            emoji = ":no_entry:"
+            begin
+            payload = {
+                :channel  => channel,
+                :text     => message,
+                :icon_url => emoji
+            }.to_json
+              cmd = "curl -X POST --data-urlencode 'payload=#{payload}' #{webhookUrl}"
+              system(cmd)
+            rescue
+                    puts "failed to notify slack, proceeding..."
+            end
+      end
+      #writeToSlack("Event was blocked: " + @event['check']['name'] + "becasuse some of dependencies was already triggered. Deps list: "+ @event['check']['dependencies'])
+
       # Make an HTTP GET request to the Sensu API, using the URI
       # path provided. Uses Sensu settings to determine how to
       # connect to the API.
       #
       # @param path [String]
       # @return [Object] http response object.
+
       def sensu_api_get_request(path)
         api = @settings[:api] || {}
         request = Net::HTTP::Get.new(path)
@@ -64,6 +85,7 @@ module Sensu
       #
       # @param event [Hash]
       # @return [Boolean]
+
       def dependency_events_exist?(event)
         if event[:check][:dependencies].is_a?(Array)
           event[:check][:dependencies].any? do |dependency|
@@ -92,7 +114,8 @@ module Sensu
           begin
             Timeout::timeout(10) do
               if dependency_events_exist?(event)
-                ["event exists for check dependency", 0]
+                ["event exists for check dependency", 1]
+                writeToSlack("Event: " + event[check][name] + " will be blocked."  "Some of dependencies were already triggered. Deps list: "+ event[check][dependencies])
               else
                 ["no current events for check dependencies", 1]
               end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: sensu-extensions-check-deps
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
 platform: ruby
 authors:
 - Vladimir Mevzos
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-05-29 00:00:00.000000000 Z
+date: 2018-05-31 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: sensu-extension
@@ -94,8 +94,7 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '0'
-description: Filter events when event exists and push message to Rabbitmq with event
-  json
+description: Filter events when event exists and post message to Slack
 email:
 - "<vladimir@appsflyer.com>"
 executables: []
@@ -105,6 +104,7 @@ files:
 - CHANGELOG.md
 - LICENSE
 - README.md
+- lib/sensu/extensions/check-deps-any.rb
 - lib/sensu/extensions/check-deps.rb
 homepage: https://github.com/mau1024/sensu-extensions-check-deps
 licenses: []
@@ -128,5 +128,5 @@ rubyforge_project:
 rubygems_version: 2.6.11
 signing_key: 
 specification_version: 4
-summary: Filter events when event exists and push message to Rabbitmq with event json
+summary: Filter events when event exists and post message to Slack
 test_files: []