blob: 7e647ddd2297cae109d37bde2d02459f022ae5ec (
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
71
72
73
74
75
76
77
|
# frozen_string_literal: true
module Packages
module CleanupArtifactWorker
extend ActiveSupport::Concern
include LimitedCapacity::Worker
include Gitlab::Utils::StrongMemoize
def perform_work
return unless artifact
begin
artifact.transaction do
log_metadata(artifact)
artifact.destroy!
end
rescue StandardError => exception
unless artifact&.destroyed?
artifact&.update_column(:status, :error)
end
Gitlab::ErrorTracking.log_exception(
exception,
class: self.class.name
)
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
|