Merge pull request ManageIQ#11857 from Fryguy/migration_helper_enhanc…

…ements

Migration helper enhancements

lib/migration_helper.rb

@@ -12,6 +12,42 @@ def sanitize_sql_for_conditions(conditions)
     DummyActiveRecordForMigrationHelper.send(:sanitize_sql_for_conditions, conditions)
   end
 
+  #
+  # Batching
+  #
+
+  def say_batch_started(count)
+    say "Processing #{count} rows", :subitem
+    @batch_total_started = Time.now.utc
+    @batch_started = Time.now.utc
+    @batch_total = count
+    @batch_count = 0
+  end
+
+  def say_batch_processed(count)
+    Thread.exclusive do
+      if count > 0
+        @batch_count += count
+
+        progress = @batch_count / @batch_total.to_f * 100
+        timing   = Time.now.utc - @batch_started
+        estimate = estimate_batch_complete(@batch_total_started, progress)
+
+        say "#{count} rows (#{"%.2f" % progress}% - #{@batch_count} total - #{"%.2f" % timing}s - ETA: #{estimate})", :subitem
+      end
+
+      @batch_started = Time.now.utc
+      @batch_count
+    end
+  end
+
+  def estimate_batch_complete(start_time, progress)
+    klass = Class.new { extend ActionView::Helpers::DateHelper }
+    estimated_end_time = start_time + (Time.now.utc - start_time) / (progress / 100.0)
+    klass.distance_of_time_in_words(Time.now.utc, estimated_end_time, :include_seconds => true)
+  end
+  private :estimate_batch_complete
+
   #
   # Triggers
   #

spec/support/migration_helper.rb

@@ -9,6 +9,7 @@ def migration_context(direction, &block)
             end
 
             around do |example|
+              Rails.logger.debug("============================================================")
               clearing_caches(&example)
             end
 
@@ -66,7 +67,9 @@ def ar_stubs
       end
 
       def migrate_under_test
+        Rails.logger.debug("========= migrate start ====================================")
         described_class.migrate(migration_direction)
+        Rails.logger.debug("========= migrate complete =================================")
       end
 
       def migration_direction