munin_manager 1.2.1

data/lib/munin_manager/starling/starling_stats.rb

@@ -0,0 +1,66 @@
+class Starling
+  def stats
+     raise MemCacheError, "No active servers" unless active?
+     server_stats = {}
+
+     @servers.each do |server|
+       sock = server.socket
+       raise MemCacheError, "No connection to server" if sock.nil?
+
+       value = nil
+       begin
+         sock.write "stats\r\n"
+         stats = {}
+         while line = sock.gets do
+           break if line == "END\r\n"
+           if line =~ /^STAT/ then
+             stat, name, value = line.split
+             stats[name] = case name
+                           when 'version'
+                             value
+                           when 'rusage_user', 'rusage_system' then
+                             seconds, microseconds = value.split(/:/, 2)
+                             microseconds ||= 0
+                             Float(seconds) + (Float(microseconds) / 1_000_000)
+                           else
+                             if value =~ /^\d+$/ then
+                               value.to_i
+                             else
+                               value
+                             end
+                           end
+           end
+         end
+         server_stats["#{server.host}:#{server.port}"] = stats
+       rescue SocketError, SystemCallError, IOError => err
+         puts err.inspect
+         server.close
+         raise MemCacheError, err.message
+       end
+     end
+
+     server_stats
+   end
+
+   ##
+   # returns the number of items in +queue+. If +queue+ is +:all+, a hash of all
+   # queue sizes will be returned.
+
+   def sizeof(queue, statistics = nil)
+     statistics ||= stats
+
+     if queue == :all
+       queue_sizes = {}
+       available_queues(statistics).each do |queue|
+         queue_sizes[queue] = sizeof(queue, statistics)
+       end
+       return queue_sizes
+     end
+
+     statistics.inject(0) { |m,(k,v)| m + v["queue_#{make_cache_key(queue)}_items"].to_i }
+   end
+   
+   def queue_names
+     return available_queues
+   end
+end

data/test/haproxy_response_time_test.rb

@@ -0,0 +1,37 @@
+require File.dirname(__FILE__) + "/test_helper"
+
+class HaproxyResponseTimeTest < Test::Unit::TestCase
+  include MuninManager::Plugins
+  
+  def setup
+    @reader = HaproxyResponseTime.new("#{TEST_DIR}/logs/haproxy.log")
+    @reader.state_dir = "#{TEST_DIR}/tmp"
+  end
+  
+  def teardown
+    File.unlink(@reader.state_file)
+  rescue
+    # Do nothing
+  end
+  
+  def test_parsing
+    @reader.collect!
+
+    expected_values = {
+      :rails_action => "0.0005463880",
+      :total => "0.5552000000",
+      :client_connect => "0.0000000000",
+      :waiting_in_queue => "0.0000000000",
+      :server_connect => "0.0000000000",
+      :server_response => "0.5547000000"
+    }
+    
+    expected_values.each do |field, val|
+      assert_equal(val, @reader.data[field], "Value mismatch in #{field}")
+    end
+  end
+  
+  def test_has_help_text
+    assert !HaproxyResponseTime.help_text.empty?
+  end
+end

data/test/log_reader_test.rb

@@ -0,0 +1,31 @@
+require File.dirname(__FILE__) + "/test_helper"
+
+class LogReaderTest < Test::Unit::TestCase
+  def setup
+    @reader = RailsLogReader.new("#{HERE}/logs/rails.log")
+    
+    File.open(@reader.file_name, 'w') do |f|
+      10.times {f.puts("Completed in")}
+      20.times {f.puts("Some other stuff")}
+    end
+  end
+  
+  def teardown
+    File.unlink(@reader.state_file)
+  end
+  
+  def test_remembers_last_position_in_log
+    @reader.collect!
+    assert_equal 10, @reader.num_requests
+    
+    File.open(@reader.file_name, 'a+') do |f|
+      10.times {f.puts("Completed in")}
+    end
+    
+    @new_reader = RailsLogReader.new(@reader.file_name)
+    @new_reader.collect!
+    
+    assert_equal 10, @new_reader.num_requests
+    assert_equal 40, File.new(@new_reader.file_name).readlines.length
+  end
+end

data/test/rails_response_time_test.rb

