#!/usr/bin/env ruby

=begin

mongrel_memory - A munin plugin for OpenSolaris to monitor memory size of
                 each individual mongrel process
Copyright (C) 2009 Matthias Marschall - mm@agileweboperations.com

Based on:
mongrel_process_memory - A munin plugin to monitor memory size of
                         each individual mongrel process
Copyright (C) 2007 Ben VandenBos and Avvo, Inc.

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License version 2
as published by the Free Software Foundation.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

Author: Ben VandenBos
Contributors: Adam Jacob (<adam@hjksolutions.com>)
              Ryan Woodrum
              Matthias Marschall (mm@agileweboperations.com)

#%# family=auto
#%# capabilities=autoconf

=end

module Munin
  class MongrelProcessMemory
    def run
      pid_port_map = get_pids
      port_list = {}
      pid_port_map.sort.each do |pid, port|
        rss = `pmap -x #{pid} | grep total`.split(' ')[3]
        puts "mongrel_#{port}.value #{rss}"
      end
    end

    def get_pids
      h = {}
      pids = []
      pids += `pgrep mongrel_rails`.split("\n")
      pids += `pgrep ruby`.split("\n")
      pids.each do |pid|
        l = `pargs -l #{pid}`
        l =~ /-p (\d+)/
        h[pid] = Regexp.last_match(1) if Regexp.last_match(1)
      end
      h
    end

    def autoconf
      get_pids.length > 0
    end
  end
end

mpm = Munin::MongrelProcessMemory.new

case ARGV[0]
when 'config'
  puts 'graph_title Mongrel Memory'
  puts 'graph_vlabel RSS'
  puts 'graph_category memory'
  puts 'graph_args --base 1024 -l 0'
  puts 'graph_scale yes'
  puts 'graph_info Tracks the size of individual mongrel processes'
  mpm.get_pids.values.sort.each do |port|
    puts "mongrel_#{port}.label mongrel_#{port}"
    puts "mongrel_#{port}.info Process memory"
    puts "mongrel_#{port}.type GAUGE"
    puts "mongrel_#{port}.min 0"
  end
when 'autoconf'
  if mpm.autoconf
    puts 'yes'
    exit 0
  end
  puts 'no'
  exit 0
else
  mpm.run
end