add librato chef handler

rubygems · Jul 13, 2014 · 3493297 · 3493297
1 parent f55b253
commit 3493297
Show file tree

Hide file tree

Showing 5 changed files with 87 additions and 2 deletions.
diff --git a/Berksfile.lock b/Berksfile.lock
@@ -218,7 +218,7 @@ GRAPH
   rubygems-cache (0.0.1)
     memcached (>= 0.0.0)
     rubygems (>= 0.0.0)
-  rubygems-chef (0.0.9)
+  rubygems-chef (0.0.10)
     chef-client (>= 0.0.0)
     omnibus_updater (>= 0.0.0)
   rubygems-ci (0.0.7)

diff --git a/cookbooks/rubygems-chef/metadata.rb b/cookbooks/rubygems-chef/metadata.rb
@@ -1,7 +1,7 @@
 name 'rubygems-chef'
 maintainer 'RubyGems.org ops team'
 
-version '0.0.9'
+version '0.0.10'
 
 depends 'chef-client'
 depends 'omnibus_updater'

diff --git a/cookbooks/rubygems-chef/recipes/default.rb b/cookbooks/rubygems-chef/recipes/default.rb
@@ -3,6 +3,8 @@
 # Recipe:: default
 #
 
+include_recipe 'rubygems-chef::librato'
+
 cron 'chef-client' do
   action :delete
 end

diff --git a/cookbooks/rubygems-chef/recipes/librato.rb b/cookbooks/rubygems-chef/recipes/librato.rb
@@ -0,0 +1,18 @@
+#
+# Cookbook Name:: rubygems-chef
+# Recipe:: librato
+#
+
+include_recipe 'chef-vault'
+librato_creds = chef_vault_item('librato', 'credentials')
+
+chef_gem 'chef-handler-librato'
+
+template '/etc/chef/client.d/librato.rb' do
+  source 'librato.rb'
+  mode '0644'
+  variables(
+    email: librato_creds['email'],
+    api_key: librato_creds['api_key']
+  )
+end
diff --git a/cookbooks/rubygems-chef/templates/default/librato.rb b/cookbooks/rubygems-chef/templates/default/librato.rb
@@ -0,0 +1,65 @@
+# Mostly taken from https://github.com/bscott/chef-handler-librato
+# and https://github.com/CozyCo/chef-handler-librato
+
+require 'rubygems'
+Gem.clear_paths
+require 'librato/metrics'
+require 'chef'
+require 'chef/handler'
+
+class LibratoReporting < Chef::Handler
+  attr_accessor :email, :api_key
+
+  def report
+    @source = node.name
+
+    Librato::Metrics.authenticate @email, @api_key
+
+    gauge_metrics = {}
+    counter_metrics = {}
+    gauge_metrics[:updated_resources] = run_status.updated_resources.length
+    gauge_metrics[:all_resources] = run_status.all_resources.length
+    gauge_metrics[:elapsed_time] = run_status.elapsed_time.to_i
+
+    if run_status.success?
+      counter_metrics[:success] = 1
+      counter_metrics[:fail] = 0
+    else
+      counter_metrics[:success] = 0
+      counter_metrics[:fail] = 1
+    end
+
+    gauge_metrics.each do |metric, value|
+      Chef::Log.debug("#{metric} #{value} #{Time.now}")
+      begin
+        Librato::Metrics.submit :"chef.#{metric}" => { type: :gauge, value: value, source: @source }
+      rescue => e
+        puts "#{e}"
+      end
+    end
+
+    counter_metrics.each do |metric, value|
+      Chef::Log.debug("#{metric} #{value} #{Time.now}")
+      begin
+        Librato::Metrics.submit :"chef.#{metric}" => { type: :counter, value: value, source: @source }
+      rescue => e
+        puts "#{e}"
+      end
+    end
+    begin
+      Librato::Metrics.annotate 'chef.runs', "Chef run #{Time.now}",
+                                source: @source,
+                                description: "updated #{run_status.updated_resources.length} resources",
+                                start_time: run_status.start_time.to_i,
+                                end_time: run_status.end_time.to_i
+    rescue => e
+      puts "#{e}"
+    end
+  end
+end
+
+librato_handler = LibratoReporting.new
+librato_handler.email = '<%= @email %>'
+librato_handler.api_key = '<%= @api_key %>'
+report_handlers << librato_handler
+exception_handlers << librato_handler