Reduce Sidekiq signature of scheduled tokens migration

1 files changed, 9 insertions, 1 deletions

diff --git a/lib/gitlab/background_migration/encrypt_runners_tokens.rb b/lib/gitlab/background_migration/encrypt_runners_tokens.rb
index 4647301f1a9..cb7a4c4d52e 100644
--- a/lib/gitlab/background_migration/encrypt_runners_tokens.rb
+++ b/lib/gitlab/background_migration/encrypt_runners_tokens.rb
@@ -15,6 +15,14 @@ module Gitlab
     #
     # https://gitlab.com/gitlab-org/gitlab-ce/issues/54328
     #
-    class EncryptRunnersTokens < EncryptColumns; end
+    class EncryptRunnersTokens < EncryptColumns
+      def perform(model, from, to)
+        resource = "::Gitlab::BackgroundMigration::Models::EncryptColumns::#{model.to_s.capitalize}"
+        model = resource.constantize
+        attributes = model.encrypted_attributes.keys
+
+        super(model, attributes, from, to)
+      end
+    end
   end
 end