vspheremonitor 0.0.7 → 0.1.0

data/bin/vspheremonitor

@@ -6,45 +6,57 @@ begin
   require "json"
   require "yaml"
   require 'optparse'
-rescue LoadError
-  puts 'Some of the required gems have not been found'
-  exit
+  require 'awesome_print'
+  require 'cri'
+rescue LoadError => e
+  puts "Unable to load library: #{e}"
+  exit 1
 end
 
-options = {}
-OptionParser.new do |opts|
-  opts.banner = "Usage: vspheremonitor [options]"
 
-  opts.on("-v", "--verbose", "Be less quiet") do |v|
-    options[:verbose] = v
-  end
+command = Cri::Command.define do
+  name __FILE__
+  usage __FILE__ + " [options]"
+  summary 'Collect metrics from VMware vSphere'
 
-  opts.on("-c", "--config FILE", "Specify the configuration file") do |conf|
-    options[:configfile] = conf
+  flag   :h, :help,  'show help for this command' do |value, cmd|
+    puts cmd.help
+    exit 0
   end
+  flag   :v, :verbose,  'Be less quiet'
+  option :c, :conf,     'Specify the configuration file', :argument => :required
 
-end.parse!
+  run do |opts, args, cmd|
+    configfile = opts[:conf] || 'etc/vsphere.yaml'
 
-def run (options)
+    if opts[:verbose]
+      starttime = Time.now
+      puts "Initializing at #{starttime}"
+    end
 
-  if options[:configfile].nil?
-    configfile = 'etc/vsphere.yaml'
-  else
-    configfile = options[:configfile]
-  end
+    puts "Loading configuration #{configfile}" if opts[:verbose]
+    config = YAML::load(File.read(configfile))
+    config[:opts] = opts
 
-  config = YAML::load(File.read(configfile))
-  vim = RbVmomi::VIM.connect :host     => config[:host],
-                             :user     => config[:user],
-                             :password => config[:password],
-                             :insecure => true
+    p = VSphereMonitoring.new(config)
 
-  data = Hash.new
-  data[:vsphere] = VSphereMonitoring.process_all_datacenters(config[:datacenters],vim)
-  data
+  end
 
 end
 
-output = run(options)
-puts output.to_json
+#options = {}
+#OptionParser.new do |opts|
+#  opts.banner = "Usage: vspheremonitor [options]"
+#
+#  opts.on("-v", "--verbose", "Be less quiet") do |v|
+#    options[:verbose] = v
+#  end
+#
+#  opts.on("-c", "--config FILE", "Specify the configuration file") do |conf|
+#    options[:configfile] = conf
+#  end
+#
+#end.parse!
+
+command.run(ARGV)
 

data/etc/vsphere.yaml.sample

@@ -1,7 +1,4 @@
 ---
-:datacenters:
-  - dc1
-  - dc2
 :host: server.example.net
 :user: monitor
 :password: secret

data/lib/vspheremonitoring.rb

@@ -4,10 +4,30 @@ require 'pp'
 require 'alchemist'
 require 'yaml'
 
-module VSphereMonitoring
-  module_function
+class VSphereMonitoring
+
+  def initialize (config)
+    @config = config
+    @verbose = config[:opts][:verbose]
+
+    speak("Building connection to vsphere")
+    @vim = RbVmomi::VIM.connect :host     => config[:host],
+                               :user     => config[:user],
+                               :password => config[:password],
+                               :insecure => true
+
+    p = Hash.new
+    p[:vsphere] = process_all_datacenters()
+    puts p.to_json
+  end
+  #module_function
+
+  def speak(what)
+    puts what if @verbose
+  end
 
   def host_memory (host)
+    speak("gathering memory stats for #{host.name}")
     data            = Hash.new
     data['total']   = host.hardware.memorySize.bytes.to.megabytes.to_f
     data['usage']   = host.summary.quickStats.overallMemoryUsage.megabytes.to_f
@@ -16,6 +36,7 @@ module VSphereMonitoring
   end
 
   def host_cpu (host)
+    speak("gathering cpu stats for #{host.name}")
     data = Hash.new
     data['Mhz']      = host.hardware.cpuInfo.hz / 1000 / 1000
     data['cores']    = host.hardware.cpuInfo.numCpuCores
@@ -26,6 +47,7 @@ module VSphereMonitoring
   end
 
   def host_stats (host)
+    speak("gathering host stats for #{host.name}")
     data = Hash.new
     data['memory']  = host_memory(host)
     data['cpu']     = host_cpu(host)
@@ -33,6 +55,7 @@ module VSphereMonitoring
   end
 
   def cluster_memory (cluster)
+    speak("gathering cluster memory stats for #{cluster.name}")
     data = Hash.new
     data['totalMemory']     = cluster.summary.totalMemory
     data['effectiveMemory'] = cluster.summary.effectiveMemory
@@ -40,6 +63,7 @@ module VSphereMonitoring
   end
 
   def cluster_cpu (cluster)
+    speak("gathering cluster cpu stats for #{cluster.name}")
     data = Hash.new
     data['totalCpu']      = cluster.summary.totalCpu
     data['numCpuCores']   = cluster.summary.numCpuCores
@@ -49,6 +73,7 @@ module VSphereMonitoring
   end
 
   def cluster_stats (cluster)
+    speak("gathering cluster stats for #{cluster.name}")
     data = Hash.new
     data['cpu']               = cluster_cpu(cluster)
     data['memory']            = cluster_memory(cluster)
@@ -61,6 +86,7 @@ module VSphereMonitoring
   end
 
   def datastore_stats (datastore)
+    speak("gathering datastore stats for #{datastore.name}")
     data = Hash.new
     datastore.RefreshDatastore
     capacity = datastore.summary.capacity
@@ -73,11 +99,14 @@ module VSphereMonitoring
   end
 
   def network_stats (network)
+    speak("gathering network stats for #{network.name}")
     data = Hash.new
     data
   end
 
   def process_datacenter (dc)
+    speak("collecting metrics for datacenter #{dc.name}")
+
     data = Hash.new
     data['hypers'] = Hash.new
     data['clusters'] = Hash.new
@@ -107,12 +136,14 @@ module VSphereMonitoring
     data
   end
 
-  def process_all_datacenters (datacenterlist,vim)
+  def process_all_datacenters ()
+
+    rootFolder = @vim.serviceInstance.content.rootFolder
+    dclist = rootFolder.children.select {|d| d.class == RbVmomi::VIM::Datacenter }
 
     data = Hash.new
-    datacenterlist.each do |datacenter|
-      dc = vim.serviceInstance.find_datacenter datacenter
-      data[dc.name] = VSphereMonitoring.process_datacenter(dc)
+    dclist.each do |datacenter|
+      data[datacenter.name] = process_datacenter(datacenter)
     end
     data
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: vspheremonitor
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.1.0
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-08-02 00:00:00.000000000 Z
+date: 2012-11-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: json
@@ -91,8 +91,9 @@ required_rubygems_version: !ruby/object:Gem::Requirement
       version: '0'
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.24
+rubygems_version: 1.8.23
 signing_key: 
 specification_version: 3
 summary: A tool to get some basic statistics out of a vSphere installation.
 test_files: []
+has_rdoc: