add contrib/nitro_exporter.rb

1 files changed, 35 insertions, 0 deletions

diff --git a/contrib/nitro_exporter.rb b/contrib/nitro_exporter.rb
new file mode 100644
index 0000000..eae3ba8
--- /dev/null
+++ b/contrib/nitro_exporter.rb
@@ -0,0 +1,35 @@
+#!/usr/bin/ruby
+# nitro_exporter - Prometheus exporter for nitro state
+
+require 'rack'
+require 'rack/handler/thin'
+
+class NitroExporter
+  HOSTNAME = `hostname -f`.strip
+
+  def call(env)
+    r = Rack::Response.new
+    r.content_type = 'text/plain'
+
+    state = `nitroctl`
+    if $?.exitstatus != 0
+      r.write %Q|nitro_up{hostname=#{HOSTNAME.dump}} 0\n|
+    else
+      r.write %Q|nitro_up{hostname=#{HOSTNAME.dump}} 1\n|
+      state.each_line { |line|
+        case line
+        when /^([A-Z]+) (.*?) (\(pid \d+\) )?\(wstatus (\d+)\) (\d+)s$/
+          r.write %Q|nitro_state{hostname=#{HOSTNAME.dump},service=#{$2.dump},state=#{$1.dump}} #{$5}\n|
+          r.write %Q|nitro_wstatus{hostname=#{HOSTNAME.dump},service=#{$2.dump}} #{$4}\n|  if $4
+        end
+      }
+
+    end
+
+    r.finish
+  end
+end
+
+if __FILE__ == $0
+  Rack::Handler::Thin.run(NitroExporter.new, :Port => 9007, :Host => '::')
+end