openldap_monitor_extractor 0.0.3 → 0.0.4

data/.gitignore

@@ -8,7 +8,11 @@ Gemfile.lock
 InstalledFiles
 _yardoc
 coverage
-examples/production.yml
+examples/master.yml
+examples/slave1.yml
+examples/slave2.yml
+examples/slave3.yml
+examples/slave4.yml
 doc/
 pkg
 rdoc

data/examples/{get_totalconnections.rb → get_all_stats.rb}

@@ -8,11 +8,12 @@ require 'yaml'
 require 'openldap_monitor_extractor'
 
 
-if ARGV[0] && File.exist?(ARGV[0])
-  
-  config = YAML.load_file(ARGV[0])
-  puts OpenldapMonitorExtractor.configure(config).get(:total_connections)
-else
-  
+unless ARGV[0] && File.exist?(ARGV[0])
+
   puts "usage: #{$0} config_file.yml"
-end
+  Kernel.exit 1
+end
+
+  
+config = YAML.load_file(ARGV[0])
+puts OpenldapMonitorExtractor.configure(config).get(:all).inspect

data/lib/openldap_monitor_extractor.rb

@@ -1,57 +1,82 @@
+require "ostruct"
 require "openldap_monitor_extractor/version"
 require "openldap_monitor_extractor/connection"
 
 
 module OpenldapMonitorExtractor
   extend self
+  
+  class ExtractorError < StandardError
+  end 
 
   MAPPER = {
-    :total_connections    =>{ :dn =>"cn=Total,cn=Connections,cn=Monitor",   :attribute =>:monitorcounter      },
-    :referals_statistics  =>{ :dn =>"cn=Referrals,cn=Statistics,cn=Monitor",:attribute =>:monitorcounter      },
-    :bytes_statistics     =>{ :dn =>"cn=Bytes,cn=Statistics,cn=Monitor",    :attribute =>:monitorcounter      },
-    :entries_statistics   =>{ :dn =>"cn=Entries,cn=Statistics,cn=Monitor",  :attribute =>:monitorcounter      },
-    :initiated_operations =>{ :dn =>"cn=Operations,cn=Monitor",             :attribute =>:monitoropinitiated  },
-    :completed_operations =>{ :dn =>"cn=Operations,cn=Monitor",             :attribute =>:monitoropcompleted  },
-    :bind_operations      =>{ :dn =>"cn=Bind,cn=Operations,cn=Monitor",     :attribute =>:monitoropcompleted  },
-    :unbind_operations    =>{ :dn =>"cn=Unbind,cn=Operations,cn=Monitor",   :attribute =>:monitoropcompleted  },
-    :search_operations    =>{ :dn =>"cn=Search,cn=Operations,cn=Monitor",   :attribute =>:monitoropcompleted  },
-    :compare_operations   =>{ :dn =>"cn=Compare,cn=Operations,cn=Monitor",  :attribute =>:monitoropcompleted  },
-    :modify_operations    =>{ :dn =>"cn=Modify,cn=Operations,cn=Monitor",   :attribute =>:monitoropcompleted  },
-    :modrdn_operations    =>{ :dn =>"cn=Modrdn,cn=Operations,cn=Monitor",   :attribute =>:monitoropcompleted  },
-    :add_operations       =>{ :dn =>"cn=Add,cn=Operations,cn=Monitor",      :attribute =>:monitoropcompleted  },
-    :delete_operations    =>{ :dn =>"cn=Delete,cn=Operations,cn=Monitor",   :attribute =>:monitoropcompleted  },
-    :abandon_operations   =>{ :dn =>"cn=Abandon,cn=Operations,cn=Monitor",  :attribute =>:monitoropcompleted  },
-    :extended_operations  =>{ :dn =>"cn=Extended,cn=Operations,cn=Monitor", :attribute =>:monitoropcompleted  },
-    :write_waiters        =>{ :dn =>"cn=Write,cn=Waiters,cn=Monitor",       :attribute =>:monitorcounter      },
-    :read_waiters         =>{ :dn =>"cn=Read,cn=Waiters,cn=Monitor",        :attribute =>:monitorcounter      }
+    :total_connections    =>{ :dn =>"cn=Total,cn=Connections,cn=Monitor",   :attribute =>:monitorcounter     },
+    :referals_statistics  =>{ :dn =>"cn=Referrals,cn=Statistics,cn=Monitor",:attribute =>:monitorcounter     },
+    :bytes_statistics     =>{ :dn =>"cn=Bytes,cn=Statistics,cn=Monitor",    :attribute =>:monitorcounter     },
+    :entries_statistics   =>{ :dn =>"cn=Entries,cn=Statistics,cn=Monitor",  :attribute =>:monitorcounter     },
+    :initiated_operations =>{ :dn =>"cn=Operations,cn=Monitor",             :attribute =>:monitoropinitiated },
+    :completed_operations =>{ :dn =>"cn=Operations,cn=Monitor",             :attribute =>:monitoropcompleted },
+    :bind_operations      =>{ :dn =>"cn=Bind,cn=Operations,cn=Monitor",     :attribute =>:monitoropcompleted },
+    :unbind_operations    =>{ :dn =>"cn=Unbind,cn=Operations,cn=Monitor",   :attribute =>:monitoropcompleted },
+    :search_operations    =>{ :dn =>"cn=Search,cn=Operations,cn=Monitor",   :attribute =>:monitoropcompleted },
+    :compare_operations   =>{ :dn =>"cn=Compare,cn=Operations,cn=Monitor",  :attribute =>:monitoropcompleted },
+    :modify_operations    =>{ :dn =>"cn=Modify,cn=Operations,cn=Monitor",   :attribute =>:monitoropcompleted },
+    :modrdn_operations    =>{ :dn =>"cn=Modrdn,cn=Operations,cn=Monitor",   :attribute =>:monitoropcompleted },
+    :add_operations       =>{ :dn =>"cn=Add,cn=Operations,cn=Monitor",      :attribute =>:monitoropcompleted },
+    :delete_operations    =>{ :dn =>"cn=Delete,cn=Operations,cn=Monitor",   :attribute =>:monitoropcompleted },
+    :abandon_operations   =>{ :dn =>"cn=Abandon,cn=Operations,cn=Monitor",  :attribute =>:monitoropcompleted },
+    :extended_operations  =>{ :dn =>"cn=Extended,cn=Operations,cn=Monitor", :attribute =>:monitoropcompleted },
+    :write_waiters        =>{ :dn =>"cn=Write,cn=Waiters,cn=Monitor",       :attribute =>:monitorcounter     },
+    :read_waiters         =>{ :dn =>"cn=Read,cn=Waiters,cn=Monitor",        :attribute =>:monitorcounter     }
   }
   
   KEYS = MAPPER.keys
   
