parallel count leaves

Author: omerfirmak

trie/zk_trie.go

@@ -239,10 +239,10 @@ func (t *ZkTrie) CountLeaves() uint64 {
 	if err != nil {
 		panic("CountLeaves cannot get root")
 	}
-	return t.countLeaves(root)
+	return t.countLeaves(root, 0)
 }
 
-func (t *ZkTrie) countLeaves(root *zkt.Hash) uint64 {
+func (t *ZkTrie) countLeaves(root *zkt.Hash, depth int) uint64 {
 	if root == nil {
 		return 0
 	}
@@ -255,7 +255,18 @@ func (t *ZkTrie) countLeaves(root *zkt.Hash) uint64 {
 	if rootNode.Type == zktrie.NodeTypeLeaf_New {
 		return 1
 	} else {
-		return t.countLeaves(rootNode.ChildL) + t.countLeaves(rootNode.ChildR)
+		count := make(chan uint64)
+		if depth < 5 {
+			go func() {
+				count <- t.countLeaves(rootNode.ChildL, depth+1)
+			}()
+			go func() {
+				count <- t.countLeaves(rootNode.ChildR, depth+1)
+			}()
+			return <-count + <-count
+		} else {
+			return t.countLeaves(rootNode.ChildL, depth+1) + t.countLeaves(rootNode.ChildR, depth+1)
+		}
 	}
 }