mongodb-graphite-agent 0.1.1 → 0.1.2

data/.gitignore

@@ -1,20 +1,20 @@
-*.gem
-*.rbc
-.bundle
-.config
-.yardoc
-Gemfile.lock
-InstalledFiles
-_yardoc
-coverage
-doc/
-lib/bundler/man
-pkg
-rdoc
-spec/reports
-test/tmp
-test/version_tmp
-tmp
-.idea/*
-lastsample
-*.iml
+*.gem
+*.rbc
+.bundle
+.config
+.yardoc
+Gemfile.lock
+InstalledFiles
+_yardoc
+coverage
+doc/
+lib/bundler/man
+pkg
+rdoc
+spec/reports
+test/tmp
+test/version_tmp
+tmp
+.idea/*
+lastsample
+*.iml

data/.travis.yml

@@ -0,0 +1,7 @@
+language: ruby
+rvm:
+  - 1.9.3
+  - rbx-19mode
+
+services:
+  - mongodb

data/Gemfile

@@ -1,3 +1,3 @@
-source 'https://rubygems.org'
-
+source 'https://rubygems.org'
+
 gemspec

data/README.md

@@ -1,29 +1,21 @@
-# Mongodb::Graphite::Agent
-
-TODO: Write a gem description
-
-## Installation
-
-Add this line to your application's Gemfile:
-
-    gem 'mongodb-graphite-agent'
-
-And then execute:
-
-    $ bundle
-
-Or install it yourself as:
-
-    $ gem install mongodb-graphite-agent
-
-## Usage
-
-TODO: Write usage instructions here
-
-## Contributing
-
-1. Fork it
-2. Create your feature branch (`git checkout -b my-new-feature`)
-3. Commit your changes (`git commit -am 'Add some feature'`)
-4. Push to the branch (`git push origin my-new-feature`)
-5. Create new Pull Request
+# Mongodb::Graphite::Agent
+
+[![Build Status](https://travis-ci.org/emmekappa/mongodb-graphite-agent.png)](https://travis-ci.org/emmekappa/mongodb-graphite-agent)
+
+Sends MongoDB metrics to Graphite.
+
+## Installation
+
+    $ gem install mongodb-graphite-agent
+
+## Usage
+
+    mongodb-graphite-agent.rb --mongodb-host localhost --graphite-host graphite.local --graphite-port 9090 --graphite-metrics-prefix "localhost.mongodb" --verbose
+
+## Contributing
+
+1. Fork it
+2. Create your feature branch (`git checkout -b my-new-feature`)
+3. Commit your changes (`git commit -am 'Add some feature'`)
+4. Push to the branch (`git push origin my-new-feature`)
+5. Create new Pull Request

data/Rakefile

@@ -1 +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
+task :default => :spec

data/bin/mongodb-graphite-agent.rb

@@ -1,28 +1,29 @@
-#!/usr/bin/env ruby
-
-$:.unshift File.expand_path("#{File.dirname(__FILE__)}/../lib")
-
-require 'trollop'
-require 'mongodb/graphite/agent'
-
-opts = Trollop::options do
-  opt :mongodb_username, "MongoDB username", :type => :string
-  opt :mongodb_host, "MongoDB host", :type => :string, :default => "localhost"
-  opt :mongodb_post, "MongoDB port", :type => :int, :default => 27017
-  opt :mongodb_password, "MongoDB password", :type => :string
-  opt :graphite_host, "Graphite host", :type => :string
-  opt :graphite_port, "Graphite port", :type => :string
-  opt :graphite_metrics_prefix, "Graphite metrics prefix", :type => :string, :default => Socket.gethostname
-  opt :dry_run, "Dry run", :type => :boolean, :default => false
-  opt :verbose, "Verbose", :type => :boolean, :default => false
-end
-
-if opts[:dry_run]
-  puts "\n\nWARNING!!! This is a dry run\n\n\n"
-  sleep 1
-end
-
-Trollop::die :graphite_host, "(or --dry-run) must be specified" if (opts[:graphite_host].blank? and opts[:dry_run].blank?)
-
-runner = Mongodb::Graphite::Agent::Runner.new(opts)
+#!/usr/bin/env ruby
+
+$:.unshift File.expand_path("#{File.dirname(__FILE__)}/../lib")
+
+require 'trollop'
+require 'mongodb/graphite/agent'
+require 'mongodb/graphite/agent/graphite_writer'
+
+opts = Trollop::options do
+  opt :mongodb_username, "MongoDB username", :type => :string
+  opt :mongodb_host, "MongoDB host", :type => :string, :default => "localhost"
+  opt :mongodb_post, "MongoDB port", :type => :int, :default => 27017
+  opt :mongodb_password, "MongoDB password", :type => :string
+  opt :graphite_host, "Graphite host", :type => :string
+  opt :graphite_port, "Graphite port", :type => :string
+  opt :graphite_metrics_prefix, "Graphite metrics prefix", :type => :string, :default => Socket.gethostname
+  opt :dry_run, "Dry run", :type => :boolean, :default => false
+  opt :verbose, "Verbose", :type => :boolean, :default => false
+end
+
+if opts[:dry_run]
+  puts "\n\nWARNING!!! This is a dry run\n\n\n"
+  sleep 1
+end
+
+Trollop::die :graphite_host, "(or --dry-run) must be specified" if (opts[:graphite_host].blank? and opts[:dry_run].blank?)
+
+runner = Mongodb::Graphite::Agent::Runner.new(opts)
 runner.run

data/lib/mongodb/graphite/agent.rb

@@ -1,82 +1,82 @@
-require 'rubygems'
-require 'bundler/setup'
-require 'mongo'
-require 'simple-graphite'
-require 'bson'
-require 'socket'
-require 'awesome_print'
-require 'time_difference'
-require 'mongodb/graphite/agent/utils'
-require 'mongodb/graphite/agent/op_counters_sample'
-require 'mongodb/graphite/agent/mongo_cient_extensions'
-
-module Mongodb
-  module Graphite
-    module Agent
-      class Runner
-
-        def initialize(opts)
-          @opts = opts
-        end
-
-        def run
-          connection = Mongo::MongoClient.new(@opts.mongodb_host, @opts.mongodb_port, :slave_ok => true)
-          unless (@opts[:mongodb_username].blank? && @opts[:mongodb_password].blank?)
-            connection["admin"].authenticate(@opts.mongodb_username, @opts.mongodb_password)
-          end
-
-          server_status_result = connection["local"].command('serverStatus' => 1)
-          metric_hash = Utils.to_hash(server_status_result).select { |k|
-            k.match('^connection|^network\.|^cursors|^mem\.mapped|^indexCounters|^repl.oplog')
-          }
-
-          opcounters_per_second_metric_hash = calculate_opcounters_per_second server_status_result["opcounters"]
-
-          if @opts[:verbose]
-            puts "Calculating metrics..."
-            ap metric_hash
-            ap opcounters_per_second_metric_hash
-          end
-
-
-          unless (@opts[:dry_run])
-            graphite_writer = GraphiteWriter.new({:host => @opts[:graphite_host],
-                                                  :port => @opts[:graphite_port],
-                                                  :verbose => @opts[:verbose],
-                                                  :metrics_prefix => @opts[:graphite_metrics_prefix]})
-            graphite_writer.write(metric_hash)
-            graphite_writer.write(opcounters_per_second_metric_hash)
-          end
-        end
-
-        def calculate_opcounters_per_second(opcounters)
-          current_sample = OpCountersSample.new Hash[opcounters]
-          previous_sample = current_sample.dup
-          result = {}
-
-          if File.exist? 'lastsample'
-            File.open('lastsample', 'r') do |file|
-              previous_sample = Marshal.load(file)
-            end
-          end
-
-          delta = TimeDifference.between(Time.parse(current_sample.sample_time), Time.parse(previous_sample.sample_time))
-          puts "Last sample was taken #{delta.in_seconds.round(0)} seconds ago"
-
-          previous_sample.values.keys.sort.each do |k|
-            previous_sample_value = previous_sample.values[k]
-            current_sample_value = current_sample.values[k]
-            value_per_seconds = ((current_sample_value - previous_sample_value) / delta.in_seconds).round(2)
-            result["#{k}_per_seconds"] = value_per_seconds
-          end
-
-          File.open('lastsample', 'w') do |file|
-            Marshal.dump(current_sample, file)
-          end
-
-          result
-        end
-      end
-    end
-  end
+require 'rubygems'
+require 'bundler/setup'
+require 'mongo'
+require 'simple-graphite'
+require 'bson'
+require 'socket'
+require 'awesome_print'
+require 'time_difference'
+require 'mongodb/graphite/agent/utils'
+require 'mongodb/graphite/agent/op_counters_sample'
+require 'mongodb/graphite/agent/mongo_cient_extensions'
+
+module Mongodb
+  module Graphite
+    module Agent
+      class Runner
+
+        def initialize(opts)
+          @opts = opts
+        end
+
+        def run
+          connection = Mongo::MongoClient.new(@opts[:mongodb_host], @opts[:mongodb_port], :slave_ok => true)
+          unless (@opts[:mongodb_username].blank? && @opts[:mongodb_password].blank?)
+            connection["admin"].authenticate(@opts.mongodb_username, @opts.mongodb_password)
+          end
+
+          server_status_result = connection["local"].command('serverStatus' => 1)
+          metric_hash = Utils.to_hash(server_status_result).select { |k|
+            k.match('^connection|^network\.|^cursors|^mem\.mapped|^indexCounters|^repl.oplog')
+          }
+
+          opcounters_per_second_metric_hash = calculate_opcounters_per_second server_status_result["opcounters"]
+
+          if @opts[:verbose]
+            puts "Calculating metrics..."
+            ap metric_hash
+            ap opcounters_per_second_metric_hash
+          end
+
+
+          unless (@opts[:dry_run])
+            graphite_writer = GraphiteWriter.new({:host => @opts[:graphite_host],
+                                                  :port => @opts[:graphite_port],
+                                                  :verbose => @opts[:verbose],
+                                                  :metrics_prefix => @opts[:graphite_metrics_prefix]})
+            graphite_writer.write(metric_hash)
+            graphite_writer.write(opcounters_per_second_metric_hash)
+          end
+        end
+
+        def calculate_opcounters_per_second(opcounters)
+          current_sample = OpCountersSample.new Hash[opcounters]
+          previous_sample = current_sample.dup
+          result = {}
+
+          if File.exist? 'lastsample'
+            File.open('lastsample', 'r') do |file|
+              previous_sample = Marshal.load(file)
+            end
+          end
+
+          delta = TimeDifference.between(Time.parse(current_sample.sample_time), Time.parse(previous_sample.sample_time))
+          puts "Last sample was taken #{delta.in_seconds.round(0)} seconds ago"
+
+          previous_sample.values.keys.sort.each do |k|
+            previous_sample_value = previous_sample.values[k]
+            current_sample_value = current_sample.values[k]
+            value_per_seconds = ((current_sample_value - previous_sample_value) / delta.in_seconds).round(2)
+            result["#{k}_per_seconds"] = value_per_seconds
+          end
+
+          File.open('lastsample', 'w') do |file|
+            Marshal.dump(current_sample, file)
+          end
+
+          result
+        end
+      end
+    end
+  end
 end

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

@@ -1,21 +1,21 @@
-module Mongodb
-  module Graphite
-    module Agent
-      class GraphiteWriter
-        def initialize(opts)
-          @graphite = ::Graphite.new({:host => opts[:host], :port => opts[:port]})
-          @opts = opts
-        end
-
-        def write(metric_hash)
-          @metric_hash_with_hostname = Hash[metric_hash.map { |k,v| ["#{@opts[:metrics_prefix]}.#{k}", v]}]
-          if @opts[:verbose]
-            puts "Sending data to graphite..."
-            ap @metric_hash_with_hostname
-          end
-          @graphite.send_metrics @metric_hash_with_hostname
-        end
-      end
-    end
-  end
-end
+module Mongodb
+  module Graphite
+    module Agent
+      class GraphiteWriter
+        def initialize(opts)
+          @graphite = ::Graphite.new({:host => opts[:host], :port => opts[:port]})
+          @opts = opts
+        end
+
+        def write(metric_hash)
+          @metric_hash_with_hostname = Hash[metric_hash.map { |k,v| ["#{@opts[:metrics_prefix]}.#{k}", v]}]
+          if @opts[:verbose]
+            puts "Sending data to graphite..."
+            ap @metric_hash_with_hostname
+          end
+          @graphite.send_metrics @metric_hash_with_hostname
+        end
+      end
+    end
+  end
+end

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

@@ -1,15 +1,15 @@
-module Mongo
-
-  class MongoClient
-    def is_replicaset?
-      begin
-        cmd = BSON::OrderedHash.new
-        cmd["replSetGetStatus"] = 1
-        result = self.db("admin").command(cmd)
-        return !result["set"].blank?
-      rescue Mongo::OperationFailure
-        return false
-      end
-    end
-  end
-end
+module Mongo
+
+  class MongoClient
+    def is_replicaset?
+      begin
+        cmd = BSON::OrderedHash.new
+        cmd["replSetGetStatus"] = 1
+        result = self.db("admin").command(cmd)
+        return !result["set"].blank?
+      rescue Mongo::OperationFailure
+        return false
+      end
+    end
+  end
+end

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

@@ -1,26 +1,26 @@
-require 'date'
-
-module Mongodb
-  module Graphite
-    module Agent
-
-      class OpCountersSample
-        attr_reader :values, :sample_time
-
-        def initialize(values, sample_time = DateTime.now.to_s)
-          @values = values
-          @sample_time = sample_time
-        end
-
-        def marshal_dump
-          [@sample_time, @values]
-        end
-
-        def marshal_load array
-          @sample_time, @values = array
-        end
-
-      end
-    end
-  end
+require 'date'
+
+module Mongodb
+  module Graphite
+    module Agent
+
+      class OpCountersSample
+        attr_reader :values, :sample_time
+
+        def initialize(values, sample_time = DateTime.now.to_s)
+          @values = values
+          @sample_time = sample_time
+        end
+
+        def marshal_dump
+          [@sample_time, @values]
+        end
+
+        def marshal_load array
+          @sample_time, @values = array
+        end
+
+      end
+    end
+  end
 end

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

@@ -1,29 +1,29 @@
-module Mongodb
-  module Graphite
-    module Agent
-
-      module Utils
-        def self.merge_all
-          self.inject({}) { |h1, h2|
-            h1.merge! h2
-          }
-        end
-
-        def self.json_descent(pre, json)
-          json.map do |k, v|
-            key = pre + [k]
-            if v.is_a? BSON::OrderedHash
-              json_descent(key, v)
-            else
-              {key.join('.') => v}
-            end
-          end
-        end
-
-        def self.to_hash(s)
-          json_descent([], s).flatten.reduce Hash.new, :merge
-        end
-      end
-    end
-  end
-end
+module Mongodb
+  module Graphite
+    module Agent
+
+      module Utils
+        def self.merge_all
+          self.inject({}) { |h1, h2|
+            h1.merge! h2
+          }
+        end
+
+        def self.json_descent(pre, json)
+          json.map do |k, v|
+            key = pre + [k]
+            if v.is_a? BSON::OrderedHash
+              json_descent(key, v)
+            else
+              {key.join('.') => v}
+            end
+          end
+        end
+
+        def self.to_hash(s)
+          json_descent([], s).flatten.reduce Hash.new, :merge
+        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.1"
-    end
-  end
-end
+module Mongodb
+  module Graphite
+    module Agent
+        VERSION = "0.1.2"
+    end
+  end
+end

data/mongodb-graphite-agent.gemspec

@@ -1,31 +1,32 @@
-# coding: utf-8
-lib = File.expand_path('../lib', __FILE__)
-$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
-require 'mongodb/graphite/agent/version'
-
-Gem::Specification.new do |spec|
-  spec.name          = "mongodb-graphite-agent"
-  spec.version       = Mongodb::Graphite::Agent::VERSION
-  spec.authors       = ["Michele Cantelli"]
-  spec.email         = ["michele.cantelli@jobrapido.com"]
-  spec.description   = ""
-  spec.summary       = ""
-  spec.homepage      = "https://github.com/emmekappa/mongodb-graphite-agent"
-  spec.license       = "MIT"
-
-  spec.files         = `git ls-files`.split($/)
-  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
-  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
-  spec.require_paths = ["lib"]
-
-  spec.add_development_dependency "bundler", "~> 1.3"
-  spec.add_development_dependency "rake"
-
-  spec.add_dependency 'mongo'
-  spec.add_dependency 'simple-graphite'
-  spec.add_dependency 'bson'
-  spec.add_dependency 'bson_ext'
-  spec.add_dependency 'awesome_print'
-  spec.add_dependency 'time_difference'
-  spec.add_dependency 'trollop'
-end
+# coding: utf-8
+lib = File.expand_path('../lib', __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require 'mongodb/graphite/agent/version'
+
+Gem::Specification.new do |spec|
+  spec.name          = "mongodb-graphite-agent"
+  spec.version       = Mongodb::Graphite::Agent::VERSION
+  spec.authors       = ["Michele Cantelli"]
+  spec.email         = ["michele.cantelli@jobrapido.com"]
+  spec.description   = ""
+  spec.summary       = ""
+  spec.homepage      = "https://github.com/emmekappa/mongodb-graphite-agent"
+  spec.license       = "MIT"
+
+  spec.files         = `git ls-files`.split($/)
+  spec.executables   = spec.files.grep(%r{^bin/}) { |f| File.basename(f) }
+  spec.test_files    = spec.files.grep(%r{^(test|spec|features)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.3"
+  spec.add_development_dependency "rake"
+  spec.add_development_dependency "rspec"
+
+  spec.add_dependency 'mongo'
+  spec.add_dependency 'simple-graphite'
+  spec.add_dependency 'bson'
+  spec.add_dependency 'bson_ext'
+  spec.add_dependency 'awesome_print'
+  spec.add_dependency 'time_difference'
+  spec.add_dependency 'trollop'
+end

data/spec/agent_spec.rb

@@ -0,0 +1,13 @@
+require 'rspec'
+require 'mongodb/graphite/agent'
+
+describe 'MongoDB Server Status' do
+
+  it 'should read server status' do
+
+    #To change this template use File | Settings | File Templates.
+    runner = Mongodb::Graphite::Agent::Runner.new({ :mongodb_host => 'localhost', :mongodb_port => '27017', :dry_run => true, :verbose => true})
+    runner.run
+
+  end
+end

data/spec/op_counters_sample_spec.rb

@@ -0,0 +1,17 @@
+require 'rspec'
+require 'mongodb/graphite/agent/op_counters_sample'
+
+describe 'OpCounters' do
+  it 'should reload last sample' do
+    current_sample = Mongodb::Graphite::Agent::OpCountersSample.new(100)
+
+    File.open('lastsample-test', 'w') do |file|
+      Marshal.dump(current_sample, file)
+    end
+
+    File.open('lastsample-test', 'r') do |file|
+      previous_sample = Marshal.load(file)
+      previous_sample.values.should eq(100)
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mongodb-graphite-agent
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-07-14 00:00:00.000000000 Z
+date: 2013-07-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -43,6 +43,22 @@ dependencies:
     - - ! '>='
       - !ruby/object:Gem::Version
         version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: mongo
   requirement: !ruby/object:Gem::Requirement
@@ -164,6 +180,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - .gitignore
+- .travis.yml
 - Gemfile
 - LICENSE.txt
 - README.md
@@ -176,6 +193,8 @@ files:
 - lib/mongodb/graphite/agent/utils.rb
 - lib/mongodb/graphite/agent/version.rb
 - mongodb-graphite-agent.gemspec
+- spec/agent_spec.rb
+- spec/op_counters_sample_spec.rb
 homepage: https://github.com/emmekappa/mongodb-graphite-agent
 licenses:
 - MIT
@@ -189,22 +208,18 @@ required_ruby_version: !ruby/object:Gem::Requirement
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
-      segments:
-      - 0
-      hash: 2967853814840194051
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
   - - ! '>='
     - !ruby/object:Gem::Version
       version: '0'
-      segments:
-      - 0
-      hash: 2967853814840194051
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.25
+rubygems_version: 1.8.24
 signing_key: 
 specification_version: 3
 summary: ''
-test_files: []
+test_files:
+- spec/agent_spec.rb
+- spec/op_counters_sample_spec.rb