sensu-transport-snssqs-ng 2.0.4

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: c92289e5673d6c1523eaff67ba2c13811d224b0b
+  data.tar.gz: 53a04fd08f78ec57eea0e4c07c447809bedc1641
+SHA512:
+  metadata.gz: dcf918e8f291f21ec353df00d6dc24c76151e7a21fb9e2c12bcf4eca8d53054446439219d81fe7c3cb532b7d70d91772be59c41bd75fb70409f2e2770a8a7693
+  data.tar.gz: 1879be3c160435f6724390b10fc1f29cd4c38b06d0ff0c4965f176040f148d9a05f9bf6c5ce9d366290fea74284f10741c99583e449dbff3f482737c17681237

data/lib/sensu/transport/snssqs.rb

@@ -0,0 +1,185 @@
+require 'sensu/transport/base'
+require 'aws-sdk'
+require 'statsd-ruby'
+
+module Sensu
+  module Transport
+    class SNSSQS < Sensu::Transport::Base
+      attr_accessor :logger
+
+      STRING_STR     = "String".freeze
+      KEEPALIVES_STR = "keepalives".freeze
+      PIPE_STR       = "pipe".freeze
+      TYPE_STR       = "type".freeze
+
+      def initialize
+        @connected = false
+        @subscribing = false
+
+        # as of sensu 0.23.0 we need to call succeed when we have
+        # successfully connected to SQS.
+        #
+        # we already have our own logic to maintain the connection to
+        # SQS, so we can always say we're connected.
+        #
+        # See:
+        # https://github.com/sensu/sensu/blob/cdc25b29169ef2dcd2e056416eab0e83dbe000bb/CHANGELOG.md#0230---2016-04-04
+        succeed()
+      end
+
+      def connected?; @connected; end
+
+      def connect(settings)
+        @settings = settings
+        @connected = true
+        @results_callback = proc {}
+        @keepalives_callback = proc {}
+        # Sensu Windows install does not include a valid cert bundle for AWS
+        Aws.use_bundled_cert! if Gem.win_platform?
+        @sqs = Aws::SQS::Client.new(region: @settings[:region])
+        @sns = Aws::SNS::Client.new(region: @settings[:region])
+
+        # connect to statsd, if necessary
+        @statsd = nil
+        if !@settings[:statsd_addr].nil? and @settings[:statsd_addr] != ""
+          pieces = @settings[:statsd_addr].split(':')
+          @statsd = Statsd.new(pieces[0], pieces[1].to_i).tap { |sd|
+            sd.namespace = @settings[:statsd_namespace]
+          }
+          @statsd_sample_rate = @settings[:statsd_sample_rate].to_f
+        end
+      end
+
+      def statsd_incr(stat)
+        @statsd.increment(stat, @statsd_sample_rate) unless @statsd.nil?
+      end
+
+      def statsd_time(stat)
+        # always measure + run the block, but only if @statsd is set
+        # do we actually report it.
+        start = Time.now
+        result = yield
+        if !@statsd.nil?
+          @statsd.timing(stat, ((Time.now - start) * 1000).round(5), @statsd_sample_rate)
+        end
+        result
+      end
+
+      # subscribe will begin "subscribing" to the consuming sqs queue.
+      #
+      # This method is intended for use by the Sensu server; fanout
+      # subscriptions initiated by the Sensu client process are
+      # treated as a no-op.
+      #
+      # What this really means is that we will start polling for
+      # messages from the SQS queue, and, depending on the message
+      # type, it will call the appropriate callback.
+      #
+      # This assumes that the SQS Queue is consuming "Raw" messages
+      # from SNS.
+      #
+      # "subscribing" means that the "callback" parameter will be
+      # called when there is a message for you to consume.
+      #
+      # "funnel" and "type" parameters are completely ignored.
+      def subscribe(type, pipe, funnel = nil, options = {}, &callback)
+        if type == :fanout
+          self.logger.debug("skipping unsupported fanout subscription type=#{type}, pipe=#{pipe}, funnel=#{funnel}")
+          return
+        end
+
+        self.logger.info("subscribing to type=#{type}, pipe=#{pipe}, funnel=#{funnel}")
+
+        if pipe == KEEPALIVES_STR
+          @keepalives_callback = callback
+        else
+          @results_callback = callback
+        end
+
+        unless @subscribing
+          do_all_the_time {
+            EM::Iterator.new(receive_messages, 10).each do |msg, iter|
+              statsd_time("sqs.#{@settings[:consuming_sqs_queue_url]}.process_timing") {
+                if msg.message_attributes[PIPE_STR].string_value == KEEPALIVES_STR
+                  @keepalives_callback.call(msg, msg.body)
+                else
+                  @results_callback.call(msg, msg.body)
+                end
+              }
+              iter.next
+            end
+          }
+          @subscribing = true
+        end
+      end
+
+      # acknowledge will delete the given message from the SQS queue.
+      def acknowledge(info, &callback)
+        EM.defer {
+          @sqs.delete_message(
+            queue_url: @settings[:consuming_sqs_queue_url],
+            receipt_handle: info.receipt_handle,
+          )
+          statsd_incr("sqs.#{@settings[:consuming_sqs_queue_url]}.message.deleted")
+          callback.call(info) if callback
+        }
+      end
+
+      # publish publishes a message to the SNS topic.
+      #
+      # The type, pipe, and options are transformed into SNS message
+      # attributes and included with the message.
+      def publish(type, pipe, message, options = {}, &callback)
+        attributes = {
+          TYPE_STR => str_attr(type),
+          PIPE_STR => str_attr(pipe)
+        }
+        options.each do |k, v|
+          attributes[k.to_s] = str_attr(v.to_s)
+        end
+        EM.defer { send_message(message, attributes, &callback) }
+      end
+
+      private
+
+      def str_attr(str)
+        { :data_type => STRING_STR, :string_value => str }
+      end
+
+      def do_all_the_time(&blk)
+        callback = proc {
+          do_all_the_time(&blk)
+        }
+        EM.defer(blk, callback)
+      end
+
+      def send_message(msg, attributes, &callback)
+        resp = @sns.publish(
+          target_arn: @settings[:publishing_sns_topic_arn],
+          message: msg,
+          message_attributes: attributes
+        )
+        statsd_incr("sns.#{@settings[:publishing_sns_topic_arn]}.message.published")
+        callback.call({ :response => resp }) if callback
+      end
+
+      PIPE_ARR = [PIPE_STR]
+
+      # receive_messages returns an array of SQS messages
+      # for the consuming queue
+      def receive_messages
+        begin
+          resp = @sqs.receive_message(
+            message_attribute_names: PIPE_ARR,
+            queue_url: @settings[:consuming_sqs_queue_url],
+            wait_time_seconds: @settings[:wait_time_seconds],
+            max_number_of_messages: @settings[:max_number_of_messages],
+          )
+          resp.messages
+        rescue Aws::SQS::Errors::ServiceError => e
+          self.logger.info(e)
+        end
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,109 @@
+--- !ruby/object:Gem::Specification
+name: sensu-transport-snssqs-ng
+version: !ruby/object:Gem::Version
+  version: 2.0.4
+platform: ruby
+authors:
+- Troy Ready
+autorequire: 
+bindir: bin
+cert_chain:
+- |
+  -----BEGIN CERTIFICATE-----
+  MIIDjjCCAnagAwIBAgIBATANBgkqhkiG9w0BAQUFADBGMRgwFgYDVQQDDA90cm95
+  LnJlYWR5X2dlbXMxFTATBgoJkiaJk/IsZAEZFgVnbWFpbDETMBEGCgmSJomT8ixk
+  ARkWA2NvbTAeFw0xNjExMDgyMzMzMjlaFw0xNzExMDgyMzMzMjlaMEYxGDAWBgNV
+  BAMMD3Ryb3kucmVhZHlfZ2VtczEVMBMGCgmSJomT8ixkARkWBWdtYWlsMRMwEQYK
+  CZImiZPyLGQBGRYDY29tMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA
+  4HhKJwPlVHMlOaxmv2FBaS/M8u/9mN2v5WWsv0+6P934jzDIYBBKE6oW5eFUg6vM
+  wpWOaF4O4mwCRoOCvfGuhg2ufVnDYh3GPIWqE1ziIBx/Y3D5VIQSnAvP4/6yDMuS
+  kKTHi3/fJcs/vOs9JiuTb7zkb53TFLlR/g2lX6I8vy0HowJcMYZzKLhEUnnIPG7a
+  DPD5Qy67p8rIy/T6AsSCCPdn+mpg1jLHsTsj6DdQxPyM9rtNZRRGvBluZNa4QxYv
+  PMj3dSRlO9eEKHayrY24pcOZqus8eyPE64bbOeR6yj34vRMJ2eeu+Pq/YzRxlqmf
+  1dtR7Nm3Uxhmw1/PLWi8eQIDAQABo4GGMIGDMAkGA1UdEwQCMAAwCwYDVR0PBAQD
+  AgSwMB0GA1UdDgQWBBTX6fBjssJCph8JWQd+lF31ZWtkLTAkBgNVHREEHTAbgRl0
+  cm95LnJlYWR5K2dlbXNAZ21haWwuY29tMCQGA1UdEgQdMBuBGXRyb3kucmVhZHkr
+  Z2Vtc0BnbWFpbC5jb20wDQYJKoZIhvcNAQEFBQADggEBADMf8GNVKPmUTX8iirDp
+  p4FJtZ5pv614QBSoxutDMTVRPUGluN9dZ/oUNTBvByrm4CJI4vMbZhfIs07L8YdG
+  D/u5A5nXD9qlKIi9CtYHIekjsetool2s4tCMTyvNO4CfV/dALT0Vdc3gXIPeC8YZ
+  lIT+5GzwvlSX+TXYUq/Q4lDLQHU28j2nC4yFl5rej5akZCjiurJWE8A9P/AiHkgc
+  ZHJ4XyeURdy0oA6kjiWzYf4puRodQTDQ2W4uis7aHSkqVrw/SCW5YgFZkEBh72KY
+  SDfvq2xI1v+iTmBkyYqft23MKnTcc5jJ5Gh0a11a2zUstWaH1XHelXYCx44ZLCtC
+  Q6k=
+  -----END CERTIFICATE-----
+date: 2016-11-08 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: aws-sdk
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: eventmachine
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: statsd-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email: troy.ready+gems@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/sensu/transport/snssqs.rb
+homepage: https://github.com/troyready/sensu-transport-snssqs-ng
+licenses:
+- Apache-2.0
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.6.7
+signing_key: 
+specification_version: 4
+summary: Sensu transport over Amazon SNS & SQS
+test_files: []

metadata.gz.sig

@@ -0,0 +1 @@
+�c7�\JCl�;�rd�/9��f���`M5C�VA��m�I��������"f�e(!�-7O��3��Wl��}`�R������]Ffa�O~F�2�{r�njJ!X�`�)Q!*��Wڃ�7�:���P��I�A[�Pxh�p��xnO�^A�3Uֹ��l��{��eh�,c�o���׿)�C����)��-� 
�@B��̄2*�"lH=?��Ո�������v�����1Z7G�'Y[�q!i���Ts��p