63 lines
2.0 KiB
Ruby
63 lines
2.0 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
class LooseForeignKeys::DeletedRecord < ApplicationRecord
|
|
PARTITION_DURATION = 1.day
|
|
|
|
include PartitionedTable
|
|
|
|
self.primary_key = :id
|
|
self.ignored_columns = %i[partition]
|
|
|
|
partitioned_by :partition, strategy: :sliding_list,
|
|
next_partition_if: -> (active_partition) do
|
|
return false if Feature.disabled?(:lfk_automatic_partition_creation, default_enabled: :yaml)
|
|
|
|
oldest_record_in_partition = LooseForeignKeys::DeletedRecord
|
|
.select(:id, :created_at)
|
|
.for_partition(active_partition)
|
|
.order(:id)
|
|
.limit(1)
|
|
.take
|
|
|
|
oldest_record_in_partition.present? && oldest_record_in_partition.created_at < PARTITION_DURATION.ago
|
|
end,
|
|
detach_partition_if: -> (partition) do
|
|
return false if Feature.disabled?(:lfk_automatic_partition_dropping, default_enabled: :yaml)
|
|
|
|
!LooseForeignKeys::DeletedRecord
|
|
.for_partition(partition)
|
|
.status_pending
|
|
.exists?
|
|
end
|
|
|
|
scope :for_table, -> (table) { where(fully_qualified_table_name: table) }
|
|
scope :for_partition, -> (partition) { where(partition: partition) }
|
|
scope :consume_order, -> { order(:partition, :consume_after, :id) }
|
|
|
|
enum status: { pending: 1, processed: 2 }, _prefix: :status
|
|
|
|
def self.load_batch_for_table(table, batch_size)
|
|
# selecting partition as partition_number to workaround the sliding partitioning column ignore
|
|
select(arel_table[Arel.star], arel_table[:partition].as('partition_number'))
|
|
.for_table(table)
|
|
.status_pending
|
|
.consume_order
|
|
.limit(batch_size)
|
|
.to_a
|
|
end
|
|
|
|
def self.mark_records_processed(all_records)
|
|
# Run a query for each partition to optimize the row lookup by primary key (partition, id)
|
|
update_count = 0
|
|
|
|
all_records.group_by(&:partition_number).each do |partition, records_within_partition|
|
|
update_count += status_pending
|
|
.for_partition(partition)
|
|
.where(id: records_within_partition.pluck(:id))
|
|
.update_all(status: :processed)
|
|
end
|
|
|
|
update_count
|
|
end
|
|
end
|