logstash-input-tqp 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 22137a39127bc2546cffb562e760b17d9b5475d68d6e2452cfa67c173a4d68f1
+  data.tar.gz: 344ae3838834dd5e868dd0fdcfb57f647a46a3f62da73c3d56c49eb0bc76251e
+SHA512:
+  metadata.gz: 1c4e6eb6ba7ad4de1024509efafc4fde483371dc41415732e5bea329e28f84db03c40d493f465c41b22d30699a206187a83ea6aeb2f777d94e1ffad818ebe3c9
+  data.tar.gz: 94e89728be7bda4ce8e6a0cb3cbdd56b17bbcca4eccf73f0781c5ecce7d38aa580e1064764372c49bafecc754aee5c35773a21577c6a542c1e8bebbe79b048bf

data/Gemfile

@@ -0,0 +1,11 @@
+source 'https://rubygems.org'
+
+gemspec
+
+logstash_path = ENV["LOGSTASH_PATH"] || "../../logstash"
+use_logstash_source = ENV["LOGSTASH_SOURCE"] && ENV["LOGSTASH_SOURCE"].to_s == "1"
+
+if Dir.exist?(logstash_path) && use_logstash_source
+  gem 'logstash-core', :path => "#{logstash_path}/logstash-core"
+  gem 'logstash-core-plugin-api', :path => "#{logstash_path}/logstash-core-plugin-api"
+end

data/LICENSE

@@ -0,0 +1,202 @@
+
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright 2020 Elastic and contributors
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

data/README.md

