diff options
-rw-r--r-- | lib/longboat.rb | 1 | ||||
-rw-r--r-- | lib/longboat/jobs.rb | 28 | ||||
-rw-r--r-- | lib/longboat/metrics.rb | 43 | ||||
-rwxr-xr-x | longboat | 10 |
4 files changed, 82 insertions, 0 deletions
diff --git a/lib/longboat.rb b/lib/longboat.rb new file mode 100644 index 0000000..0459cfc --- /dev/null +++ b/lib/longboat.rb @@ -0,0 +1 @@ +require 'longboat/metrics' diff --git a/lib/longboat/jobs.rb b/lib/longboat/jobs.rb new file mode 100644 index 0000000..eadae03 --- /dev/null +++ b/lib/longboat/jobs.rb @@ -0,0 +1,28 @@ +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/metrics.rb b/lib/longboat/metrics.rb new file mode 100644 index 0000000..58b1cbe --- /dev/null +++ b/lib/longboat/metrics.rb @@ -0,0 +1,43 @@ +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 diff --git a/longboat b/longboat new file mode 100755 index 0000000..3907d87 --- /dev/null +++ b/longboat @@ -0,0 +1,10 @@ +#!/usr/bin/env ruby +$LOAD_PATH << './lib' + +require 'longboat' +require 'longboat/jobs' +#require 'longboat/server' + +Longboat::Metrics.collect! +puts Longboat::Metrics +#Longboat::Server.serve! |