Munin contributed plugins

#!/usr/bin/ruby

# this plugin reports various performance statistics from a puppetmaster

# server and client

#

# linked as puppet_mem, it will display memory usage of the puppet

# client and puppet master on the local server.

#

# linked as puppet_clients, it will report the average compile time, the

# number of clients that checked in in the last 5 minutes and 24h and

# the number of known clients.

#

# it requires read access to the puppet logfile.

#

# CONFIGURATION

#

# [puppet*]

# env.puppet_logfile /var/log/message

# env.puppet_logformat "^%b %d"

#

# the logfile is where the puppetmaster is expected to log its

# compilation statistics. the format is the format of the date syslog

# writes to the file, which may vary according to locale and

# configuration.

# returns the mem usage of a given process

def plist(psname)

	counter = 0

	%x{ps h -o rss,size,vsize,pcpu -C #{psname}}.each do |ps|

		rss,size,vsize,cpu = ps.split

		counter += 1

		puts "#{psname}_#{counter}.value #{rss}" 

	end

	return

end

# reports how many clients compiled in the last 5 minutes

# repotrs how many unique clients compiled since the begining of the day

# report the average compilation time for all clients in the last 5 minutes.

def phaselog

	logfile = ENV['puppet_logfile'] || '/var/log/messages'

	count,avg,day_count_unique,day_count = 0 ,0 ,0, 0

	t = Time.now

	dateformat = ENV['puppet_logformat'] || "^%b %d"

	today = t.strftime(dateformat)

	hour = t.strftime(dateformat + " %H:")

	m = t.min.to_i

	last5m = ""

	6.times do |i|

		last5m += hour

		last5m += "0" if (m-i) < 10

		last5m += (m-i).to_s  

		last5m += "|" unless i==5

	end

	hosts = Array.new

	regexp = ".* for (.*) in (.*) seconds"

	File.open(logfile).grep(/#{today}/).grep(/Compiled configuration|Compiled catalog/).each do |line|

		case line

			when /#{last5m}/ then

				if line =~ /#{regexp}/

					avg += $2.to_f

					count += 1

					unless hosts.include?($1)

						hosts << $1

					end

				end

			when /#{regexp}/ then

				day_count += 1

				unless hosts.include?($1)

					hosts << $1 

					day_count_unique += 1

				end

		end

	end

	puts "avg_compile.value #{(avg / count).to_s[0..3]}" unless count == 0

	puts "last5m_count.value #{count}" 

	puts "last24h_unique_count.value #{day_count_unique}" 

end

case ARGV[0]

	when 'config'

		case $0

		when /puppet_mem/

			puts "graph_title puppet memory usage"

			puts "graph_vlabel memory"

			puts "puppetd_1.label puppetd"

			# find out how many mongrel process we have - if any

			File.open('/etc/sysconfig/puppetmaster') do |line|

				@pm_process = line.grep(/PUPPETMASTER_INSTANCES/).to_s.split('=')[1].to_i

			end

			if @pm_process > 0 

				@pm_process.times do |i|

					puts "puppetmasterd_#{i+1}.label puppetmasterd #{i+1}"

				end

			else

				puts "puppetmaster.label puppetmasterd"

			end

		when /puppet_clients/

			puts "graph_title puppet clients usage"

			puts "graph_vlabel clients"

			puts "known_clients.label Known Clients"

			puts "avg_compile.label Average configuration compile, in seconds"

			puts "last5m_count.label Clients in the last 5 minutes"

			puts "last24h_unique_count.label unique clients in the last 24 hours"

		end

		puts "graph_category puppet"

		exit 0

	when 'autoconf'

		case $0

			when /puppet_mem/,/puppet_clients/

				puts "yes"

			else 

				puts "no"

			exit 0

		end

	else

		plist("puppetmasterd") if $0 =~ /puppet_mem$/

		plist("puppetd") if $0 =~ /puppet_mem$/

		if $0 =~ /puppet_clients$/		

			puts "known_clients.value #{Dir.entries('/var/lib/puppet/yaml/facts/').size-2}"

			phaselog

		end

end