mq-tail 0.0.2

Sign up to get free protection for your applications and to get access to all the features.
data/.document ADDED
@@ -0,0 +1,5 @@
1
+ lib/**/*.rb
2
+ bin/*
3
+ -
4
+ features/**/*.feature
5
+ LICENSE.txt
data/.rspec ADDED
@@ -0,0 +1 @@
1
+ --color
data/Gemfile ADDED
@@ -0,0 +1,19 @@
1
+ source "http://rubygems.org"
2
+ # Add dependencies required to use your gem here.
3
+ # Example:
4
+ # gem "activesupport", ">= 2.3.5"
5
+
6
+ # Add dependencies to develop your gem here.
7
+ # Include everything needed to run rake, tests, features, etc.
8
+ group :development do
9
+ gem "rspec", "~> 2.8.0"
10
+ gem "yard", "~> 0.7"
11
+ gem "bundler", "~> 1.1.5"
12
+ gem "jeweler", "~> 1.8.4"
13
+ gem "pry", "~> 0.9.10"
14
+ end
15
+
16
+ gem "main", "~> 5.0.1"
17
+ gem "amqp", "~> 0.9.7"
18
+ gem "em-synchrony", "~> 1.0.2"
19
+ gem "json", "~> 1.7.4"
data/LICENSE.txt ADDED
@@ -0,0 +1,20 @@
1
+ Copyright (c) 2012 Lucas Hansen
2
+
3
+ Permission is hereby granted, free of charge, to any person obtaining
4
+ a copy of this software and associated documentation files (the
5
+ "Software"), to deal in the Software without restriction, including
6
+ without limitation the rights to use, copy, modify, merge, publish,
7
+ distribute, sublicense, and/or sell copies of the Software, and to
8
+ permit persons to whom the Software is furnished to do so, subject to
9
+ the following conditions:
10
+
11
+ The above copyright notice and this permission notice shall be
12
+ included in all copies or substantial portions of the Software.
13
+
14
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
15
+ EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
16
+ MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
17
+ NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
18
+ LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
19
+ OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
20
+ WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
data/README.rdoc ADDED
@@ -0,0 +1,9 @@
1
+ = mq-tail
2
+
3
+ A tail like utility for watching a RabbitMQ queue.
4
+
5
+ == Copyright
6
+
7
+ Copyright (c) 2012 Lucas Hansen. See LICENSE.txt for
8
+ further details.
9
+
data/Rakefile ADDED
@@ -0,0 +1,42 @@
1
+ # encoding: utf-8
2
+
3
+ require 'rubygems'
4
+ require 'bundler'
5
+ begin
6
+ Bundler.setup(:default, :development)
7
+ rescue Bundler::BundlerError => e
8
+ $stderr.puts e.message
9
+ $stderr.puts "Run `bundle install` to install missing gems"
10
+ exit e.status_code
11
+ end
12
+ require 'rake'
13
+
14
+ require 'jeweler'
15
+ Jeweler::Tasks.new do |gem|
16
+ # gem is a Gem::Specification... see http://docs.rubygems.org/read/chapter/20 for more options
17
+ gem.name = "mq-tail"
18
+ gem.homepage = "http://github.com/Veered/mq-tail"
19
+ gem.license = "MIT"
20
+ gem.summary = %Q{A tail like utility for watching a RabbitMQ queue. }
21
+ gem.description = %Q{A tail like utility for watching a RabbitMQ queue. }
22
+ gem.email = "lucas@likeness.com"
23
+ gem.authors = ["Lucas Hansen"]
24
+ # dependencies defined in Gemfile
25
+ end
26
+ Jeweler::RubygemsDotOrgTasks.new
27
+
28
+ require 'rspec/core'
29
+ require 'rspec/core/rake_task'
30
+ RSpec::Core::RakeTask.new(:spec) do |spec|
31
+ spec.pattern = FileList['spec/**/*_spec.rb']
32
+ end
33
+
34
+ RSpec::Core::RakeTask.new(:rcov) do |spec|
35
+ spec.pattern = 'spec/**/*_spec.rb'
36
+ spec.rcov = true
37
+ end
38
+
39
+ task :default => :spec
40
+
41
+ require 'yard'
42
+ YARD::Rake::YardocTask.new
data/VERSION ADDED
@@ -0,0 +1 @@
1
+ 0.0.2
data/bin/mqtail ADDED
@@ -0,0 +1,173 @@
1
+ #!/usr/bin/env ruby
2
+ require 'main'
3
+
4
+ require 'em-synchrony'
5
+ require 'em-synchrony/amqp'
6
+
7
+ require 'json'
8
+
9
+ Main {
10
+
11
+ option('config', 'c'){
12
+ description 'A config file to load settings from.'
13
+ argument :optional
14
+ }
15
+
16
+ option('broker', 'b'){
17
+ description 'The address of the AMQP broker.'
18
+ argument :optional
19
+ defaults ''
20
+ }
21
+
22
+ option('exchange', 'e'){
23
+ description 'The name of the exchange to subscribe bind to.'
24
+ argument :optional
25
+ defaults ''
26
+ }
27
+
28
+ option('routing_key', 'r'){
29
+ description 'The routing key to bind on. Matches all keys by default.'
30
+ argument :optional
31
+ defaults ''
32
+ }
33
+
34
+ option('filters', 'f'){
35
+ description 'A list of regex filters to pass the message stream through. If none are given, no messages will be filtered out.'
36
+ argument :optional
37
+ arity -1
38
+ }
39
+
40
+ option('throttle', 't'){
41
+ description 'Throttles the number of messages shown per second. Defaults to 1.'
42
+ argument :optional
43
+ cast :int
44
+ defaults 100
45
+ }
46
+
47
+ option('keys', 'k'){
48
+ description 'A list of keys in the message to display. If none are given, the raw JSON will be displayed.'
49
+ argument :optional
50
+ arity -1
51
+ }
52
+
53
+ # The main run loop of the application
54
+ def run
55
+ capture_signals
56
+ load_config
57
+
58
+ EM.synchrony do
59
+ connect
60
+ start_throttle
61
+ end
62
+ end
63
+
64
+ # Closes the connection to AMQP and stops the EM reactor
65
+ def stop
66
+
67
+ if EM.reactor_running?
68
+ EM::Synchrony.next_tick do
69
+ @amqp.close
70
+ EM.stop
71
+ end
72
+ end
73
+
74
+ end
75
+
76
+ # Captures interupt signals for this process and stops the node.
77
+ def capture_signals
78
+ %w[ TERM INT QUIT HUP ].each do |signal|
79
+ Signal.trap(signal) { stop }
80
+ end
81
+ end
82
+
83
+ # Loads a json config file
84
+ def load_config
85
+ if params[:config].given?
86
+ @config = File.open(File.expand_path(params[:config].value)) { |f| JSON.load(f) }
87
+
88
+ @config.each do |key, value|
89
+ if params.has_key?(key) and params[key].values == params[key].defaults
90
+ params[key].values = [*value]
91
+ params[key].given = true
92
+ end
93
+ end
94
+
95
+ end
96
+ end
97
+
98
+ # Connects to AMQP
99
+ def connect
100
+ @amqp = AMQP.connect("amqp://#{ params[:broker].value }")
101
+ @channel = AMQP::Channel.new(@amqp)
102
+
103
+ @queue = @channel.queue('', {
104
+ :auto_delete => true,
105
+ :exclusive => true,
106
+ :arguments => { 'x-message-ttl' => 30000 }
107
+ } )
108
+ @queue.bind(params[:exchange].value, :routing_key => params[:routing_key].value)
109
+ @queue.subscribe(&method(:process_message))
110
+ end
111
+
112
+ # Processes a new message
113
+ def process_message(headers, payload)
114
+ return if throttled?
115
+ return unless filtered?(payload)
116
+
117
+ @throttle += 1
118
+ display_message(payload)
119
+ end
120
+
121
+ # Displays a message to the console
122
+ def display_message(message)
123
+ if params[:keys].given?
124
+ output = ""
125
+
126
+ parsed = JSON.parse(message)
127
+ params[:keys].values.each do |key|
128
+ output << get_value(key, parsed).to_s
129
+ output << " "
130
+ end
131
+
132
+ puts output
133
+
134
+ else
135
+ puts message, ''
136
+ end
137
+ end
138
+
139
+ # Takes a key path (each key seperated by a space), and returns the corresponding value in the hash
140
+ def get_value(path, hash)
141
+ path.split(' ').each do |key|
142
+ hash = hash[key]
143
+ end
144
+ raise if hash.nil?
145
+
146
+ hash
147
+ rescue
148
+ "ValueNotFound"
149
+ end
150
+
151
+ # Returns true if the message passes all of the filters
152
+ def filtered?(message)
153
+ params[:filters].values.all? do |filter|
154
+ regex = Regexp.new(filter)
155
+ message.match(regex)
156
+ end
157
+ end
158
+
159
+ # Sets up the throttle reset
160
+ def start_throttle
161
+ @throttle = 0
162
+
163
+ EM::Synchrony.add_periodic_timer(1) do
164
+ @throttle = 0
165
+ end
166
+ end
167
+
168
+ # Checks if the maximum number of messages per second has been received
169
+ def throttled?
170
+ @throttle >= params[:throttle].value
171
+ end
172
+
173
+ }
@@ -0,0 +1,13 @@
1
+ {
2
+ "broker": "my-broker.company.com",
3
+ "exchange": "my-exchange",
4
+ "routing_key": "#",
5
+ "filters": [
6
+ "\"appname\":\"checkin\""
7
+ ],
8
+ "keys": [
9
+ "@source",
10
+ "@fields appname",
11
+ "@fields severity"
12
+ ]
13
+ }
data/mq-tail.gemspec ADDED
@@ -0,0 +1,76 @@
1
+ # Generated by jeweler
2
+ # DO NOT EDIT THIS FILE DIRECTLY
3
+ # Instead, edit Jeweler::Tasks in Rakefile, and run 'rake gemspec'
4
+ # -*- encoding: utf-8 -*-
5
+
6
+ Gem::Specification.new do |s|
7
+ s.name = "mq-tail"
8
+ s.version = "0.0.2"
9
+
10
+ s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
11
+ s.authors = ["Lucas Hansen"]
12
+ s.date = "2012-08-06"
13
+ s.description = "A tail like utility for watching a RabbitMQ queue. "
14
+ s.email = "lucas@likeness.com"
15
+ s.executables = ["mqtail"]
16
+ s.extra_rdoc_files = [
17
+ "LICENSE.txt",
18
+ "README.rdoc"
19
+ ]
20
+ s.files = [
21
+ ".document",
22
+ ".rspec",
23
+ "Gemfile",
24
+ "LICENSE.txt",
25
+ "README.rdoc",
26
+ "Rakefile",
27
+ "VERSION",
28
+ "bin/mqtail",
29
+ "config/mqtail.json",
30
+ "mq-tail.gemspec",
31
+ "spec/mq-tail_spec.rb",
32
+ "spec/spec_helper.rb"
33
+ ]
34
+ s.homepage = "http://github.com/Veered/mq-tail"
35
+ s.licenses = ["MIT"]
36
+ s.require_paths = ["lib"]
37
+ s.rubygems_version = "1.8.24"
38
+ s.summary = "A tail like utility for watching a RabbitMQ queue."
39
+
40
+ if s.respond_to? :specification_version then
41
+ s.specification_version = 3
42
+
43
+ if Gem::Version.new(Gem::VERSION) >= Gem::Version.new('1.2.0') then
44
+ s.add_runtime_dependency(%q<main>, ["~> 5.0.1"])
45
+ s.add_runtime_dependency(%q<amqp>, ["~> 0.9.7"])
46
+ s.add_runtime_dependency(%q<em-synchrony>, ["~> 1.0.2"])
47
+ s.add_runtime_dependency(%q<json>, ["~> 1.7.4"])
48
+ s.add_development_dependency(%q<rspec>, ["~> 2.8.0"])
49
+ s.add_development_dependency(%q<yard>, ["~> 0.7"])
50
+ s.add_development_dependency(%q<bundler>, ["~> 1.1.5"])
51
+ s.add_development_dependency(%q<jeweler>, ["~> 1.8.4"])
52
+ s.add_development_dependency(%q<pry>, ["~> 0.9.10"])
53
+ else
54
+ s.add_dependency(%q<main>, ["~> 5.0.1"])
55
+ s.add_dependency(%q<amqp>, ["~> 0.9.7"])
56
+ s.add_dependency(%q<em-synchrony>, ["~> 1.0.2"])
57
+ s.add_dependency(%q<json>, ["~> 1.7.4"])
58
+ s.add_dependency(%q<rspec>, ["~> 2.8.0"])
59
+ s.add_dependency(%q<yard>, ["~> 0.7"])
60
+ s.add_dependency(%q<bundler>, ["~> 1.1.5"])
61
+ s.add_dependency(%q<jeweler>, ["~> 1.8.4"])
62
+ s.add_dependency(%q<pry>, ["~> 0.9.10"])
63
+ end
64
+ else
65
+ s.add_dependency(%q<main>, ["~> 5.0.1"])
66
+ s.add_dependency(%q<amqp>, ["~> 0.9.7"])
67
+ s.add_dependency(%q<em-synchrony>, ["~> 1.0.2"])
68
+ s.add_dependency(%q<json>, ["~> 1.7.4"])
69
+ s.add_dependency(%q<rspec>, ["~> 2.8.0"])
70
+ s.add_dependency(%q<yard>, ["~> 0.7"])
71
+ s.add_dependency(%q<bundler>, ["~> 1.1.5"])
72
+ s.add_dependency(%q<jeweler>, ["~> 1.8.4"])
73
+ s.add_dependency(%q<pry>, ["~> 0.9.10"])
74
+ end
75
+ end
76
+
@@ -0,0 +1,5 @@
1
+ require File.expand_path(File.dirname(__FILE__) + '/spec_helper')
2
+
3
+ describe "MqTail" do
4
+
5
+ end
@@ -0,0 +1,12 @@
1
+ $LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
2
+ $LOAD_PATH.unshift(File.dirname(__FILE__))
3
+ require 'rspec'
4
+ require 'mq-tail'
5
+
6
+ # Requires supporting files with custom matchers and macros, etc,
7
+ # in ./support/ and its subdirectories.
8
+ Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each {|f| require f}
9
+
10
+ RSpec.configure do |config|
11
+
12
+ end
metadata ADDED
@@ -0,0 +1,205 @@
1
+ --- !ruby/object:Gem::Specification
2
+ name: mq-tail
3
+ version: !ruby/object:Gem::Version
4
+ version: 0.0.2
5
+ prerelease:
6
+ platform: ruby
7
+ authors:
8
+ - Lucas Hansen
9
+ autorequire:
10
+ bindir: bin
11
+ cert_chain: []
12
+ date: 2012-08-06 00:00:00.000000000 Z
13
+ dependencies:
14
+ - !ruby/object:Gem::Dependency
15
+ name: main
16
+ requirement: !ruby/object:Gem::Requirement
17
+ none: false
18
+ requirements:
19
+ - - ~>
20
+ - !ruby/object:Gem::Version
21
+ version: 5.0.1
22
+ type: :runtime
23
+ prerelease: false
24
+ version_requirements: !ruby/object:Gem::Requirement
25
+ none: false
26
+ requirements:
27
+ - - ~>
28
+ - !ruby/object:Gem::Version
29
+ version: 5.0.1
30
+ - !ruby/object:Gem::Dependency
31
+ name: amqp
32
+ requirement: !ruby/object:Gem::Requirement
33
+ none: false
34
+ requirements:
35
+ - - ~>
36
+ - !ruby/object:Gem::Version
37
+ version: 0.9.7
38
+ type: :runtime
39
+ prerelease: false
40
+ version_requirements: !ruby/object:Gem::Requirement
41
+ none: false
42
+ requirements:
43
+ - - ~>
44
+ - !ruby/object:Gem::Version
45
+ version: 0.9.7
46
+ - !ruby/object:Gem::Dependency
47
+ name: em-synchrony
48
+ requirement: !ruby/object:Gem::Requirement
49
+ none: false
50
+ requirements:
51
+ - - ~>
52
+ - !ruby/object:Gem::Version
53
+ version: 1.0.2
54
+ type: :runtime
55
+ prerelease: false
56
+ version_requirements: !ruby/object:Gem::Requirement
57
+ none: false
58
+ requirements:
59
+ - - ~>
60
+ - !ruby/object:Gem::Version
61
+ version: 1.0.2
62
+ - !ruby/object:Gem::Dependency
63
+ name: json
64
+ requirement: !ruby/object:Gem::Requirement
65
+ none: false
66
+ requirements:
67
+ - - ~>
68
+ - !ruby/object:Gem::Version
69
+ version: 1.7.4
70
+ type: :runtime
71
+ prerelease: false
72
+ version_requirements: !ruby/object:Gem::Requirement
73
+ none: false
74
+ requirements:
75
+ - - ~>
76
+ - !ruby/object:Gem::Version
77
+ version: 1.7.4
78
+ - !ruby/object:Gem::Dependency
79
+ name: rspec
80
+ requirement: !ruby/object:Gem::Requirement
81
+ none: false
82
+ requirements:
83
+ - - ~>
84
+ - !ruby/object:Gem::Version
85
+ version: 2.8.0
86
+ type: :development
87
+ prerelease: false
88
+ version_requirements: !ruby/object:Gem::Requirement
89
+ none: false
90
+ requirements:
91
+ - - ~>
92
+ - !ruby/object:Gem::Version
93
+ version: 2.8.0
94
+ - !ruby/object:Gem::Dependency
95
+ name: yard
96
+ requirement: !ruby/object:Gem::Requirement
97
+ none: false
98
+ requirements:
99
+ - - ~>
100
+ - !ruby/object:Gem::Version
101
+ version: '0.7'
102
+ type: :development
103
+ prerelease: false
104
+ version_requirements: !ruby/object:Gem::Requirement
105
+ none: false
106
+ requirements:
107
+ - - ~>
108
+ - !ruby/object:Gem::Version
109
+ version: '0.7'
110
+ - !ruby/object:Gem::Dependency
111
+ name: bundler
112
+ requirement: !ruby/object:Gem::Requirement
113
+ none: false
114
+ requirements:
115
+ - - ~>
116
+ - !ruby/object:Gem::Version
117
+ version: 1.1.5
118
+ type: :development
119
+ prerelease: false
120
+ version_requirements: !ruby/object:Gem::Requirement
121
+ none: false
122
+ requirements:
123
+ - - ~>
124
+ - !ruby/object:Gem::Version
125
+ version: 1.1.5
126
+ - !ruby/object:Gem::Dependency
127
+ name: jeweler
128
+ requirement: !ruby/object:Gem::Requirement
129
+ none: false
130
+ requirements:
131
+ - - ~>
132
+ - !ruby/object:Gem::Version
133
+ version: 1.8.4
134
+ type: :development
135
+ prerelease: false
136
+ version_requirements: !ruby/object:Gem::Requirement
137
+ none: false
138
+ requirements:
139
+ - - ~>
140
+ - !ruby/object:Gem::Version
141
+ version: 1.8.4
142
+ - !ruby/object:Gem::Dependency
143
+ name: pry
144
+ requirement: !ruby/object:Gem::Requirement
145
+ none: false
146
+ requirements:
147
+ - - ~>
148
+ - !ruby/object:Gem::Version
149
+ version: 0.9.10
150
+ type: :development
151
+ prerelease: false
152
+ version_requirements: !ruby/object:Gem::Requirement
153
+ none: false
154
+ requirements:
155
+ - - ~>
156
+ - !ruby/object:Gem::Version
157
+ version: 0.9.10
158
+ description: ! 'A tail like utility for watching a RabbitMQ queue. '
159
+ email: lucas@likeness.com
160
+ executables:
161
+ - mqtail
162
+ extensions: []
163
+ extra_rdoc_files:
164
+ - LICENSE.txt
165
+ - README.rdoc
166
+ files:
167
+ - .document
168
+ - .rspec
169
+ - Gemfile
170
+ - LICENSE.txt
171
+ - README.rdoc
172
+ - Rakefile
173
+ - VERSION
174
+ - bin/mqtail
175
+ - config/mqtail.json
176
+ - mq-tail.gemspec
177
+ - spec/mq-tail_spec.rb
178
+ - spec/spec_helper.rb
179
+ homepage: http://github.com/Veered/mq-tail
180
+ licenses:
181
+ - MIT
182
+ post_install_message:
183
+ rdoc_options: []
184
+ require_paths:
185
+ - lib
186
+ required_ruby_version: !ruby/object:Gem::Requirement
187
+ none: false
188
+ requirements:
189
+ - - ! '>='
190
+ - !ruby/object:Gem::Version
191
+ version: '0'
192
+ required_rubygems_version: !ruby/object:Gem::Requirement
193
+ none: false
194
+ requirements:
195
+ - - ! '>='
196
+ - !ruby/object:Gem::Version
197
+ version: '0'
198
+ requirements: []
199
+ rubyforge_project:
200
+ rubygems_version: 1.8.24
201
+ signing_key:
202
+ specification_version: 3
203
+ summary: A tail like utility for watching a RabbitMQ queue.
204
+ test_files: []
205
+ has_rdoc: