#!/usr/bin/env ruby
# 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

module Munin
  class MongrelProcessMemory

    def run
      pid_port_map = get_pids()
      port_list = Hash.new
      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 = Hash.new
      pids = []
      pids += `pgrep mongrel_rails`.split("\n")
      pids += `pgrep ruby`.split("\n")
      pids.each { |pid|
        l = `pargs -l #{pid}`
      	l =~ /-p (\d+)/
      	h[pid] = $1 if $1
      }
      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