Add a new Squid graph to count the IPs being delayed due to no-referer, and restrict...
[chef.git] / cookbooks / munin / files / default / plugins / api_calls_
index 58797b8330ffa276cfed75f6b1e52b380aec3881..17bba1625aeb57b7b4e6823de1b423f89065f437 100755 (executable)
@@ -1,18 +1,18 @@
 #!/usr/bin/ruby
 
-require 'rubygems'
-require 'date'
-require 'hpricot'
-require 'open-uri'
+require "rubygems"
+require "date"
+require "hpricot"
+require "open-uri"
 
 def uris_from_status(server)
   file = open("http://#{server}/server-status").read
   doc = Hpricot.parse(file)
-  tables = doc / 'table'
-  rows = (tables[0] / 'tr')[1..-1]
-  data = rows.collect { |r| (r  / 'td').collect(&:inner_html) }
+  tables = doc / "table"
+  rows = (tables[1] / "tr")[1..-1]
+  data = rows.collect { |r| (r / "td").collect(&:inner_html) }
   # filter where the PID is numeric, status is 'W' and host matches the server
-  matching_data = data.select { |r| (r[1].to_i > 0) && r[3].match(/W/) && r[11].match(server) }
+  matching_data = data.select { |r| r[1].to_i > 0 && r[3].match(/W/) && r[11].match(server) }
   # return only the URI part
   matching_data.collect { |r| r[12] }
 end
@@ -26,7 +26,7 @@ CALL_TYPES = {
   :trkpts => "GPX trackpoints calls",
   :web => "Web site traffic",
   :other => "Other API calls"
-}
+}.freeze
 
 def categorise_uri(line)
   uri = line.split(" ")[1]
@@ -45,7 +45,7 @@ end
 
 server = $PROGRAM_NAME.match("api_calls_(.*)")[1]
 
-if ARGV[0] == 'config'
+if ARGV[0] == "config"
   puts "graph_title Active requests"
   puts "graph_vlabel Number of requests"
   puts "graph_category api"
@@ -54,7 +54,7 @@ if ARGV[0] == 'config'
 else
   counts = uris_from_status(server)
            .collect { |x| categorise_uri(x) }
-           .each_with_object(Hash.new) do |h, e|
+           .each_with_object({}) do |e, h|
     if h.key? e
       h[e] += 1
     else