gitlab-org--gitlab-foss/lib/gitlab/database/background_migration_job.rb

# frozen_string_literal: true

module Gitlab
  module Database
    class BackgroundMigrationJob < SharedModel
      include EachBatch
      include BulkInsertSafe

      self.table_name = :background_migration_jobs

      scope :for_migration_class, -> (class_name) { where(class_name: normalize_class_name(class_name)) }
      scope :for_migration_execution, -> (class_name, arguments) do
        for_migration_class(class_name).where('arguments = ?', arguments.to_json) # rubocop:disable Rails/WhereEquals
      end

      scope :for_partitioning_migration, -> (class_name, table_name) do
        for_migration_class(class_name).where('arguments ->> 2 = ?', table_name)
      end

      enum status: {
        pending: 0,
        succeeded: 1
      }

      def self.mark_all_as_succeeded(class_name, arguments)
        self.pending.for_migration_execution(class_name, arguments)
          .update_all("status = #{statuses[:succeeded]}, updated_at = NOW()")
      end

      def self.normalize_class_name(class_name)
        return class_name unless class_name.present? && class_name.start_with?('::')

        class_name[2..]
      end

      def class_name=(value)
        write_attribute(:class_name, self.class.normalize_class_name(value))
      end
    end
  end
end