slurm_number_jobs.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

class SlurmNumberJobs
  def initialize(collector, config)
    @collector = collector
    @config = config
  end

  def raid
    raw = `squeue --format="%P,%u,%T" --noheader`
    raw = raw.lines
    raw = raw.map(&:strip)
    raw = raw.map{ |line| line.split(',') }

    tally = raw.tally

    @collector.redact!("slurm_number_jobs")

    tally.each do |labelset, number|
      @collector.report!(
        "slurm_number_jobs",
        number,
        help: "Number of jobs for a given user, partition, and state",
        type: "gauge",
        labels: {
          partition: labelset[0],
          user: labelset[1],
          state: labelset[2]
        }
      )
    end
  end
end