Small refactor and fix for RuboCop

1 files changed, 4 insertions, 2 deletions

diff --git a/db/post_migrate/20171005130944_schedule_create_gpg_key_subkeys_from_gpg_keys.rb b/db/post_migrate/20171005130944_schedule_create_gpg_key_subkeys_from_gpg_keys.rb
index c8bbfbccc08..08c9ce1d3bb 100644
--- a/db/post_migrate/20171005130944_schedule_create_gpg_key_subkeys_from_gpg_keys.rb
+++ b/db/post_migrate/20171005130944_schedule_create_gpg_key_subkeys_from_gpg_keys.rb
@@ -8,11 +8,13 @@ class ScheduleCreateGpgKeySubkeysFromGpgKeys < ActiveRecord::Migration
 
   class GpgKey < ActiveRecord::Base
     self.table_name = 'gpg_keys'
+
+    include EachBatch
   end
 
   def up
-    GpgKey.select(:id).in_batches do |relation|
-      jobs = relation.pluck(:id).map do |id|
+    GpgKey.select(:id).each_batch do |gpg_keys|
+      jobs = gpg_keys.pluck(:id).map do |id|
         ['CreateGpgKeySubkeysFromGpgKeys', [id]]
       end