From 4298e18655cb192c31c9b916d5c3c0680bf72d18 Mon Sep 17 00:00:00 2001
From: Jonathan Rainville <rainville.jonathan@gmail.com>
Date: Wed, 20 Jun 2018 11:15:47 -0400
Subject: [PATCH] make groups of dependencyCount to do async by group

---
 lib/contracts/contracts.js      | 36 ++++++++++++++++++++++++++++++++-
 lib/contracts/deploy_manager.js | 16 +++++++++------
 2 files changed, 45 insertions(+), 7 deletions(-)

diff --git a/lib/contracts/contracts.js b/lib/contracts/contracts.js
index f53d5fd53..54023d5df 100644
--- a/lib/contracts/contracts.js
+++ b/lib/contracts/contracts.js
@@ -261,8 +261,42 @@ class ContractsManager {
           });
         }
         callback();
+      },
+      function setDependencyCount(callback) {
+        let className;
+
+        function getDependencyCount(contractName, cycleDetector) {
+          if (!self.contractDependencies[contractName] || !self.contractDependencies[contractName].length) {
+            self.contracts[contractName].dependencyCount = 0;
+            return 0;
+          }
+          if (self.contracts[contractName].dependencyCount) {
+            // Already have that count
+            return self.contracts[contractName].dependencyCount;
+          }
+          let total = self.contractDependencies[contractName].length;
+          self.contractDependencies[contractName].some(dependencyName => {
+            if (cycleDetector.indexOf(dependencyName) > -1) {
+              // We are in a cycle because of the dependency, set both to Infinity
+              self.contracts[dependencyName].dependencyCount = Infinity;
+              total = Infinity;
+              return true;
+            }
+            cycleDetector.push(dependencyName);
+            total += getDependencyCount(dependencyName, cycleDetector);
+          });
+          self.contracts[contractName].dependencyCount = total;
+          return total;
+        }
+
+        let cycleDetector;
+        for (className in self.contracts) {
+          cycleDetector = [];
+          getDependencyCount(className, cycleDetector);
+        }
+        callback();
       }
-    ], function (err, _result) {
+    ], function (err) {
       if (err) {
         self.compileError = true;
         self.events.emit("status", __("Compile/Build error"));
diff --git a/lib/contracts/deploy_manager.js b/lib/contracts/deploy_manager.js
index 7f63a4d43..b53c4b0d1 100644
--- a/lib/contracts/deploy_manager.js
+++ b/lib/contracts/deploy_manager.js
@@ -1,4 +1,5 @@
 let async = require('async');
+const _  = require('underscore');
 
 class DeployManager {
   constructor(options) {
@@ -31,12 +32,15 @@ class DeployManager {
       self.logger.info(__("deploying contracts"));
       self.events.emit("deploy:beforeAll");
 
-      async.eachOfSeries(contracts,
-        function (contract, key, callback) {
-          contract._gasLimit = self.gasLimit;
-          self.events.request('deploy:contract', contract, (err) => {
-            callback(err);
-          });
+      const contractsPerDependencyCount = _.groupBy(contracts, 'dependencyCount');
+      async.eachSeries(contractsPerDependencyCount,
+        function (parallelGroups, callback) {
+          async.each(parallelGroups, (contract, eachCb) => {
+            contract._gasLimit = self.gasLimit;
+            self.events.request('deploy:contract', contract, (err) => {
+              eachCb(err);
+            });
+          }, callback);
         },
         function (err, _results) {
           if (err) {