@@ -0,0 +1,27 @@
+# logstash-input-tqp plugin
+
+This is a plugin for [Logstash](https://github.com/elastic/logstash).
+
+It is fully free and fully open source. The license is Apache 2.0, meaning you are pretty much free to use it however you want in whatever way.
+
+## Documentation
+
+TQP is an an opinionated library for pub/sub over SQS and SNS
+
+`logstash-input-tqp` is an input plugin for logstash to implements the same tqp conventions for SQS queue construction and topic subscription.
+
+```
+input {
+  tqp {
+    queue_name => "logstash-poller",
+    topics => ["my-topic"]
+    access_key_id => ENV["AWS_ACCESS_KEY_ID"],
+    secret_access_key => ENV["AWS_SECRET_ACCESS_KEY"],
+    region" => ENV["AWS_REGION"]
+  }
+}
+```
+
+The plugin will create a SQS queue (and an associated DLQ) if it doesn't exist and subscribe
+to each configured topic. SNS messages are parsed from `json`, and set as the logstash event.
+the topic name is also added to the event `@metadata`.

data/lib/logstash/inputs/sqs/patch.rb

@@ -0,0 +1,21 @@
+# This patch was stolen from logstash-plugins/logstash-output-sqs#20.
+#
+# This patch is a workaround for a JRuby issue which has been fixed in JRuby
+# 9000, but not in JRuby 1.7. See https://github.com/jruby/jruby/issues/3645
+# and https://github.com/jruby/jruby/issues/3920. This is necessary because the
+# `aws-sdk` is doing tricky name discovery to generate the correct error class.
+#
+# As per https://github.com/aws/aws-sdk-ruby/issues/1301#issuecomment-261115960,
+# this patch may be short-lived anyway.
+require 'aws-sdk'
+
+begin
+  old_stderr = $stderr
+  $stderr = StringIO.new
+
+  module Aws
+    const_set(:SQS, Aws::SQS)
+  end
+ensure
+  $stderr = old_stderr
+end

data/lib/logstash/inputs/tqp.rb

@@ -0,0 +1,256 @@
+# encoding: utf-8
+require "logstash/inputs/threadable"
+require "logstash/json"
+require "logstash/namespace"
+require "logstash/timestamp"
+require "logstash/plugin_mixins/aws_config"
+require "logstash/errors"
+require 'logstash/inputs/sqs/patch'
+
+# Forcibly load all modules marked to be lazily loaded.
+#
+# It is recommended that this is called prior to launching threads. See
+# https://aws.amazon.com/blogs/developer/threading-with-the-aws-sdk-for-ruby/.
+Aws.eager_autoload!
+
+def jsonify_hash(hash)
+  hash.map { |k,v| [k.to_s, LogStash::Json.dump(v)] }.to_h
+end
+
+def create_queue_raw(sqs_client, queue_name, attributes, tags)
+  sqs_resource = Aws::SQS::Resource.new(client: sqs_client)
+
+  return sqs_resource.create_queue(
+      queue_name: queue_name, 
+      attributes: jsonify_hash(attributes), 
+      tags: tags,
+  )
+rescue Aws::SQS::Errors::QueueNameExists
+  sqs_resource = Aws::SQS::Resource.new(client: sqs_client)
+
+  queue_url = sqs_client.get_queue_url(queue_name).queue_url
+
+  existing_tags = sqs_client.list_queue_tags(queue_url: queue_url).tags
+
+  tags_to_remove = existing_tags.keys - tags.keys
+
+  sqs_client.tag_queue(queue_url: queue_url, tags: tags)
+
+  if tags_to_remove
+      sqs_client.untag_queue(queue_url: queue_url, tag_keys: tags_to_remove)
+  end
+
+  # Run create again to make sure everything matches.
+  return sqs_resource.create_queue(
+      queue_name: queue_name, 
+      attributes: jsonify_hash(attributes), 
+      tags: tags,
+  )
+end
+
+
+def create_queue(sqs_client, queue_name, **kwargs) 
+  tags = kwargs.delete('tags') || {}
+
+  dead_letter_queue = create_queue_raw(
+      sqs_client,
+      "#{queue_name}-dead-letter",
+      {"MessageRetentionPeriod" => 1209600},  # maximum (14 days)
+      {"dlq" => "true", **tags},
+  )
+
+  dead_letter_queue_arn = dead_letter_queue.attributes["QueueArn"]
+
+  redrive_policy_kwargs = kwargs.delete("RedrivePolicy") || {}
+
+  redrivePolicy = {"maxReceiveCount" => 5}.merge(redrive_policy_kwargs)
+  redrivePolicy["deadLetterTargetArn"] = dead_letter_queue_arn
+
+  return create_queue_raw(
+      sqs_client,
+      queue_name, 
+      {"RedrivePolicy" => redrivePolicy }, 
+      tags,
+  )
+end
+
+
+class LogStash::Inputs::TQP < LogStash::Inputs::Threadable
+  include LogStash::PluginMixins::AwsConfig::V2
+
+  MAX_TIME_BEFORE_GIVING_UP = 60
+  MAX_MESSAGES_TO_FETCH = 10 # Between 1-10 in the AWS-SDK doc
+  SQS_ATTRIBUTES = ['All']
+  BACKOFF_SLEEP_TIME = 1
+  BACKOFF_FACTOR = 2
+  DEFAULT_POLLING_FREQUENCY = 20
+
+  config_name "tqp"
+
+  default :codec, "json"
+
+  config :additional_settings, :validate => :hash, :default => {}
+
+  config :queue_name, :validate => :string
+  config :prefix, :validate => :string
+  config :topics, :validate => :array
+  
+  # Polling frequency, default is 20 seconds
+  config :polling_frequency, :validate => :number, :default => DEFAULT_POLLING_FREQUENCY
+
+  attr_reader :poller
+
+  def register
+    require "aws-sdk"
+    @logger.info("Registering SQS input", 
+      queue_name: @queue_name,
+    )
+
+    setup_queue
+  end
+
+  def setup_queue
+    sqs_client = Aws::SQS::Client.new()
+    
+    @sqs_queue = create_queue(sqs_client, "#{prefix}--#{queue_name}")
+
+    subscribe_to_topics(sqs_client)
+
+    poller = Aws::SQS::QueuePoller.new(@sqs_queue.url, client: sqs_client)
+    poller.before_request { |stats| throw :stop_polling if stop? }
+
+    @poller = poller
+  rescue Aws::SQS::Errors::ServiceError, Seahorse::Client::NetworkingError => e
+    @logger.error("Cannot establish connection to Amazon SQS", exception_details(e))
+    raise LogStash::ConfigurationError, "Verify the SQS queue name and your credentials"
+  end
+
+  def subscribe_to_topics(sqs_client)
+    sns_client = Aws::SNS::Resource.new()
+
+    queue_arn = @sqs_queue.attributes["QueueArn"]
+
+    topic_arns = []
+
+    topics.each do |topic|
+      topic = "#{topic[0]}--#{topic[1]}" if topic.kind_of?(Array)
+
+      @logger.debug("Subscribing to topic", topic: "#{prefix}--#{topic}")
+
+      topic = sns_client.create_topic(name: "#{prefix}--#{topic}")
+      topic.subscribe(protocol: "sqs", endpoint: queue_arn)
+
+      topic_arns.push(topic.arn)
+    end 
+
+
+    @sqs_queue.set_attributes(
+      attributes: jsonify_hash({
+        "Policy" => {
+          "Version" => "2012-10-17", 
+          "Statement" => [
+              {
+                "Sid" => "sns",
+                "Effect" => "Allow",
+                "Principal" => {"AWS" => "*"},
+                "Action" => "SQS:SendMessage",
+                "Resource" => queue_arn,
+                "Condition" => {"ArnEquals" => {"aws:SourceArn" => topic_arns}},
+            }
+          ]
+        }
+      })
+    )
+
+  end 
+
+  def polling_options
+    { 
+      :max_number_of_messages => MAX_MESSAGES_TO_FETCH,
+      :attribute_names => SQS_ATTRIBUTES,
+      :wait_time_seconds => @polling_frequency
+    }
+  end
+
+  def add_sns_data(event, message)
+    body = LogStash::Json.load(message.body)
+
+    topic = body["TopicArn"].split(":")[-1]
+    topic = topic.delete_prefix("#{prefix}--") if prefix
+
+    message = LogStash::Json.load(body.delete("Message") || '{}')
+
+    event.set('[@metadata][topic]', topic)
+    event.set('[@metadata][body]', body)
+    
+    message.each do |key, value| 
+      event.set(key, value)
+    end
+  end 
+
+  def handle_message(message, output_queue)
+    event = LogStash::Event.new("message" => @message, "host" => @host)
+
+    add_sns_data(event, message)
+
+    decorate(event)
+    output_queue << event
+  end
+
+  def run(output_queue)
+    @logger.debug("Polling SQS queue", polling_options: polling_options)
+
+    run_with_backoff do
+      poller.poll(polling_options) do |messages, stats|
+        break if stop?
+        
+        messages.each {|message| 
+          handle_message(message, output_queue) 
+        }
+
+        @logger.debug(
+          "SQS Stats:", 
+          request_count: stats.request_count,
+          received_message_count: stats.received_message_count,
+          last_message_received_at: stats.last_message_received_at
+        ) if @logger.debug?
+      end
+    end
+  end
+
+  private
+
+  # Runs an AWS request inside a Ruby block with an exponential backoff in case
+  # we experience a ServiceError.
+  #
+  # @param [Block] block Ruby code block to execute.
+  def run_with_backoff(&block)
+    sleep_time = BACKOFF_SLEEP_TIME
+    begin
+      block.call
+    rescue Aws::SQS::Errors::ServiceError, Seahorse::Client::NetworkingError => e
+      @logger.warn("SQS error ... retrying with exponential backoff", exception_details(e, sleep_time))
+      sleep_time = backoff_sleep(sleep_time)
+      retry
+    end
+  end
+
+  def backoff_sleep(sleep_time)
+    sleep(sleep_time)
+    sleep_time > MAX_TIME_BEFORE_GIVING_UP ? sleep_time : sleep_time * BACKOFF_FACTOR
+  end
+
+  def convert_epoch_to_timestamp(time)
+    LogStash::Timestamp.at(time.to_i / 1000)
+  end
+
+  def exception_details(e, sleep_time = nil)
+    details = { :queue_name => @queue_name, :exception => e.class, :message => e.message }
+    details[:code] = e.code if e.is_a?(Aws::SQS::Errors::ServiceError) && e.code
+    details[:cause] = e.original_error if e.respond_to?(:original_error) && e.original_error # Seahorse::Client::NetworkingError
+    details[:sleep_time] = sleep_time if sleep_time
+    details[:backtrace] = e.backtrace if @logger.debug?
+    details
+  end
+
+end # class LogStash::Inputs::TQP

data/logstash-input-tqp.gemspec

@@ -0,0 +1,29 @@
+Gem::Specification.new do |s|
+  s.name            = 'logstash-input-tqp'
+  s.version         = '0.1.0'
+  s.licenses        = ['Apache-2.0']
+  s.summary         = "An opinionated library for pub/sub over SQS and SNS"
+  s.description     = "This gem is a Logstash plugin required to be installed on top of the Logstash core pipeline using $LS_HOME/bin/logstash-plugin install gemname. This gem is not a stand-alone program"
+  s.authors         = ["jason quense"]
+  s.email           = 'monastic.panic@gmailc.om'
+  s.require_paths = ["lib"]
+
+  # Files
+  s.files = Dir["lib/**/*","spec/**/*","*.gemspec","*.md","CONTRIBUTORS","Gemfile","LICENSE","NOTICE.TXT", "vendor/jar-dependencies/**/*.jar", "vendor/jar-dependencies/**/*.rb", "VERSION", "docs/**/*"]
+
+  # Tests
+  s.test_files = s.files.grep(%r{^(test|spec|features)/})
+
+  # Special flag to let us know this is actually a logstash plugin
+  s.metadata = { "logstash_plugin" => "true", "logstash_group" => "input" }
+
+  # Gem dependencies
+  s.add_runtime_dependency "logstash-core-plugin-api", ">= 1.60", "<= 2.99"
+
+  s.add_runtime_dependency 'logstash-codec-json'
+  s.add_runtime_dependency 'logstash-mixin-aws', '>= 5.1.0'
+
+  s.add_development_dependency 'logstash-devutils'
+  s.add_development_dependency "logstash-codec-json_lines"
+end
+

data/spec/inputs/tqp_spec.rb

@@ -0,0 +1,178 @@
+# # encoding: utf-8
+# require "logstash/devutils/rspec/spec_helper"
+# require "logstash/devutils/rspec/shared_examples"
+# require "logstash/inputs/tqp"
+# require "logstash/errors"
+# require "logstash/event"
+# require "logstash/json"
+# require "aws-sdk"
+# require "ostruct"
+
+# describe LogStash::Inputs::TQP do
+
+#   let(:queue_name) { "my-poller" }
+#   let(:queue_url) { "https://sqs.test.local/#{queue_name}" }
+#   let(:config) do
+#     {
+#       "region" => "us-east-1",
+#       "access_key_id" =>  "123",
+#       "secret_access_key" =>  "secret",
+#       "queue_name" =>  :queue_name,
+#       "prefix" => 'my-service',
+#       "topics" => ['topic--1', 'topic--2']
+#     }
+#   end
+
+#   let(:input) { LogStash::Inputs::TQP.new(config) }
+#   # let(:decoded_message) { { "bonjour" => "awesome" }  }
+#   # let(:encoded_message)  { double("sqs_message", :body => LogStash::Json::dump(decoded_message)) }
+
+#   subject { input }
+
+#   let(:mock_sqs) {
+#    sqs = Aws::SQS::Client.new({ :stub_responses => true })
+#    sqs.stub_responses(:get_queue_url, {queue_url: queue_url})
+#    sqs.stub_responses(:get_queue_attributes, {attributes: {"QueueArn" => 'aws::sqs:::coolthing'}})
+#    sqs
+#   }
+
+#   context "valid credentials" do
+#     let(:queue) { [] }
+
+#     it "doesn't raise an error with valid credentials" do
+#       expect(Aws::SQS::Client).to receive(:new).and_return(mock_sqs)
+#       # expect(mock_sqs).to receive(:get_queue_url).with(queue_name: queue_name).and_return(queue_url: 'hi')
+#       mock_sqs
+
+#       # puts 'hi', mock_sqs.get_queue_url(queue_name: queue_name )
+#       subject.register
+#       # expect { subject.register }.not_to raise_error
+#     end
+
+#     # context "when interrupting the plugin" do
+#     #   before do
+#     #     expect(Aws::SQS::Client).to receive(:new).and_return(mock_sqs)
+#     #     expect(mock_sqs).to receive(:get_queue_url).with({ :queue_name => queue_name }).and_return({:queue_url => queue_url })
+#     #     expect(subject).to receive(:poller).and_return(mock_sqs).at_least(:once)
+
+#     #     # We have to make sure we create a bunch of events
+#     #     # so we actually really try to stop the plugin.
+
+#     #     #
+#     #     # rspec's `and_yield` allow you to define a fix amount of possible
+#     #     # yielded values and doesn't allow you to create infinite loop.
+#     #     # And since we are actually creating thread we need to make sure
+#     #     # we have enough work to keep the thread working until we kill it..
+#     #     #
+#     #     # I haven't found a way to make it rspec friendly
+#     #     mock_sqs.instance_eval do
+#     #       def poll(polling_options = {})
+#     #         loop do
+#     #           yield [OpenStruct.new(:body => LogStash::Json::dump({ "message" => "hello world"}))], OpenStruct.new
+#     #         end
+#     #       end
+#     #     end
+#     #   end
+
+#     #   it_behaves_like "an interruptible input plugin"
+#     # end
+
+#     # context "enrich event" do
+#     #   let(:event) { LogStash::Event.new }
+
+#     #   let(:message_id) { "123" }
+#     #   let(:md5_of_body) { "dr strange" }
+#     #   let(:sent_timestamp) { LogStash::Timestamp.new }
+#     #   let(:epoch_timestamp) { (sent_timestamp.utc.to_f * 1000).to_i }
+
+#     #   let(:id_field) { "my_id_field" }
+#     #   let(:md5_field) { "my_md5_field" }
+#     #   let(:sent_timestamp_field) { "my_sent_timestamp_field" }
+
+#     #   let(:message) do
+#     #     double("message", :message_id => message_id, :md5_of_body => md5_of_body, :attributes => { LogStash::Inputs::TQP::SENT_TIMESTAMP  => epoch_timestamp } )
+
+#     #   end
+
+#     #   subject { input.add_sqs_data(event, message) }
+
+#     #   context "when the option is specified" do
+#     #     let(:config) do
+#     #       {
+#     #         "region" => "us-east-1",
+#     #         "access_key_id" => "123",
+#     #         "secret_access_key" => "secret",
+#     #         "queue" => queue_name,
+#     #         "id_field" => id_field,
+#     #         "md5_field" => md5_field,
+#     #         "sent_timestamp_field" => sent_timestamp_field
+#     #       }
+#     #     end
+
+#     #     it "add the `message_id`" do
+#     #       expect(subject.get(id_field)).to eq(message_id)
+#     #     end
+
+#     #     it "add the `md5_of_body`" do
+#     #       expect(subject.get(md5_field)).to eq(md5_of_body)
+#     #     end
+
+#     #     it "add the `sent_timestamp`" do
+#     #       expect(subject.get(sent_timestamp_field).to_i).to eq(sent_timestamp.to_i)
+#     #     end
+#     #   end
+
+#     #   context "when the option isn't specified" do
+#     #     it "doesnt add the `message_id`" do
+#     #       expect(subject).not_to include(id_field)
+#     #     end
+
+#     #     it "doesnt add the `md5_of_body`" do
+#     #       expect(subject).not_to include(md5_field)
+#     #     end
+
+#     #     it "doesnt add the `sent_timestamp`" do
+#     #       expect(subject).not_to include(sent_timestamp_field)
+#     #     end
+#     #   end
+#     # end
+
+#     # context "when decoding body" do
+#     #   subject { LogStash::Inputs::TQP::new(config.merge({ "codec" => "json" })) }
+
+#     #   it "uses the specified codec" do
+#     #     subject.handle_message(encoded_message, queue)
+#     #     expect(queue.pop.get("bonjour")).to eq(decoded_message["bonjour"])
+#     #   end
+#     # end
+
+#     # context "receiving messages" do
+
+#     #   before do
+#     #     expect(subject).to receive(:poller).and_return(mock_sqs).at_least(:once)
+#     #   end
+
+#     #   it "creates logstash event" do
+#     #     expect(mock_sqs).to receive(:poll).with(anything()).and_yield([encoded_message], double("stats"))
+#     #     subject.run(queue)
+#     #     expect(queue.pop.get("bonjour")).to eq(decoded_message["bonjour"])
+#     #   end
+
+#     #   context 'can create multiple events' do
+#     #     require "logstash/codecs/json_lines"
+#     #     let(:config) { super().merge({ "codec" => "json_lines" }) }
+#     #     let(:first_message) { { "sequence" => "first" } }
+#     #     let(:second_message) { { "sequence" => "second" } }
+#     #     let(:encoded_message)  { double("sqs_message", :body => "#{LogStash::Json::dump(first_message)}\n#{LogStash::Json::dump(second_message)}\n") }
+
+#     #     it 'creates multiple events' do
+#     #       expect(mock_sqs).to receive(:poll).with(anything()).and_yield([encoded_message], double("stats"))
+#     #       subject.run(queue)
+#     #       events = queue.map{ |e|e.get('sequence')}
+#     #       expect(events).to match_array([first_message['sequence'], second_message['sequence']])
+#     #     end
+#     #   end
+#     # end
+
+#   end
+# end

data/spec/integration/tqp_spec.rb

@@ -0,0 +1,74 @@
+# encoding: utf-8
+require "spec_helper"
+require "logstash/inputs/tqp"
+require "logstash/event"
+require "logstash/json"
+require "aws-sdk"
+require_relative "../support/helpers"
+require "thread"
+
+Thread.abort_on_exception = true
+
+describe "LogStash::Inputs::TQP integration" do
+  let(:decoded_message) { { "drstrange" => "is-he-really-that-strange" } }
+  let(:encoded_message) { LogStash::Json.dump(decoded_message) }
+  let(:queue) { Queue.new }
+  let(:prefix) { "jq" }
+  let(:input) { 
+    puts options
+    LogStash::Inputs::TQP.new(options) 
+  }
+
+  context "with invalid credentials" do
+    let(:options) do
+      {
+        "queue_name" => 'test-queue-1',
+        "access_key_id" => "bad_access",
+        "secret_access_key" => "bad_secret_key",
+        "region" => ENV["AWS_REGION"],
+        "topics": []
+      }
+    end
+
+    subject { input }
+
+    it "raises a Configuration error if the credentials are bad" do
+      expect { subject.register }.to raise_error(LogStash::ConfigurationError)
+    end
+  end
+
+  context "with valid credentials" do
+    let(:options) do
+      {
+        "queue_name" => 'test-queue-1',
+        "prefix" => prefix,
+        "topics" => ['studies--updated'],
+        "access_key_id" => ENV['AWS_ACCESS_KEY_ID'],
+        "secret_access_key" => ENV['AWS_SECRET_ACCESS_KEY'],
+        "region" => ENV["AWS_REGION"]
+      }
+    end
+
+    before :each do
+      input.register
+
+      @server = Thread.new { input.run(queue) }
+    end
+
+    after do
+      @server.kill
+    end
+
+    subject { 
+      push_sns_event("#{prefix}--studies--updated", {"foo" => "bar"})
+      queue.pop 
+    }
+
+    it "creates logstash events" do
+      puts subject.to_json()
+
+      expect(subject.get('[@metadata][topic]')).to eq('studies--updated')
+      expect(subject.get('foo')).to eq('bar')
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -0,0 +1,2 @@
+# encoding: utf-8
+require "logstash/devutils/rspec/spec_helper"

data/spec/support/helpers.rb

@@ -0,0 +1,18 @@
+require 'logstash/json'
+
+# encoding: utf-8
+def push_sqs_event(message)
+  client = Aws::SQS::Client.new
+  queue_url = client.get_queue_url(queue_name: ENV['SQS_QUEUE_NAME'])
+
+  client.send_message({ queue_url: queue_url.queue_url, message_body: message })
+end
+
+def push_sns_event(topic, message)
+  resource = Aws::SNS::Resource.new
+
+  puts "sending message #{topic}"
+
+  topic = resource.create_topic(name: topic)
+  topic.publish(message: LogStash::Json.dump(message))
+end

metadata

@@ -0,0 +1,137 @@
+--- !ruby/object:Gem::Specification
+name: logstash-input-tqp
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- jason quense
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2022-05-27 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.60'
+    - - "<="
+      - !ruby/object:Gem::Version
+        version: '2.99'
+  name: logstash-core-plugin-api
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.60'
+    - - "<="
+      - !ruby/object:Gem::Version
+        version: '2.99'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: logstash-codec-json
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.1.0
+  name: logstash-mixin-aws
+  prerelease: false
+  type: :runtime
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.1.0
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: logstash-devutils
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  name: logstash-codec-json_lines
+  prerelease: false
+  type: :development
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: This gem is a Logstash plugin required to be installed on top of the
+  Logstash core pipeline using $LS_HOME/bin/logstash-plugin install gemname. This
+  gem is not a stand-alone program
+email: monastic.panic@gmailc.om
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- LICENSE
+- README.md
+- lib/logstash/inputs/sqs/patch.rb
+- lib/logstash/inputs/tqp.rb
+- logstash-input-tqp.gemspec
+- spec/inputs/tqp_spec.rb
+- spec/integration/tqp_spec.rb
+- spec/spec_helper.rb
+- spec/support/helpers.rb
+homepage:
+licenses:
+- Apache-2.0
+metadata:
+  logstash_plugin: 'true'
+  logstash_group: input
+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.7.10
+signing_key:
+specification_version: 4
+summary: An opinionated library for pub/sub over SQS and SNS
+test_files:
+- spec/inputs/tqp_spec.rb
+- spec/integration/tqp_spec.rb
+- spec/spec_helper.rb
+- spec/support/helpers.rb