From cd0f6b103553f547bc21a8447e785c31f7ce858c Mon Sep 17 00:00:00 2001
From: Nat Lasseter <nat.lasseter@york.ac.uk>
Date: Mon, 2 Mar 2020 10:21:32 +0000
Subject: Use classes properly

---
 lib/longboat/jobs/slurm_job_states.rb | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100644 lib/longboat/jobs/slurm_job_states.rb

(limited to 'lib/longboat/jobs')

diff --git a/lib/longboat/jobs/slurm_job_states.rb b/lib/longboat/jobs/slurm_job_states.rb
new file mode 100644
index 0000000..74c7999
--- /dev/null
+++ b/lib/longboat/jobs/slurm_job_states.rb
@@ -0,0 +1,31 @@
+module Longboat
+  module Jobs
+    class SlurmJobStates
+      def initialize(collector)
+        @collector = collector
+        @collector.register!(self)
+      end
+
+      def run
+        start_time = (Time.now - 15 * 60).strftime("%H:%M:%S")
+        raw = `sacct -a -P -o State -S #{start_time}`.lines.map(&:strip)[1..-1]
+
+        tally = Hash.new{0}
+
+        raw.each do |state|
+          tally[state] += 1
+        end
+
+        tally.each do |state, number|
+          @collector.report!(
+            "longboat_slurm_job_states",
+            number,
+            help: "Number of jobs in each state",
+            type: "gauge",
+            labels: {state: state}
+          )
+        end
+      end
+    end
+  end
+end
-- 
cgit v1.2.1