blob: a01d7e8abbafcf922f713a100ecdc7971a8dca18 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
# frozen_string_literal: true
module Packages
module CleanupArtifactWorker
extend ActiveSupport::Concern
include LimitedCapacity::Worker
include Gitlab::Utils::StrongMemoize
def perform_work
return unless artifact
artifact.transaction do
log_metadata(artifact)
artifact.destroy!
rescue StandardError
unless artifact&.destroyed?
artifact&.update_column(:status, :error)
end
end
after_destroy
end
def remaining_work_count
artifacts_pending_destruction.limit(max_running_jobs + 1).count
end
private
def model
raise NotImplementedError
end
def log_metadata
raise NotImplementedError
end
def log_cleanup_item
raise NotImplementedError
end
def after_destroy
# no op
end
def artifact
strong_memoize(:artifact) do
model.transaction do
to_delete = next_item
if to_delete
to_delete.update_column(:status, :processing)
log_cleanup_item(to_delete)
end
to_delete
end
end
end
def artifacts_pending_destruction
model.pending_destruction
end
def next_item
model.next_pending_destruction(order_by: :updated_at)
end
end
end
|