move final reporting of results to rake task

lib/broken_record/scanner.rb

@@ -5,24 +5,23 @@
 module BrokenRecord
   class Scanner
     def run(class_names)
-      classes = classes_to_validate(class_names)
+      ResultAggregator.new.tap do |aggregator|
+        classes = classes_to_validate(class_names)
 
-      BrokenRecord::Config.before_scan_callbacks.each { |callback| callback.call }
+        BrokenRecord::Config.before_scan_callbacks.each { |callback| callback.call }
 
-      jobs = BrokenRecord::Job.build_jobs(classes)
-      aggregator = ResultAggregator.new
+        jobs = BrokenRecord::Job.build_jobs(classes)
 
-      callback = proc do |_, _, result|
-        aggregator.add_result result if result.is_a? BrokenRecord::JobResult
-      end
+        callback = proc do |_, _, result|
+          aggregator.add_result result if result.is_a? BrokenRecord::JobResult
+        end
 
-      Parallel.each(jobs, :finish => callback) do |job|
-        ActiveRecord::Base.connection.reconnect!
-        BrokenRecord::Config.after_fork_callbacks.each { |callback| callback.call }
-        job.perform
+        Parallel.each(jobs, :finish => callback) do |job|
+          ActiveRecord::Base.connection.reconnect!
+          BrokenRecord::Config.after_fork_callbacks.each { |callback| callback.call }
+          job.perform
+        end
       end
-
-      aggregator.report_final_results
     end
 
     private

lib/broken_record/tasks.rb

@@ -6,6 +6,7 @@
     scanner = BrokenRecord::Scanner.new
     class_names = args[:class_name] ? [args[:class_name]] : []
     class_names += args.extras
-    scanner.run(class_names)
+    aggregator = scanner.run(class_names)
+    aggregator.report_final_results
   end
 end