From f6d7cf5965c849e8d00cb9dbe6cdaf8927c66e51 Mon Sep 17 00:00:00 2001
From: Jasper Lievisse Adriaanse <jasper@humppa.nl>
Date: Tue, 12 Aug 2014 22:50:12 +0200
Subject: [PATCH] Add plugin to retrieve the time puppet agent took to apply
 the catalog

---
 plugins/puppet/puppet_runtime | 45 +++++++++++++++++++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100755 plugins/puppet/puppet_runtime

diff --git a/plugins/puppet/puppet_runtime b/plugins/puppet/puppet_runtime
new file mode 100755
index 00000000..c4eea5ca
--- /dev/null
+++ b/plugins/puppet/puppet_runtime
@@ -0,0 +1,45 @@
+#!/usr/bin/ruby
+
+# This plugin reports the duration of the most recent puppet agent run.
+# It requires read access to the puppet logfile (defaults to /var/log/messages).
+#
+# CONFIGURATION
+#
+# [puppet*]
+# env.puppet_logfile /var/log/message
+# env.puppet_logformat "^%b %d"
+#
+# The logfile is where the puppet agent is expected to log its run time statistics.
+# The format is the format of the date syslog writes to the file, which may vary
+# according to locale and configuration.
+
+# reports how long the puppet agent took to apply the catalog
+def get_runtime
+  logfile = ENV['puppet_logfile'] || '/var/log/messages'
+  t = Time.now
+  dateformat = ENV['puppet_logformat'] || "^%b %d"
+  today = t.strftime(dateformat)
+  File.open(logfile).grep(/#{today}/).grep(/Finished catalog run in/).reverse_each do |line|
+    if line =~ /in (.*) seconds/
+      puts "runtime.value #{$1}"
+      exit 0
+    end
+  end
+end
+
+case ARGV[0]
+	when 'config'
+  		puts "graph_category puppet"
+                puts "graph_args --base 1000 -l 0"
+                puts "graph_scale no"
+  		puts "graph_title puppet catalog run time"
+  		puts "graph_vlabel Seconds"
+  		puts "runtime.label Catalog application time"
+		exit 0
+	when 'autoconf'
+		puts "yes"
+		exit 0
+	else
+  		get_runtime
+end
+