WIP

internal/trie/node/branch.go

@@ -33,6 +33,7 @@ type Branch struct {
 	// inserted, moved or iterated over.
 	Generation uint64
 	sync.RWMutex
+	Stats Stats
 }
 
 // NewBranch creates a new branch using the arguments given.

internal/trie/node/stats.go

@@ -0,0 +1,23 @@
+package node
+
+// Stats contains statistical information for a node.
+type Stats struct {
+	// Descendants is the number of descendant nodes for
+	// this particular node.
+	Descendants uint32
+}
+
+// GetDescendants returns the number of descendants in the branch.
+func (b *Branch) GetDescendants() (descendants uint32) {
+	return b.Stats.Descendants
+}
+
+// AddDescendants adds descendant nodes count to the node stats.
+func (b *Branch) AddDescendants(n uint32) {
+	b.Stats.Descendants += n
+}
+
+// SubDescendants subtracts descendant nodes count from the node stats.
+func (b *Branch) SubDescendants(n uint32) {
+	b.Stats.Descendants -= n
+}

internal/trie/node/stats_test.go

@@ -0,0 +1 @@
+package node