mongodb-graphite-agent 0.1.2 → 0.1.3

data/.gitignore

@@ -16,5 +16,5 @@ test/tmp
 test/version_tmp
 tmp
 .idea/*
-lastsample
+lastsample*
 *.iml

data/LICENSE.txt

@@ -1,22 +1,22 @@
-Copyright (c) 2013 Michele Cantelli
-
-MIT License
-
-Permission is hereby granted, free of charge, to any person obtaining
-a copy of this software and associated documentation files (the
-"Software"), to deal in the Software without restriction, including
-without limitation the rights to use, copy, modify, merge, publish,
-distribute, sublicense, and/or sell copies of the Software, and to
-permit persons to whom the Software is furnished to do so, subject to
-the following conditions:
-
-The above copyright notice and this permission notice shall be
-included in all copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
-EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
-MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
-NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
-LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
-OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
-WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+Copyright (c) 2013 Michele Cantelli
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/Rakefile

@@ -1,7 +1,7 @@
-require "bundler/gem_tasks"
-require 'rspec/core/rake_task'
-
-RSpec::Core::RakeTask.new('spec')
-
-# If you want to make this the default task
+require "bundler/gem_tasks"
+require 'rspec/core/rake_task'
+
+RSpec::Core::RakeTask.new('spec')
+
+# If you want to make this the default task
 task :default => :spec

data/lib/mongodb/graphite/agent.rb

@@ -9,6 +9,7 @@ require 'time_difference'
 require 'mongodb/graphite/agent/utils'
 require 'mongodb/graphite/agent/op_counters_sample'
 require 'mongodb/graphite/agent/mongo_cient_extensions'
+require 'mongodb/graphite/agent/collection_size_calculator'
 
 module Mongodb
   module Graphite
@@ -31,11 +32,13 @@ module Mongodb
           }
 
           opcounters_per_second_metric_hash = calculate_opcounters_per_second server_status_result["opcounters"]
+          total_collections_hash = CollectionSizeCalculator.new(connection).calculate
 
           if @opts[:verbose]
             puts "Calculating metrics..."
             ap metric_hash
             ap opcounters_per_second_metric_hash
+            ap total_collections_hash
           end
 
 
@@ -46,6 +49,7 @@ module Mongodb
                                                   :metrics_prefix => @opts[:graphite_metrics_prefix]})
             graphite_writer.write(metric_hash)
             graphite_writer.write(opcounters_per_second_metric_hash)
+            graphite_writer.write(total_collections_hash)
           end
         end
 

data/lib/mongodb/graphite/agent/collection_size_calculator.rb

@@ -0,0 +1,28 @@
+require 'awesome_print'
+
+module Mongodb
+  module Graphite
+    module Agent
+      class CollectionSizeCalculator
+
+        def initialize(mongo_client)
+          @mongo_client = mongo_client
+        end
+
+        def calculate
+          collection_count_hash = {}
+          @mongo_client.database_names.each { |db_name|
+            @mongo_client[db_name].collection_names.each { |collection_name|
+
+
+              collection_count_hash["collection_sizes.#{db_name}.#{collection_name}"] = @mongo_client[db_name][collection_name].stats()["count"]
+            } unless db_name == 'local'
+          }
+
+          return collection_count_hash
+        end
+
+      end
+    end
+  end
+end

data/lib/mongodb/graphite/agent/version.rb

@@ -1,7 +1,7 @@
 module Mongodb
   module Graphite
     module Agent
-        VERSION = "0.1.2"
+        VERSION = "0.1.3"
     end
   end
 end

data/spec/collection_size_calculator_spec.rb

@@ -0,0 +1,10 @@
+require 'rspec'
+require 'mongodb/graphite/agent/collection_size_calculator'
+require 'mongo'
+
+describe 'Collection size calculator' do
+  it 'should calculate the number of documents in all db collections' do
+    collection_size_calculator = Mongodb::Graphite::Agent::CollectionSizeCalculator.new(Mongo::MongoClient.new())
+    collection_size_calculator.calculate.should have_at_least(1).items
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongodb-graphite-agent
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-07-15 00:00:00.000000000 Z
+date: 2013-07-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -187,6 +187,7 @@ files:
 - Rakefile
 - bin/mongodb-graphite-agent.rb
 - lib/mongodb/graphite/agent.rb
+- lib/mongodb/graphite/agent/collection_size_calculator.rb
 - lib/mongodb/graphite/agent/graphite_writer.rb
 - lib/mongodb/graphite/agent/mongo_cient_extensions.rb
 - lib/mongodb/graphite/agent/op_counters_sample.rb
@@ -194,6 +195,7 @@ files:
 - lib/mongodb/graphite/agent/version.rb
 - mongodb-graphite-agent.gemspec
 - spec/agent_spec.rb
+- spec/collection_size_calculator_spec.rb
 - spec/op_counters_sample_spec.rb
 homepage: https://github.com/emmekappa/mongodb-graphite-agent
 licenses:
@@ -222,4 +224,5 @@ specification_version: 3
 summary: ''
 test_files:
 - spec/agent_spec.rb
+- spec/collection_size_calculator_spec.rb
 - spec/op_counters_sample_spec.rb