Run multiple node diffs in parallel

Author: EdgeJ

lib/octocatalog-diff/cli.rb

@@ -11,6 +11,7 @@
 require_relative 'version'
 
 require 'logger'
+require 'parallel'
 require 'socket'
 
 module OctocatalogDiff
@@ -118,7 +119,9 @@ def self.cli(argv = ARGV, logger = Logger.new(STDERR), opts = {})
       # Compile catalogs and do catalog-diff
       node_set = options.delete(:node)
       all_diffs = []
-      node_set.each do |node|
+
+      # run multiple node diffs in parallel
+      Parallel.map(node_set) do |node|
         options[:node] = node
         catalog_diff = OctocatalogDiff::API::V1.catalog_diff(options.merge(logger: logger))
         diffs = catalog_diff.diffs
@@ -131,8 +134,8 @@ def self.cli(argv = ARGV, logger = Logger.new(STDERR), opts = {})
         all_diffs << diffs
       end
 
-      # Return the resulting diff object if requested (generally for testing) or otherwise return exit code
-      return catalog_diff if opts[:INTEGRATION]
+      # Return the resulting diff objects if requested (generally for testing) or otherwise return exit code
+      return all_diffs if opts[:INTEGRATION]
       all_diffs.any? ? EXITCODE_SUCCESS_WITH_DIFFS : EXITCODE_SUCCESS_NO_DIFFS
     end
 

octocatalog-diff.gemspec

@@ -27,6 +27,7 @@ EOF
   s.add_runtime_dependency 'diffy', '>= 3.1.0'
   s.add_runtime_dependency 'httparty', '>= 0.11.0'
   s.add_runtime_dependency 'hashdiff', '>= 0.3.0'
+  s.add_runtime_dependency 'parallel'
   s.add_runtime_dependency 'rugged', '>= 0.25.0b2'
 
   s.add_development_dependency 'rspec', '~> 3.4.0'