From 96537e39ae2faa4c7801423055db26b86f875431 Mon Sep 17 00:00:00 2001
From: Nat Lasseter <nat.lasseter@york.ac.uk>
Date: Mon, 2 Mar 2020 14:30:43 +0000
Subject: Broke apart job running and metric collection. Jobs are now picked up
 automagically from lib/jobs

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

(limited to 'lib/jobs')

diff --git a/lib/jobs/slurm_job_states.rb b/lib/jobs/slurm_job_states.rb
new file mode 100644
index 0000000..bc3e8a1
--- /dev/null
+++ b/lib/jobs/slurm_job_states.rb
@@ -0,0 +1,26 @@
+class SlurmJobStates
+  def initialize(collector)
+    @collector = collector
+  end
+
+  def run
+    start_time = (Time.now - 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
-- 
cgit v1.2.1