logstash-output-sqs 0.1.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,15 @@
1
+ ---
2
+ !binary "U0hBMQ==":
3
+ metadata.gz: !binary |-
4
+ NDJlY2U2NWQ5OTNlYmZkOGQyZTUxYmFjNzM4YmVjNDFlODIwMmExNw==
5
+ data.tar.gz: !binary |-
6
+ MDNiY2NlNWM0N2ZlNDU4NTZmMDY3OWI2MWRkYzg1YzQzZTNlNjk3OA==
7
+ SHA512:
8
+ metadata.gz: !binary |-
9
+ YzgzZjBjN2M5ZjM1ZjEyNzdkNTQwNTcxN2QwYzE4YzhlZmQ3NDhiODRiOWVk
10
+ ZTNlN2MyYzNlNDRiZjdmNmU0NzZhM2VhMWI0YTNlNjg2ZGUwMmNhM2MxYjNh
11
+ MmI1M2I3NDU3Nzk3YzQzYzBiMDA5NzFiY2NkNzBjMjk1Y2FmMjM=
12
+ data.tar.gz: !binary |-
13
+ YzNhZTY4OWRhZjIwOTM5OWZkNzU4N2UxOTgzMGFmZTE4M2VhMGU1YjZkYjZl
14
+ MmVkMTVmZjhkZjA3MGFlZTg0OWFlYjFiN2ZmMDAzNmNjZjUzYjBjYTI2MmY3
15
+ ZDk4ZTM0ZjZlOGE1MTBjNzY0MjM1ZjQzYTU1Nzc5MmVhNjMwZGY=
@@ -0,0 +1,4 @@
1
+ *.gem
2
+ Gemfile.lock
3
+ .bundle
4
+ vendor
data/Gemfile ADDED
@@ -0,0 +1,4 @@
1
+ source 'http://rubygems.org'
2
+ gem 'rake'
3
+ gem 'gem_publisher'
4
+ gem 'archive-tar-minitar'
data/LICENSE ADDED
@@ -0,0 +1,13 @@
1
+ Copyright (c) 2012-2014 Elasticsearch <http://www.elasticsearch.org>
2
+
3
+ Licensed under the Apache License, Version 2.0 (the "License");
4
+ you may not use this file except in compliance with the License.
5
+ You may obtain a copy of the License at
6
+
7
+ http://www.apache.org/licenses/LICENSE-2.0
8
+
9
+ Unless required by applicable law or agreed to in writing, software
10
+ distributed under the License is distributed on an "AS IS" BASIS,
11
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12
+ See the License for the specific language governing permissions and
13
+ limitations under the License.
@@ -0,0 +1,6 @@
1
+ @files=[]
2
+
3
+ task :default do
4
+ system("rake -T")
5
+ end
6
+
@@ -0,0 +1,140 @@
1
+ # encoding: utf-8
2
+ require "logstash/outputs/base"
3
+ require "logstash/namespace"
4
+ require "logstash/plugin_mixins/aws_config"
5
+ require "stud/buffer"
6
+ require "digest/sha2"
7
+
8
+ # Push events to an Amazon Web Services Simple Queue Service (SQS) queue.
9
+ #
10
+ # SQS is a simple, scalable queue system that is part of the
11
+ # Amazon Web Services suite of tools.
12
+ #
13
+ # Although SQS is similar to other queuing systems like AMQP, it
14
+ # uses a custom API and requires that you have an AWS account.
15
+ # See http://aws.amazon.com/sqs/ for more details on how SQS works,
16
+ # what the pricing schedule looks like and how to setup a queue.
17
+ #
18
+ # To use this plugin, you *must*:
19
+ #
20
+ # * Have an AWS account
21
+ # * Setup an SQS queue
22
+ # * Create an identify that has access to publish messages to the queue.
23
+ #
24
+ # The "consumer" identity must have the following permissions on the queue:
25
+ #
26
+ # * sqs:ChangeMessageVisibility
27
+ # * sqs:ChangeMessageVisibilityBatch
28
+ # * sqs:GetQueueAttributes
29
+ # * sqs:GetQueueUrl
30
+ # * sqs:ListQueues
31
+ # * sqs:SendMessage
32
+ # * sqs:SendMessageBatch
33
+ #
34
+ # Typically, you should setup an IAM policy, create a user and apply the IAM policy to the user.
35
+ # A sample policy is as follows:
36
+ #
37
+ # {
38
+ # "Statement": [
39
+ # {
40
+ # "Sid": "Stmt1347986764948",
41
+ # "Action": [
42
+ # "sqs:ChangeMessageVisibility",
43
+ # "sqs:ChangeMessageVisibilityBatch",
44
+ # "sqs:DeleteMessage",
45
+ # "sqs:DeleteMessageBatch",
46
+ # "sqs:GetQueueAttributes",
47
+ # "sqs:GetQueueUrl",
48
+ # "sqs:ListQueues",
49
+ # "sqs:ReceiveMessage"
50
+ # ],
51
+ # "Effect": "Allow",
52
+ # "Resource": [
53
+ # "arn:aws:sqs:us-east-1:200850199751:Logstash"
54
+ # ]
55
+ # }
56
+ # ]
57
+ # }
58
+ #
59
+ # See http://aws.amazon.com/iam/ for more details on setting up AWS identities.
60
+ #
61
+ class LogStash::Outputs::SQS < LogStash::Outputs::Base
62
+ include LogStash::PluginMixins::AwsConfig
63
+ include Stud::Buffer
64
+
65
+ config_name "sqs"
66
+ milestone 1
67
+
68
+ # Name of SQS queue to push messages into. Note that this is just the name of the queue, not the URL or ARN.
69
+ config :queue, :validate => :string, :required => true
70
+
71
+ # Set to true if you want send messages to SQS in batches with batch_send
72
+ # from the amazon sdk
73
+ config :batch, :validate => :boolean, :default => true
74
+
75
+ # If batch is set to true, the number of events we queue up for a batch_send.
76
+ config :batch_events, :validate => :number, :default => 10
77
+
78
+ # If batch is set to true, the maximum amount of time between batch_send commands when there are pending events to flush.
79
+ config :batch_timeout, :validate => :number, :default => 5
80
+
81
+ public
82
+ def aws_service_endpoint(region)
83
+ return {
84
+ :sqs_endpoint => "sqs.#{region}.amazonaws.com"
85
+ }
86
+ end
87
+
88
+ public
89
+ def register
90
+ require "aws-sdk"
91
+
92
+ @sqs = AWS::SQS.new(aws_options_hash)
93
+
94
+ if @batch
95
+ if @batch_events > 10
96
+ raise RuntimeError.new(
97
+ "AWS only allows a batch_events parameter of 10 or less"
98
+ )
99
+ elsif @batch_events <= 1
100
+ raise RuntimeError.new(
101
+ "batch_events parameter must be greater than 1 (or its not a batch)"
102
+ )
103
+ end
104
+ buffer_initialize(
105
+ :max_items => @batch_events,
106
+ :max_interval => @batch_timeout,
107
+ :logger => @logger
108
+ )
109
+ end
110
+
111
+ begin
112
+ @logger.debug("Connecting to AWS SQS queue '#{@queue}'...")
113
+ @sqs_queue = @sqs.queues.named(@queue)
114
+ @logger.info("Connected to AWS SQS queue '#{@queue}' successfully.")
115
+ rescue Exception => e
116
+ @logger.error("Unable to access SQS queue '#{@queue}': #{e.to_s}")
117
+ end # begin/rescue
118
+ end # def register
119
+
120
+ public
121
+ def receive(event)
122
+ if @batch
123
+ buffer_receive(event.to_json)
124
+ return
125
+ end
126
+ @sqs_queue.send_message(event.to_json)
127
+ end # def receive
128
+
129
+ # called from Stud::Buffer#buffer_flush when there are events to flush
130
+ def flush(events, teardown=false)
131
+ @sqs_queue.batch_send(events)
132
+ end
133
+
134
+ public
135
+ def teardown
136
+ buffer_flush(:final => true)
137
+ @sqs_queue = nil
138
+ finished
139
+ end # def teardown
140
+ end
@@ -0,0 +1,29 @@
1
+ Gem::Specification.new do |s|
2
+
3
+ s.name = 'logstash-output-sqs'
4
+ s.version = '0.1.0'
5
+ s.licenses = ['Apache License (2.0)']
6
+ s.summary = "Push events to an Amazon Web Services Simple Queue Service (SQS) queue."
7
+ s.description = "Push events to an Amazon Web Services Simple Queue Service (SQS) queue."
8
+ s.authors = ["Elasticsearch"]
9
+ s.email = 'richard.pijnenburg@elasticsearch.com'
10
+ s.homepage = "http://logstash.net/"
11
+ s.require_paths = ["lib"]
12
+
13
+ # Files
14
+ s.files = `git ls-files`.split($\)+::Dir.glob('vendor/*')
15
+
16
+ # Tests
17
+ s.test_files = s.files.grep(%r{^(test|spec|features)/})
18
+
19
+ # Special flag to let us know this is actually a logstash plugin
20
+ s.metadata = { "logstash_plugin" => "true", "group" => "output" }
21
+
22
+ # Gem dependencies
23
+ s.add_runtime_dependency 'logstash', '>= 1.4.0', '< 2.0.0'
24
+
25
+ s.add_runtime_dependency 'aws-sdk'
26
+ s.add_runtime_dependency 'stud'
27
+
28
+ end
29
+
@@ -0,0 +1,9 @@
1
+ require "gem_publisher"
2
+
3
+ desc "Publish gem to RubyGems.org"
4
+ task :publish_gem do |t|
5
+ gem_file = Dir.glob(File.expand_path('../*.gemspec',File.dirname(__FILE__))).first
6
+ gem = GemPublisher.publish_if_updated(gem_file, :rubygems)
7
+ puts "Published #{gem}" if gem
8
+ end
9
+
@@ -0,0 +1,169 @@
1
+ require "net/http"
2
+ require "uri"
3
+ require "digest/sha1"
4
+
5
+ def vendor(*args)
6
+ return File.join("vendor", *args)
7
+ end
8
+
9
+ directory "vendor/" => ["vendor"] do |task, args|
10
+ mkdir task.name
11
+ end
12
+
13
+ def fetch(url, sha1, output)
14
+
15
+ puts "Downloading #{url}"
16
+ actual_sha1 = download(url, output)
17
+
18
+ if actual_sha1 != sha1
19
+ fail "SHA1 does not match (expected '#{sha1}' but got '#{actual_sha1}')"
20
+ end
21
+ end # def fetch
22
+
23
+ def file_fetch(url, sha1)
24
+ filename = File.basename( URI(url).path )
25
+ output = "vendor/#{filename}"
26
+ task output => [ "vendor/" ] do
27
+ begin
28
+ actual_sha1 = file_sha1(output)
29
+ if actual_sha1 != sha1
30
+ fetch(url, sha1, output)
31
+ end
32
+ rescue Errno::ENOENT
33
+ fetch(url, sha1, output)
34
+ end
35
+ end.invoke
36
+
37
+ return output
38
+ end
39
+
40
+ def file_sha1(path)
41
+ digest = Digest::SHA1.new
42
+ fd = File.new(path, "r")
43
+ while true
44
+ begin
45
+ digest << fd.sysread(16384)
46
+ rescue EOFError
47
+ break
48
+ end
49
+ end
50
+ return digest.hexdigest
51
+ ensure
52
+ fd.close if fd
53
+ end
54
+
55
+ def download(url, output)
56
+ uri = URI(url)
57
+ digest = Digest::SHA1.new
58
+ tmp = "#{output}.tmp"
59
+ Net::HTTP.start(uri.host, uri.port, :use_ssl => (uri.scheme == "https")) do |http|
60
+ request = Net::HTTP::Get.new(uri.path)
61
+ http.request(request) do |response|
62
+ fail "HTTP fetch failed for #{url}. #{response}" if [200, 301].include?(response.code)
63
+ size = (response["content-length"].to_i || -1).to_f
64
+ count = 0
65
+ File.open(tmp, "w") do |fd|
66
+ response.read_body do |chunk|
67
+ fd.write(chunk)
68
+ digest << chunk
69
+ if size > 0 && $stdout.tty?
70
+ count += chunk.bytesize
71
+ $stdout.write(sprintf("\r%0.2f%%", count/size * 100))
72
+ end
73
+ end
74
+ end
75
+ $stdout.write("\r \r") if $stdout.tty?
76
+ end
77
+ end
78
+
79
+ File.rename(tmp, output)
80
+
81
+ return digest.hexdigest
82
+ rescue SocketError => e
83
+ puts "Failure while downloading #{url}: #{e}"
84
+ raise
85
+ ensure
86
+ File.unlink(tmp) if File.exist?(tmp)
87
+ end # def download
88
+
89
+ def untar(tarball, &block)
90
+ require "archive/tar/minitar"
91
+ tgz = Zlib::GzipReader.new(File.open(tarball))
92
+ # Pull out typesdb
93
+ tar = Archive::Tar::Minitar::Input.open(tgz)
94
+ tar.each do |entry|
95
+ path = block.call(entry)
96
+ next if path.nil?
97
+ parent = File.dirname(path)
98
+
99
+ mkdir_p parent unless File.directory?(parent)
100
+
101
+ # Skip this file if the output file is the same size
102
+ if entry.directory?
103
+ mkdir path unless File.directory?(path)
104
+ else
105
+ entry_mode = entry.instance_eval { @mode } & 0777
106
+ if File.exists?(path)
107
+ stat = File.stat(path)
108
+ # TODO(sissel): Submit a patch to archive-tar-minitar upstream to
109
+ # expose headers in the entry.
110
+ entry_size = entry.instance_eval { @size }
111
+ # If file sizes are same, skip writing.
112
+ next if stat.size == entry_size && (stat.mode & 0777) == entry_mode
113
+ end
114
+ puts "Extracting #{entry.full_name} from #{tarball} #{entry_mode.to_s(8)}"
115
+ File.open(path, "w") do |fd|
116
+ # eof? check lets us skip empty files. Necessary because the API provided by
117
+ # Archive::Tar::Minitar::Reader::EntryStream only mostly acts like an
118
+ # IO object. Something about empty files in this EntryStream causes
119
+ # IO.copy_stream to throw "can't convert nil into String" on JRuby
120
+ # TODO(sissel): File a bug about this.
121
+ while !entry.eof?
122
+ chunk = entry.read(16384)
123
+ fd.write(chunk)
124
+ end
125
+ #IO.copy_stream(entry, fd)
126
+ end
127
+ File.chmod(entry_mode, path)
128
+ end
129
+ end
130
+ tar.close
131
+ File.unlink(tarball) if File.file?(tarball)
132
+ end # def untar
133
+
134
+ def ungz(file)
135
+
136
+ outpath = file.gsub('.gz', '')
137
+ tgz = Zlib::GzipReader.new(File.open(file))
138
+ begin
139
+ File.open(outpath, "w") do |out|
140
+ IO::copy_stream(tgz, out)
141
+ end
142
+ File.unlink(file)
143
+ rescue
144
+ File.unlink(outpath) if File.file?(outpath)
145
+ raise
146
+ end
147
+ tgz.close
148
+ end
149
+
150
+ desc "Process any vendor files required for this plugin"
151
+ task "vendor" do |task, args|
152
+
153
+ @files.each do |file|
154
+ download = file_fetch(file['url'], file['sha1'])
155
+ if download =~ /.tar.gz/
156
+ prefix = download.gsub('.tar.gz', '').gsub('vendor/', '')
157
+ untar(download) do |entry|
158
+ if !file['files'].nil?
159
+ next unless file['files'].include?(entry.full_name.gsub(prefix, ''))
160
+ out = entry.full_name.split("/").last
161
+ end
162
+ File.join('vendor', out)
163
+ end
164
+ elsif download =~ /.gz/
165
+ ungz(download)
166
+ end
167
+ end
168
+
169
+ end
@@ -0,0 +1 @@
1
+ require 'spec_helper'
metadata ADDED
@@ -0,0 +1,103 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: logstash-output-sqs
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.1.0
5
+ platform: ruby
6
+ authors:
7
+ - Elasticsearch
8
+ autorequire:
9
+ bindir: bin
10
+ cert_chain: []
11
+ date: 2014-11-06 00:00:00.000000000 Z
12
+ dependencies:
13
+ - !ruby/object:Gem::Dependency
14
+ name: logstash
15
+ requirement: !ruby/object:Gem::Requirement
16
+ requirements:
17
+ - - ! '>='
18
+ - !ruby/object:Gem::Version
19
+ version: 1.4.0
20
+ - - <
21
+ - !ruby/object:Gem::Version
22
+ version: 2.0.0
23
+ type: :runtime
24
+ prerelease: false
25
+ version_requirements: !ruby/object:Gem::Requirement
26
+ requirements:
27
+ - - ! '>='
28
+ - !ruby/object:Gem::Version
29
+ version: 1.4.0
30
+ - - <
31
+ - !ruby/object:Gem::Version
32
+ version: 2.0.0
33
+ - !ruby/object:Gem::Dependency
34
+ name: aws-sdk
35
+ requirement: !ruby/object:Gem::Requirement
36
+ requirements:
37
+ - - ! '>='
38
+ - !ruby/object:Gem::Version
39
+ version: '0'
40
+ type: :runtime
41
+ prerelease: false
42
+ version_requirements: !ruby/object:Gem::Requirement
43
+ requirements:
44
+ - - ! '>='
45
+ - !ruby/object:Gem::Version
46
+ version: '0'
47
+ - !ruby/object:Gem::Dependency
48
+ name: stud
49
+ requirement: !ruby/object:Gem::Requirement
50
+ requirements:
51
+ - - ! '>='
52
+ - !ruby/object:Gem::Version
53
+ version: '0'
54
+ type: :runtime
55
+ prerelease: false
56
+ version_requirements: !ruby/object:Gem::Requirement
57
+ requirements:
58
+ - - ! '>='
59
+ - !ruby/object:Gem::Version
60
+ version: '0'
61
+ description: Push events to an Amazon Web Services Simple Queue Service (SQS) queue.
62
+ email: richard.pijnenburg@elasticsearch.com
63
+ executables: []
64
+ extensions: []
65
+ extra_rdoc_files: []
66
+ files:
67
+ - .gitignore
68
+ - Gemfile
69
+ - LICENSE
70
+ - Rakefile
71
+ - lib/logstash/outputs/sqs.rb
72
+ - logstash-output-sqs.gemspec
73
+ - rakelib/publish.rake
74
+ - rakelib/vendor.rake
75
+ - spec/outputs/sqs_spec.rb
76
+ homepage: http://logstash.net/
77
+ licenses:
78
+ - Apache License (2.0)
79
+ metadata:
80
+ logstash_plugin: 'true'
81
+ group: output
82
+ post_install_message:
83
+ rdoc_options: []
84
+ require_paths:
85
+ - lib
86
+ required_ruby_version: !ruby/object:Gem::Requirement
87
+ requirements:
88
+ - - ! '>='
89
+ - !ruby/object:Gem::Version
90
+ version: '0'
91
+ required_rubygems_version: !ruby/object:Gem::Requirement
92
+ requirements:
93
+ - - ! '>='
94
+ - !ruby/object:Gem::Version
95
+ version: '0'
96
+ requirements: []
97
+ rubyforge_project:
98
+ rubygems_version: 2.4.1
99
+ signing_key:
100
+ specification_version: 4
101
+ summary: Push events to an Amazon Web Services Simple Queue Service (SQS) queue.
102
+ test_files:
103
+ - spec/outputs/sqs_spec.rb