bloopletech-webstats 0.1.0 → 0.2.0

data/Rakefile

@@ -8,10 +8,10 @@ begin
 
       s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
       s.authors = ["Brenton Fletcher"]
-      s.date = %q{2009-04-22}
+      s.date = Date.today.strftime("%Y-%m-%d")
       s.description = s.summary = %q{Display server CPU/Memory/Disk Usage on a web page, suitable for remote performance monitoring.}
       s.email = %q{i@bloople.net}
-      s.files = Dir['**/*']
+      s.files = Dir['**/*'].reject { |fn| fn =~ /(\.o|\.so|Makefile)$/ }
       s.executables = ['webstats']
       s.extensions = ["server/data_providers/extconf.rb"]
       s.has_rdoc = false

data/VERSION.yml

@@ -1,4 +1,4 @@
 --- 
 :major: 0
-:minor: 1
+:minor: 2
 :patch: 0

data/server/data_providers/cpu_info.rb

@@ -1,5 +1,7 @@
 class DataProviders::CpuInfo
-  def initialize
+  def initialize(settings)
+    @settings = self.class.default_settings.merge(settings)
+
     @readings = []
     @mutex = Mutex.new
 
@@ -31,7 +33,7 @@ class DataProviders::CpuInfo
         last_idle = idle
         last_iowait = iowait
         last_time = time
-        sleep(2.5)
+        sleep(@settings[:update_rate])
       end
     end
   end  
@@ -41,8 +43,8 @@ class DataProviders::CpuInfo
     @mutex.synchronize do
       unless @readings.empty?
         out[:usage] = @readings.first
-        out[:status] = 'warning' unless @readings.detect { |r| out[:usage] < 95 }
-        out[:status] = 'danger' unless @readings.detect { |r| out[:usage] < 99.5 }
+        out[:status] = 'warning' unless @readings.detect { |r| out[:usage] < @settings[:usage_warning_level] }
+        out[:status] = 'danger' unless @readings.detect { |r| out[:usage] < @settings[:usage_danger_level] }
       end
     end
     out[:loadavg_1], out[:loadavg_5], out[:loadavg_15] = IO.readlines("/proc/loadavg").first.split(' ', 4).map { |v| v.to_f }
@@ -58,12 +60,12 @@ sc.innerHTML = "<div class='major_figure'><span class='title'>Usage</span><span
 } })
   end
 
-  def information
-    { :name => "CPU Info", :in_sentence => 'CPU load', :importance => importance }
+  def self.default_settings
+    { :update_rate => 2.5, :usage_warning_level => 95, :usage_danger_level => 99.5 }
   end
 
-  def importance
-    100
+  def information
+    { :name => "CPU Info", :in_sentence => 'CPU load', :importance => 100 }
   end
 
   def kill

data/server/data_providers/disk_activity.rb

@@ -1,5 +1,7 @@
 class DataProviders::DiskActivity
-  def initialize
+  def initialize(settings)
+    @settings = self.class.default_settings.merge(settings)
+
     @reads_sec = 0
     @writes_sec = 0
 
@@ -20,7 +22,7 @@ class DataProviders::DiskActivity
         last_reads = reads
         last_writes = writes
         last_time = time
-        sleep(2.5)
+        sleep(@settings[:update_rate])
       end
     end
   end
@@ -30,18 +32,18 @@ class DataProviders::DiskActivity
   end
 
   def renderer