@@ -0,0 +1,51 @@
+require File.dirname(__FILE__) + "/test_helper"
+
+class RailsResponseTimeTest < Test::Unit::TestCase
+  include MuninManager::Plugins
+  
+  def setup
+    @reader = RailsResponseTime.new("#{TEST_DIR}/logs/rails.log")
+    @reader.state_dir = "#{TEST_DIR}/tmp"
+  end
+  
+  def teardown
+    File.unlink(@reader.state_file)
+  rescue
+    # Do nothing
+  end
+
+  def test_has_default_search_pattern
+    assert_kind_of Regexp, @reader.search_pattern
+  end
+  
+  def test_parsing_with_default_search_pattern
+    @reader.collect!
+    assert @reader.data.keys.length > 0
+
+    @reader.data.each_pair do |k, v|
+      assert v > 0, "Value mismatch for #{k}"
+    end
+  end
+
+  def test_parsing_with_non_default_search_string
+    @reader.search_pattern = /UsersController.*/
+    @reader.collect!
+
+    @reader.data.each_pair do |action_name, time|
+      assert_not_nil /UsersController.*/.match(action_name), "'%s' does not match search pattern" % action_name
+    end
+  end
+
+  def test_config_parses_log
+    assert @reader.config.split("\n").select {|line| not line.starts_with?("graph_")}.length > 0
+  end
+
+  def test_config_does_not_save_state
+    @reader.config
+    assert !File.exists?(@reader.state_file)
+  end
+  
+  def test_has_help_text
+    assert !RailsResponseTime.help_text.empty?
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,32 @@
+TEST_DIR = File.dirname(__FILE__)
+%w(lib ext bin test).each do |dir| 
+  $LOAD_PATH.unshift "#{TEST_DIR}/../#{dir}"
+end
+
+%w(rubygems test/unit ruby-debug munin_manager).each do |f|
+  require f
+end
+
+class RailsLogReader < MuninManager::LogReader
+  def initialize(*params)
+    super
+    @state_dir = "#{TEST_DIR}/tmp"
+    @state_file = File.join(@state_dir, @me)
+  end
+  
+  def scan(log_file)
+    @req_counter = 0
+    loop do
+      line = log_file.readline
+      @req_counter += 1 if line =~ /Completed in/
+    end
+  end
+
+  def process!
+    # Do nothing
+  end
+
+  def num_requests
+    @req_counter
+  end
+end

metadata

@@ -0,0 +1,86 @@
+--- !ruby/object:Gem::Specification 
+name: munin_manager
+version: !ruby/object:Gem::Version 
+  version: 1.2.1
+platform: ruby
+authors: 
+- Rohith Ravi
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-10-07 00:00:00 -07:00
+default_executable: munin_manager
+dependencies: []
+
+description: 
+email: entombedvirus@gmail.com
+executables: 
+- munin_manager
+extensions: []
+
+extra_rdoc_files: 
+- README.markdown
+files: 
+- Manifest
+- README.markdown
+- Rakefile
+- VERSION
+- bin/munin_manager
+- bin/runner
+- lib/munin_manager.rb
+- lib/munin_manager/acts_as_munin_plugin.rb
+- lib/munin_manager/log_reader.rb
+- lib/munin_manager/plugins.rb
+- lib/munin_manager/plugins/fbproxy_latency.rb
+- lib/munin_manager/plugins/haproxy_app_response_time.rb
+- lib/munin_manager/plugins/haproxy_response_time.rb
+- lib/munin_manager/plugins/haproxy_rps.rb
+- lib/munin_manager/plugins/network_latency.rb
+- lib/munin_manager/plugins/notification_classification.rb
+- lib/munin_manager/plugins/packet_loss.rb
+- lib/munin_manager/plugins/rails_rendering.rb
+- lib/munin_manager/plugins/rails_response_time.rb
+- lib/munin_manager/plugins/scribe_net.rb
+- lib/munin_manager/plugins/starling_age.rb
+- lib/munin_manager/plugins/starling_net.rb
+- lib/munin_manager/plugins/starling_ops.rb
+- lib/munin_manager/plugins/starling_queue.rb
+- lib/munin_manager/starling/starling_stats.rb
+- test/haproxy_response_time_test.rb
+- test/log_reader_test.rb
+- test/rails_response_time_test.rb
+- test/test_helper.rb
+has_rdoc: true
+homepage: http://github.com/entombedvirus/jeweler
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.2
+signing_key: 
+specification_version: 3
+summary: Tool to maintain and install munin plugins written in Ruby
+test_files: 
+- test/haproxy_response_time_test.rb
+- test/log_reader_test.rb
+- test/rails_response_time_test.rb
+- test/test_helper.rb