Move find_builds_from_stale_live_traces method to Ci::Build

1 files changed, 3 insertions, 3 deletions

diff --git a/app/workers/ci/rescue_stale_live_trace_worker.rb b/app/workers/ci/rescue_stale_live_trace_worker.rb
index c4a462446d4..6c8a20f64dd 100644
--- a/app/workers/ci/rescue_stale_live_trace_worker.rb
+++ b/app/workers/ci/rescue_stale_live_trace_worker.rb
@@ -7,12 +7,12 @@ module Ci
       # Archive live traces which still resides in redis or database
       # This could happen when sidekiq-jobs for archivements are lost by SIGKILL
       # Issue: https://gitlab.com/gitlab-org/gitlab-ce/issues/36791
-      Ci::BuildTraceChunk.find_builds_from_stale_live_trace do |build_ids|
-        Ci::Build.where(id: build_ids).find_each do |build|
+      Ci::Build.find_builds_from_stale_live_traces do |builds|
+        builds.each do |build|
           begin
             build.trace.archive!
           rescue => e
-            Rails.logger.info "Failed to archive stale live trace. id: #{build.id} message: #{e.message}"
+            Rails.logger.error "Failed to archive stale live trace. id: #{build.id} message: #{e.message}"
           end
         end
       end