summaryrefslogtreecommitdiff
path: root/app/services/ci/runners/stale_machines_cleanup_service.rb
diff options
context:
space:
mode:
Diffstat (limited to 'app/services/ci/runners/stale_machines_cleanup_service.rb')
-rw-r--r--app/services/ci/runners/stale_machines_cleanup_service.rb33
1 files changed, 33 insertions, 0 deletions
diff --git a/app/services/ci/runners/stale_machines_cleanup_service.rb b/app/services/ci/runners/stale_machines_cleanup_service.rb
new file mode 100644
index 00000000000..3e5706d24a6
--- /dev/null
+++ b/app/services/ci/runners/stale_machines_cleanup_service.rb
@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+module Ci
+ module Runners
+ class StaleMachinesCleanupService
+ MAX_DELETIONS = 1000
+
+ def execute
+ ServiceResponse.success(payload: {
+ # the `stale` relationship can return duplicates, so we don't try to return a precise count here
+ deleted_machines: delete_stale_runner_machines > 0
+ })
+ end
+
+ private
+
+ def delete_stale_runner_machines
+ total_deleted_count = 0
+ loop do
+ sub_batch_limit = [100, MAX_DELETIONS].min
+
+ # delete_all discards part of the `stale` scope query, so we expliclitly wrap it with a SELECT as a workaround
+ deleted_count = Ci::RunnerMachine.id_in(Ci::RunnerMachine.stale.limit(sub_batch_limit)).delete_all
+ total_deleted_count += deleted_count
+
+ break if deleted_count == 0 || total_deleted_count >= MAX_DELETIONS
+ end
+
+ total_deleted_count
+ end
+ end
+ end
+end