+  private 
+  def get_monitor_parameter(dn, attribute)
+    
+    @connection.search(
+      :filter     =>Net::LDAP::Filter.eq("objectClass", "*"),
+      :base       =>dn, 
+      :attributes =>[attribute]) 
+  end
+    
+    
+  public
   def configure(config) 
     
     @connection ||= Connection.new(config).connection
-    
     self
   end
   
-  def get(key)
+  
+  def validate_key(key)
+  
+    key == :all || KEYS.include?(key)
+  end
+  
+  
+  def get(key=:all)
 
-    raise StandardError.new("The key: #{key} do not exist") unless KEYS.include?(key)
+    raise ExtractorError.new("You need to configure a valid LDAP connection") unless @connection
+    raise ArgumentError.new("Invalid key: #{key}, valid keys are: #{KEYS.inspect}") unless validate_key(key)
     
-    dn        = MAPPER[key][:dn]
-    attribute = MAPPER[key][:attribute]
+    result = OpenStruct.new()
+    keys   = (key == :all ? KEYS : [key]) 
+
+    keys.each do |k|
     
-    raise ArgumentError.new("You need to configure a valid LDAP connection before call this") unless @connection
-    raise ArgumentError.new("Invalid key: #{key}") unless KEYS.include?(key)
+      dn    = MAPPER[k][:dn]
+      attr  = MAPPER[k][:attribute]
+      entry = get_monitor_parameter(dn, attr)
+        
+      raise ExtractorError.new("Problems getting #{dn} data") unless entry
     
-    entry = @connection.search( 
-      :filter     =>Net::LDAP::Filter.eq("objectClass", "*"),
-      :base       =>dn, 
-      :attributes =>[attribute])
-   
-    raise StandardError.new("Problems getting information: #{dn} from Monitor backend") unless entry
+      result.send("#{k}=", entry[0][attr][0].to_i)
+    end
     
-    entry[0][attribute][0]
+    result
+  rescue Exception => exc
+    raise ExtractorError.new(exc.message)
   end
 end

data/lib/openldap_monitor_extractor/connection.rb

@@ -13,6 +13,7 @@ module OpenldapMonitorExtractor
 
     def self.builder(parameters={ })
 
+      parameters  = { :base =>DEFAULT_MONITOR_BASE }.merge(parameters)
       uri         = URI.parse(parameters[:url])
       auth        = { 
         :method   =>:simple, 
@@ -23,7 +24,7 @@ module OpenldapMonitorExtractor
       cparameters = {
 	      :host =>uri.host,
 	      :port =>uri.port,
-	      :base =>DEFAULT_MONITOR_BASE,
+	      :base =>parameters[:base],
 	      :auth =>auth
       }
       

data/lib/openldap_monitor_extractor/version.rb

@@ -1,4 +1,4 @@
 module OpenldapMonitorExtractor
 
-  VERSION = "0.0.3"
+  VERSION = "0.0.4"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: openldap_monitor_extractor
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.4
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2013-05-28 00:00:00.000000000 Z
+date: 2013-05-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: net-ldap
@@ -87,7 +87,7 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
-- examples/get_totalconnections.rb
+- examples/get_all_stats.rb
 - examples/sample_settings.yml
 - lib/openldap_monitor_extractor.rb
 - lib/openldap_monitor_extractor/connection.rb
@@ -110,7 +110,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -566327192966787389
+      hash: 720276235540630360
 required_rubygems_version: !ruby/object:Gem::Requirement
   none: false
   requirements:
@@ -119,7 +119,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
       segments:
       - 0
-      hash: -566327192966787389
+      hash: 720276235540630360
 requirements: []
 rubyforge_project: 
 rubygems_version: 1.8.25