riemann-aws-dynamodb 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8864f136ccfb703586bd4663d78cc17b116e46e5
-  data.tar.gz: f337738b618e4795fe12d4b7621471a58efe415d
+  metadata.gz: 34cd73b6d0d8f8ede15b4439662dbda0030b59bf
+  data.tar.gz: e29e08b94c46c71d10f8804d51f0e4a9357dc0d1
 SHA512:
-  metadata.gz: a5a937dbb16bbbe8f5f4a62184f64200a6a3243b9bee68fd83553cb025504e226b02e1597e06b03024c08a82279d5415bc3406ffff2a2f3e9ac32a4fe181f675
-  data.tar.gz: 97f4bdf4e8de5a22bb7e454962fd0d748e4b539b104354301d6e151858ae4d69c8b6eaa4c835b85dbd450fa09f05079318cdd1d243632dc69ed7dc1e8b2ae4d4
+  metadata.gz: f3956d8962ebc5224cd1ef1bb325a2316acb148b1d916dfa3e15d0d062674667c4ca50b10a57629bb3a86508f844004bd35417ac4324e6b6753e948a395c8387
+  data.tar.gz: 7d31d5d48db74b874e07fe05f6b9ec236f8c067a293c73ee8589303cdc6716c4abbdbf008741c365ab234d68fe33cb39d775c2e4ef582b85f404935bd58ab144

data/bin/riemann-aws-dynamodb

@@ -12,20 +12,24 @@ class Riemann::Tools::AwsDynamoDB
   opt :access_key, "AWS access key", :type => String
   opt :secret_key, "Secret access key", :type => String
   opt :region, "AWS region", :type => String, :default => 'us-east-1'
-  opt :tables, "Table names", :type => String, :multi => true
+  opt :only_table, "Only monitor this table. Can specify multiple times.", :type => String, :multi => true
+  opt :ign_table, "Don't monitor this table. Can specify multiple times.", :type => String, :multi => true
   # opt :metric, "Metric names with comma-separated warning/critical thresholds (eg., 'CPUUtilization,50,90'). Multiple tuples can be set.", :type => String, :multi => true
 
   def initialize
-    abort "FATAL: specify table names, see --help for usage" unless opts[:tables_given]
-    @tables = opts.fetch(:tables)
+    @only_tables = opts.fetch(:only_table)
+    @ign_tables = opts.fetch(:ign_table)
+
     if opts.has_key?('access_key') and opts.has_key?('secret_key')
       creds = { :aws_access_key_id     => opts[:access_key],
                 :aws_secret_access_key => opts[:secret_key] }
     else
       creds = { :use_iam_profile => true }
     end
+
     creds['region'] = opts[:region]
     @cloudwatch = Fog::AWS::CloudWatch.new(creds)
+    @dynamodb = Fog::AWS::DynamoDB.new(creds)
   end
 
   def metrics_opts
@@ -39,6 +43,15 @@ class Riemann::Tools::AwsDynamoDB
 
   def tick
     time = Time.new
+    ddb = @dynamodb.list_tables
+    @tables = ddb[:body]["TableNames"]
+
+    if opts[:only_table_given]
+      @tables &= @only_tables
+    elsif opts[:ign_table_given]
+      @tables -= @ign_tables
+    end
+
     @tables.each do |table_name|
       metrics_opts.each_pair do |metric_name, opts|
         result = @cloudwatch.get_metric_statistics({

data/riemann-aws-dynamodb.gemspec

@@ -4,7 +4,7 @@ $LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
 
 Gem::Specification.new do |spec|
   spec.name          = "riemann-aws-dynamodb"
-  spec.version       = '0.0.2'
+  spec.version       = '0.0.4'
   spec.authors       = ["Fernando Alonso"]
   spec.email         = ["krakatoa1987@gmail.com"]
   spec.summary       = %q{Riemann agent to collect Aws DynamoDB metrics}

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: riemann-aws-dynamodb
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
 platform: ruby
 authors:
 - Fernando Alonso
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-01-19 00:00:00.000000000 Z
+date: 2016-01-20 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: riemann-tools