Merge branch '331101-expiry-notification-worker-slowness' into 'master'

Speed up SSH key expiration notification queries

See merge request gitlab-org/gitlab!61907

app/workers/ssh_keys/expired_notification_worker.rb

@@ -14,7 +14,8 @@ module SshKeys
     def perform
       return unless ::Feature.enabled?(:ssh_key_expiration_email_notification, default_enabled: :yaml)
 
-      User.with_ssh_key_expired_today.find_each do |user|
+      # rubocop:disable CodeReuse/ActiveRecord
+      User.with_ssh_key_expired_today.find_each(batch_size: 10_000) do |user|
         with_context(user: user) do
           Gitlab::AppLogger.info "#{self.class}: Notifying User #{user.id} about expired ssh key(s)"
 
@@ -22,6 +23,7 @@ module SshKeys
 
           Keys::ExpiryNotificationService.new(user, { keys: keys, expiring_soon: false }).execute
         end
+        # rubocop:enable CodeReuse/ActiveRecord
       end
     end
   end

app/workers/ssh_keys/expiring_soon_notification_worker.rb

@@ -14,7 +14,8 @@ module SshKeys
     def perform
       return unless ::Feature.enabled?(:ssh_key_expiration_email_notification, default_enabled: :yaml)
 
-      User.with_ssh_key_expiring_soon.find_each do |user|
+      # rubocop:disable CodeReuse/ActiveRecord
+      User.with_ssh_key_expiring_soon.find_each(batch_size: 10_000) do |user|
         with_context(user: user) do
           Gitlab::AppLogger.info "#{self.class}: Notifying User #{user.id} about expiring soon ssh key(s)"
 
@@ -23,6 +24,7 @@ module SshKeys
           Keys::ExpiryNotificationService.new(user, { keys: keys, expiring_soon: true }).execute
         end
       end
+      # rubocop:enable CodeReuse/ActiveRecord
     end
   end
 end