-    information.merge({ :name => "Disk Activity", :in_sentence => "Disk Activity", :importance => importance, :contents => %{
+    information.merge({ :contents => %{
 sc.innerHTML = "<div class='major_figure'><span class='title'>Reads</span><span class='figure'>" + data_source['reads'] + "</span><span class='unit'>mb/s</span></div>" +
 "<div class='major_figure'><span class='title'>Writes</span><span class='figure'>" + data_source['writes'] + "</span><span class='unit'>mb/s</span></div>";
 } })
   end
 
-  def information
-    { :name => "Disk Activity", :in_sentence => "Disk Activity", :importance => importance }
+  def self.default_settings
+    { :update_rate => 2.5 }
   end
 
-  def importance
-    70
+  def information
+    { :name => "Disk Activity", :in_sentence => "Disk Activity", :importance => 70 }
   end
 
   def kill

data/server/data_providers/disk_usage.rb

@@ -1,5 +1,6 @@
 class DataProviders::DiskUsage
-  def initialize
+  def initialize(settings)
+    @settings = self.class.default_settings.merge(settings)
   end
 
   def get
@@ -16,15 +17,15 @@ class DataProviders::DiskUsage
     out[:mounts].map do |mp|
       mp[1]['free'] /= (1024.0 * 1024)
       mp[1]['total'] /= (1024.0 * 1024)
-      out[:status] = "warning" if mp[1]['free'] < 50 and mp[1]['total'] > 100 and out[:status] != 'danger'
-      out[:status] = "danger" if mp[1]['free'] < 10 and mp[1]['total'] > 20
+      out[:status] = "warning" if mp[1]['free'] < @settings[:warning_threshold] and mp[1]['total'] > @settings[:warning_minimum_mount_point_size] and out[:status] != 'danger'
+      out[:status] = "danger" if mp[1]['free'] < @settings[:danger_threshold] and mp[1]['total'] > @settings[:danger_minimum_mount_point_size]
     end
 
     out
   end
 
   def renderer
-    information.merge({ :name => "Disk Usage by Mount Point", :in_sentence => "Disk Usage", :importance => importance, :contents => %{
+    information.merge({ :contents => %{
 var temp = "";
 for(var i = 0; i < data_source['mounts'].length; i++)
 {
@@ -38,12 +39,12 @@ sc.innerHTML = temp;
 } })
   end
 
-  def information
-    { :name => "Disk Usage by Mount Point", :in_sentence => "Disk Usage", :importance => importance }
+  def self.default_settings
+    { :warning_threshold => 50, :danger_threshold => 10, :warning_minimum_mount_point_size => 100, :danger_minimum_mount_point_size => 20 }
   end
 
-  def importance
-    80
+  def information
+    { :name => "Disk Usage by Mount Point", :in_sentence => "Disk Usage", :importance => 80 }
   end
 
   def kill

data/server/data_providers/mem_info.rb

@@ -1,5 +1,7 @@
 class DataProviders::MemInfo
-  def initialize
+  def initialize(settings)
+    @settings = self.class.default_settings.merge(settings)
+
     @readings = []
     @mutex = Mutex.new
 
@@ -13,7 +15,7 @@ class DataProviders::MemInfo
           @readings.unshift(out)
           @readings.pop while @readings.length > 5
         end
-        sleep(2.5)
+        sleep(@settings[:update_rate])
       end
     end
   end
@@ -38,12 +40,12 @@ sc.innerHTML = "<div class='major_figure'><span class='title'>Free</span><span c
 } })
   end
 
-  def information
-    { :name => "Memory Info", :in_sentence => 'Memory Usage', :importance => importance }
+  def self.default_settings
+    { :update_rate => 2.5 }
   end
 
-  def importance
-    90
+  def information
+    { :name => "Memory Info", :in_sentence => 'Memory Usage', :importance => 90 }
   end
 
   def kill

data/server/data_providers/url_monitor.rb

@@ -0,0 +1,67 @@
+require 'net/http'
+
+class DataProviders::UrlMonitor
+  def initialize(settings)
+    @settings = self.class.default_settings.merge(settings)
+
+    @readings = []
+
+    @mutex = Mutex.new
+
+    @thread = Thread.new do
+      while(true)
+        @mutex.synchronize { @readings = [] }
+
+        @settings[:urls].sort.each do |url|
+          duration = -1
+          works = false
+          begin
+            start = Time.now
+            result = Net::HTTP.get(URI.parse(url))
+            duration = Time.now - start
+            works = true
+          rescue Exception => e
+          end
+          @mutex.synchronize { @readings << [url, { :response_time => duration * 1000, :works => works }] }
+        end
+        sleep(@settings[:update_rate])
+      end
+    end
+          
+  end
+
+  def get
+    out = {}
+    @mutex.synchronize { out[:urls] = @readings }
+    out[:urls].each do |(url, info)|
+      out[:status] = 'warning' if !info[:works] or info[:response_time] > @settings[:warning_response_time_threshold] and !out[:status] == 'danger'
+      out[:status] = 'danger' if !info[:works] or info[:response_time] > @settings[:danger_response_time_threshold]
+    end
+    out
+  end
+
+  def renderer
+    information.merge({ :contents => %{
+var temp = "";
+for(var i = 0; i < data_source['urls'].length; i++)
+{
+   var ud = data_source['urls'][i][1];
+   temp += "<div class='major_figure'><span class='title'>" + data_source['urls'][i][0] + "</span><span class='figure'>" +
+    (!ud['works'] ? 'Failed</span>' : ud['response_time'] + "</span><span class='unit'>ms</span>") + "</div>";
+}
+
+sc.innerHTML = temp;
+} })
+  end
+
+  def self.default_settings
+    { :update_rate => 30, :warning_response_time_threshold => 5000, :danger_response_time_threshold => 30000, :urls => ['http://localhost/'] }
+  end
+
+  def information
+    { :name => "URL Monitor", :in_sentence => "URL Monitor", :importance => 60 }
+  end
+
+  def kill
+  end
+end

data/server/webstats.rb

@@ -1,4 +1,5 @@
 require 'webrick'
+require 'yaml'
 
 if $DEBUG
   Thread.abort_on_exception
@@ -15,6 +16,24 @@ Thread.new do
   end
 end
 
+class NilClass
+  def to_json
+    "null"
+  end
+end
+
+class TrueClass
+  def to_json
+    "true"
+  end
+end
+
+class FalseClass
+  def to_json
+    "false"
+  end
+end
+
 class String
   def underscore
     self.gsub(/::/, '/').
@@ -62,17 +81,39 @@ class Symbol
 end
 
 module DataProviders
+  DATA_SOURCES_CLASSES = {}
   DATA_SOURCES = {}
-  def self.setup
+  def self.preload
     Dir.glob("#{File.dirname(__FILE__)}/data_providers/*.rb").each { |file| load file }
     DataProviders.constants.each do |c|
       c = DataProviders.const_get(c)
-      DATA_SOURCES[c.to_s.gsub(/^DataProviders::/, '').underscore] = c.new if c.is_a? Class
+      DATA_SOURCES_CLASSES[c.to_s.gsub(/^DataProviders::/, '').underscore] = c if c.is_a? Class
     end
   end
+  def self.setup(settings)
+    DATA_SOURCES_CLASSES.each_pair { |k, v| DATA_SOURCES[k] = v.new(settings[k]) }
+  end
+end
+
+DataProviders.preload
+
+WEBSTATS_PATH = File.expand_path("~/.webstats")
+
+settings = {}
+
+if File.exists?(WEBSTATS_PATH)
+  settings = YAML.load(IO.read(WEBSTATS_PATH))
+else
+  DataProviders::DATA_SOURCES_CLASSES.each_pair do |k, v|
+    settings[k] = v.default_settings
+  end
+
+  File.open(WEBSTATS_PATH, "w") do |f|
+    YAML.dump(settings, f)
+  end
 end
 
-DataProviders.setup
+DataProviders.setup(settings)
 
 class Webstats < WEBrick::HTTPServlet::AbstractServlet
   def do_GET(req, res)
@@ -144,7 +185,7 @@ body << <<-EOF
     <div id="main">
       <h1><span>Stats for #{req.host}</span></h1>
 EOF
-      DataProviders::DATA_SOURCES.sort { |a, b| b[1].importance <=> a[1].importance }.each do |(k, v)|
+      DataProviders::DATA_SOURCES.sort { |a, b| b[1].information[:importance] <=> a[1].information[:importance] }.each do |(k, v)|
         r = v.renderer
         body << %{<div class="source" id="source_#{k}"><h2><span>#{r[:name]}</span></h2><div class="source_contents" id="source_contents_#{k}">Loading...</div></div>}
       end
@@ -157,7 +198,7 @@ EOF
     elsif req.path_info == '/update'
       out = {}
       DataProviders::DATA_SOURCES.each_pair do |k, v|
-        out[k] = v.get
+        out[k] = v.get.dup
       end
 
       fix_leaves_hash(out)
@@ -179,9 +220,9 @@ EOF
       if v.is_a? Numeric
         array[i] = v.formatted
       elsif v.is_a? Hash
-        array[i] = fix_leaves_hash(array[i])
+        array[i] = fix_leaves_hash(array[i].dup)
       elsif v.is_a? Array
-        array[i] = fix_leaves_array(array[i])
+        array[i] = fix_leaves_array(array[i].dup)
       end
     end
   end
@@ -191,9 +232,9 @@ EOF
       if v.is_a? Numeric
         hash[k] = v.formatted
       elsif v.is_a? Hash
-        hash[k] = fix_leaves_hash(hash[k])
+        hash[k] = fix_leaves_hash(hash[k].dup)
       elsif v.is_a? Array
-        hash[k] = fix_leaves_array(hash[k])
+        hash[k] = fix_leaves_array(hash[k].dup)
       end
     end
   end

data/webstats.gemspec

@@ -2,11 +2,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{webstats}
-  s.version = "0.1.0"
+  s.version = "0.2.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Brenton Fletcher"]
-  s.date = %q{2009-04-22}
+  s.date = %q{2009-04-25}
   s.default_executable = %q{webstats}
   s.description = %q{Display server CPU/Memory/Disk Usage on a web page, suitable for remote performance monitoring.}
   s.email = %q{i@bloople.net}
@@ -31,6 +31,7 @@ Gem::Specification.new do |s|
     "server/data_providers/disk_usage.rb",
     "server/data_providers/extconf.rb",
     "server/data_providers/mem_info.rb",
+    "server/data_providers/url_monitor.rb",
     "server/webstats.rb",
     "webstats.gemspec"
   ]

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: bloopletech-webstats
 version: !ruby/object:Gem::Version 
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors: 
 - Brenton Fletcher
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-04-22 00:00:00 -07:00
+date: 2009-04-25 00:00:00 -07:00
 default_executable: webstats
 dependencies: []
 
@@ -37,6 +37,7 @@ files:
 - server/data_providers/disk_usage.rb
 - server/data_providers/extconf.rb
 - server/data_providers/mem_info.rb
+- server/data_providers/url_monitor.rb
 - server/webstats.rb
 - webstats.gemspec
 has_rdoc: false