lib/longboat/jobs/slurm_job_states.rb


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
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