Add age/expiry duration constants to `BulkImport` class (#32839)

pull/2898/head
Matt Jankowski 2024-11-12 03:57:06 -05:00 committed by GitHub
parent 897cb1803e
commit 5d9dde3ec0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 56 additions and 2 deletions

View File

@ -9,10 +9,16 @@ class Vacuum::ImportsVacuum
private private
def clean_unconfirmed_imports! def clean_unconfirmed_imports!
BulkImport.state_unconfirmed.where(created_at: ..10.minutes.ago).in_batches.delete_all BulkImport
.confirmation_missed
.in_batches
.delete_all
end end
def clean_old_imports! def clean_old_imports!
BulkImport.where(created_at: ..1.week.ago).in_batches.delete_all BulkImport
.archival_completed
.in_batches
.delete_all
end end
end end

View File

@ -21,6 +21,9 @@
class BulkImport < ApplicationRecord class BulkImport < ApplicationRecord
self.inheritance_column = false self.inheritance_column = false
ARCHIVE_PERIOD = 1.week
CONFIRM_PERIOD = 10.minutes
belongs_to :account belongs_to :account
has_many :rows, class_name: 'BulkImportRow', inverse_of: :bulk_import, dependent: :delete_all has_many :rows, class_name: 'BulkImportRow', inverse_of: :bulk_import, dependent: :delete_all
@ -42,6 +45,9 @@ class BulkImport < ApplicationRecord
validates :type, presence: true validates :type, presence: true
scope :archival_completed, -> { where(created_at: ..ARCHIVE_PERIOD.ago) }
scope :confirmation_missed, -> { state_unconfirmed.where(created_at: ..CONFIRM_PERIOD.ago) }
def self.progress!(bulk_import_id, imported: false) def self.progress!(bulk_import_id, imported: false)
# Use `increment_counter` so that the incrementation is done atomically in the database # Use `increment_counter` so that the incrementation is done atomically in the database
BulkImport.increment_counter(:processed_items, bulk_import_id) BulkImport.increment_counter(:processed_items, bulk_import_id)

View File

@ -0,0 +1,42 @@
# frozen_string_literal: true
require 'rails_helper'
RSpec.describe BulkImport do
describe 'Associations' do
it { is_expected.to belong_to(:account).required }
it { is_expected.to have_many(:rows).class_name('BulkImportRow').inverse_of(:bulk_import).dependent(:delete_all) }
end
describe 'Validations' do
subject { Fabricate.build :bulk_import }
it { is_expected.to validate_presence_of(:type) }
end
describe 'Scopes' do
describe '.archival_completed' do
let!(:old_import) { Fabricate :bulk_import, created_at: 1.month.ago }
let!(:new_import) { Fabricate :bulk_import, created_at: 1.day.ago }
it 'returns imports which have passed the archive window period' do
expect(described_class.archival_completed)
.to include(old_import)
.and not_include(new_import)
end
end
describe '.confirmation_missed' do
let!(:old_unconfirmed_import) { Fabricate :bulk_import, created_at: 1.week.ago, state: :unconfirmed }
let!(:old_scheduled_import) { Fabricate :bulk_import, created_at: 1.week.ago, state: :scheduled }
let!(:new_unconfirmed_import) { Fabricate :bulk_import, created_at: 1.minute.ago, state: :unconfirmed }
it 'returns imports which have passed the confirmation window without confirming' do
expect(described_class.confirmation_missed)
.to include(old_unconfirmed_import)
.and not_include(old_scheduled_import)
.and not_include(new_unconfirmed_import)
end
end
end
end