app/models/good_job/setting.rb in good_job-4.7.0 vs app/models/good_job/setting.rb in good_job-4.8.0

@@ -2,10 +2,11 @@
 
 module GoodJob
   class Setting < BaseRecord
     CRON_KEYS_ENABLED = "cron_keys_enabled"
     CRON_KEYS_DISABLED = "cron_keys_disabled"
+    PAUSES = "pauses"
 
     self.table_name = 'good_job_settings'
     self.implicit_order_column = 'created_at'
 
     def self.cron_key_enabled?(key, default: true)
@@ -44,8 +45,85 @@
       disabled_setting = find_or_initialize_by(key: CRON_KEYS_DISABLED) do |record|
         record.value = []
       end
       disabled_setting.value << key unless disabled_setting.value.include?(key_string)
       disabled_setting.save!
+    end
+
+    def self.pause(queue: nil, job_class: nil, label: nil)
+      raise ArgumentError, "Must provide exactly one of queue, job_class, or label" unless [queue, job_class, label].count(&:present?) == 1
+
+      setting = find_or_initialize_by(key: PAUSES) do |record|
+        record.value = { "queues" => [], "job_classes" => [], "labels" => [] }
+      end
+
+      if queue
+        setting.value["queues"] ||= []
+        setting.value["queues"] << queue.to_s unless setting.value["queues"].include?(queue.to_s)
+      elsif job_class
+        setting.value["job_classes"] ||= []
+        setting.value["job_classes"] << job_class.to_s unless setting.value["job_classes"].include?(job_class.to_s)
+      else
+        setting.value["labels"] ||= []
+        setting.value["labels"] << label.to_s unless setting.value["labels"].include?(label.to_s)
+      end
+      setting.save!
+    end
+
+    def self.unpause(queue: nil, job_class: nil, label: nil)
+      raise ArgumentError, "Must provide exactly one of queue, job_class, or label" unless [queue, job_class, label].count(&:present?) == 1
+
+      setting = find_by(key: PAUSES)
+      return unless setting
+
+      if queue
+        return unless setting.value["queues"]&.include?(queue.to_s)
+
+        setting.value["queues"].delete(queue.to_s)
+      elsif job_class
+        return unless setting.value["job_classes"]&.include?(job_class.to_s)
+
+        setting.value["job_classes"].delete(job_class.to_s)
+      else
+        return unless setting.value["labels"]&.include?(label.to_s)
+
+        setting.value["labels"].delete(label.to_s)
+      end
+      setting.save!
+    end
+
+    def self.paused?(queue: nil, job_class: nil, label: nil)
+      raise ArgumentError, "Must provide at most one of queue, job_class, or label" if [queue, job_class, label].count(&:present?) > 1
+
+      if queue
+        queue.in? paused(:queues)
+      elsif job_class
+        job_class.in? paused(:job_classes)
+      elsif label
+        label.in? paused(:labels)
+      else
+        paused.values.any?(&:any?)
+      end
+    end
+
+    def self.paused(type = nil)
+      setting = find_by(key: PAUSES)
+      pauses = setting&.value&.deep_dup || { "queues" => [], "job_classes" => [], "labels" => [] }
+      pauses = pauses.with_indifferent_access
+
+      case type
+      when :queues
+        pauses["queues"]
+      when :job_classes
+        pauses["job_classes"]
+      when :labels
+        pauses["labels"]
+      else
+        {
+          queues: pauses["queues"] || [],
+          job_classes: pauses["job_classes"] || [],
+          labels: pauses["labels"] || [],
+        }
+      end
     end
   end
 end