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/collector.rb             | 44 +++++++++++++++++++++++++++++++++++
 lib/longboat/jobs.rb                  | 28 ----------------------
 lib/longboat/jobs/slurm_job_states.rb | 31 ++++++++++++++++++++++++
 lib/longboat/metrics.rb               | 43 ----------------------------------
 4 files changed, 75 insertions(+), 71 deletions(-)
 create mode 100644 lib/longboat/collector.rb
 delete mode 100644 lib/longboat/jobs.rb
 create mode 100644 lib/longboat/jobs/slurm_job_states.rb
 delete mode 100644 lib/longboat/metrics.rb

(limited to 'lib/longboat')

diff --git a/lib/longboat/collector.rb b/lib/longboat/collector.rb
new file mode 100644
index 0000000..ccf103b
--- /dev/null
+++ b/lib/longboat/collector.rb
@@ -0,0 +1,44 @@
+module Longboat
+  class Collector
+    def initialize
+      @metrics = {}
+      @jobs = []
+    end
+
+    def report!(name, value, help: nil, type: nil, labels: {})
+      @metrics[name] ||= {help: help, type: type}
+      @metrics[name][labels] = value
+    end
+
+    def metrics
+      timestamp = (Time.now.to_f * 1000).to_i
+      res = ""
+      @metrics.each do |name, metric|
+        res << "#HELP #{name} #{metric[:help]}\n" unless metric[:help].nil?
+        res << "#TYPE #{name} #{metric[:type]}\n" unless metric[:type].nil?
+
+        metric.each do |labels, value|
+          next if labels == :help
+          next if labels == :type
+          labellist = []
+          labels.each do |k, v|
+            labellist << "#{k}=\"#{v}\""
+          end
+          labellist = labellist.join(",")
+          res << "#{name}{#{labellist}} #{value} #{timestamp}\n"
+        end
+      end
+      res
+    end
+
+    def register!(job)
+      @jobs << job
+    end
+
+    def collect!
+      @jobs.each do |job|
+        job.run
+      end
+    end
+  end
+end
diff --git a/lib/longboat/jobs.rb b/lib/longboat/jobs.rb
deleted file mode 100644
index eadae03..0000000
--- a/lib/longboat/jobs.rb
+++ /dev/null
@@ -1,28 +0,0 @@
-module Longboat
-  module Jobs
-    class << self
-      def collect!
-        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|
-          Longboat::Metrics.report!(
-            "longboat_slurm_job_state",
-            number,
-            help: "Number of jobs in each state",
-            type: "gauge",
-            labels: {state: state}
-          )
-        end
-      end
-    end
-
-    Longboat::Metrics.register!(self)
-  end
-end
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
diff --git a/lib/longboat/metrics.rb b/lib/longboat/metrics.rb
deleted file mode 100644
index 58b1cbe..0000000
--- a/lib/longboat/metrics.rb
+++ /dev/null
@@ -1,43 +0,0 @@
-module Longboat
-  module Metrics
-    class << self
-      def report!(name, value, help: nil, type: nil, labels: {})
-        @metrics ||= {}
-        @metrics[name] ||= {help: help, type: type}
-        @metrics[name][labels] = value
-      end
-
-      def to_s
-        timestamp = (Time.now.to_f * 1000).to_i
-        res = ""
-        (@metrics || {}).each do |name, metric|
-          res << "#HELP #{name} #{metric[:help]}\n" unless metric[:help].nil?
-          res << "#TYPE #{name} #{metric[:type]}\n" unless metric[:type].nil?
-
-          metric.each do |labels, value|
-            next if labels == :help
-            next if labels == :type
-            labellist = []
-            labels.each do |k, v|
-              labellist << "#{k}=\"#{v}\""
-            end
-            labellist = labellist.join(",")
-            res << "#{name}{#{labellist}} #{value} #{timestamp}\n"
-          end
-        end
-        res
-      end
-
-      def register!(collector)
-        @collectors ||= []
-        @collectors << collector
-      end
-
-      def collect!
-        (@collectors || []).each do |collector|
-          collector.collect!
-        end
-      end
-    end
-  end
-end
-- 
cgit v1.2.1