module RocketJob
  module Jobs
    # Applies Retention policies to how long jobs are kept.
    #
    # Retentions are specific to each state so that for example completed
    # jobs can be cleaned up before jobs that are running.
    #
    # Only one active instance of this housekeeping job is permitted at a time.
    #
    # Example:
    #   RocketJob::Jobs::HousekeepingJob.create!
    #
    # Example, with the default values that can be modified:
    #   RocketJob::Jobs::HousekeepingJob.create!(
    #     aborted_retention:   7.days,
    #     completed_retention: 7.days,
    #     failed_retention:    14.days,
    #     paused_retention:    90.days,
    #     queued_retention:    nil
    #   )
    #
    # Example, overriding defaults and disabling removal of paused jobs:
    #   RocketJob::Jobs::HousekeepingJob.create!(
    #     aborted_retention:   1.day,
    #     completed_retention: 12.hours,
    #     failed_retention:    7.days,
    #     paused_retention:    nil
    #   )
    class HousekeepingJob < RocketJob::Job
      include RocketJob::Plugins::Cron
      include RocketJob::Plugins::Singleton

      rocket_job do |job|
        job.priority      = 50
        job.description   = 'Cleans out historical jobs'
        job.cron_schedule = '0 0 * * * America/New_York'
      end

      # Retention intervals in seconds
      # Set to nil to not
      key :aborted_retention, Integer, default: 7.days
      key :completed_retention, Integer, default: 7.days
      key :failed_retention, Integer, default: 14.days
      key :paused_retention, Integer, default: 90.days
      key :queued_retention, Integer

      def perform
        RocketJob::Job.where(state: :aborted, created_at: {'$lte' => aborted_retention.ago}).destroy_all if aborted_retention
        RocketJob::Job.where(state: :completed, created_at: {'$lte' => completed_retention.ago}).destroy_all if completed_retention
        RocketJob::Job.where(state: :failed, created_at: {'$lte' => failed_retention.ago}).destroy_all if failed_retention
        RocketJob::Job.where(state: :paused, created_at: {'$lte' => paused_retention.ago}).destroy_all if paused_retention
        RocketJob::Job.where(state: :queued, created_at: {'$lte' => queued_retention.ago}).destroy_all if queued_retention
      end

    end
  end
end