diff options
| -rw-r--r-- | .rubocop.yml | 5 | ||||
| -rw-r--r-- | pending_on_qos.rb | 44 |
2 files changed, 49 insertions, 0 deletions
diff --git a/.rubocop.yml b/.rubocop.yml new file mode 100644 index 0000000..48211dc --- /dev/null +++ b/.rubocop.yml @@ -0,0 +1,5 @@ +Layout/LineLength: + Max: 80 + +AllCops: + NewCops: enable diff --git a/pending_on_qos.rb b/pending_on_qos.rb new file mode 100644 index 0000000..7f7b3df --- /dev/null +++ b/pending_on_qos.rb @@ -0,0 +1,44 @@ +# frozen_string_literal: true + +# For each Viking partition, report the number of jobs pending due to QoS +# reasons. +class PendingOnQos + def initialize(collector, config) + @collector = collector + @partition_thresholds = { + nodes: 604_800, + week: 1_209_600, + month: 2_419_200, + himem: 604_800, + himem_week: 1_209_600, + gpu: 604_800, + interactive: 900, + test: 900, + preempt: 2_419_200 + } + end + + def raid + @partition_thresholds.each do |partition, threshold| + start_time = (Time.now - @partition_thresholds[:partition]) + .strftime('%Y-%m-:%d') + + squeue_cmd = [ + 'squeue', + '--format="%A,%R,%V"', + '--noheader', + '--parsable2', + "--partition=#{partition}", + '--state=PENDING' + ].join(' ') + end + + @collector.report!( + name: 'pending_on_qos', + value: 255, + help: 'Number of jobs pending for QoS reasons', + type: 'gauge', + labels: { partition: 'nodes' } + ) + end +end |