chore: add printHamt() for debugging

Author: rvagg

hamt_test.go

@@ -129,12 +129,12 @@ func addAndRemoveKeys(t *testing.T, keys []string, extraKeys []string, options .
 		}
 	}
 
-	fmt.Println("start flush")
-	bef := time.Now()
+	// fmt.Println("start flush")
+	// bef := time.Now()
 	if err := begn.Flush(ctx); err != nil {
 		t.Fatal(err)
 	}
-	fmt.Println("flush took: ", time.Since(bef))
+	// fmt.Println("flush took: ", time.Since(bef))
 	c, err := cs.Put(ctx, begn)
 	if err != nil {
 		t.Fatal(err)
@@ -158,6 +158,8 @@ func addAndRemoveKeys(t *testing.T, keys []string, extraKeys []string, options .
 		}
 	}
 
+	printHamt(begn)
+
 	// create second hamt by adding and deleting the extra keys
 	for i := 0; i < len(extraKeys); i++ {
 		begn.Set(ctx, extraKeys[i], randValue())
@@ -188,6 +190,37 @@ func addAndRemoveKeys(t *testing.T, keys []string, extraKeys []string, options .
 	}
 }
 
+func printHamt(hamt *Node) {
+	ctx := context.Background()
+
+	var printNode func(n *Node, depth int)
+
+	printNode = func(n *Node, depth int) {
+		c, err := n.store.Put(ctx, n)
+		if err != nil {
+			panic(err)
+		}
+		fmt.Printf("%s‣ %v:\n", strings.Repeat("  ", depth), c)
+		for _, p := range n.Pointers {
+			if p.isShard() {
+				child, err := p.loadChild(ctx, n.store, n.bitWidth, n.hash)
+				if err != nil {
+					panic(err)
+				}
+				printNode(child, depth+1)
+			} else {
+				var keys []string
+				for _, pt := range p.KVs {
+					keys = append(keys, string(pt.Key))
+				}
+				fmt.Printf("%s⇶ [ %s ]\n", strings.Repeat("  ", depth+1), strings.Join(keys, ", "))
+			}
+		}
+	}
+
+	printNode(hamt, 0)
+}
+
 func dotGraphRec(n *Node, name *int) {
 	cur := *name
 	for _, p := range n.Pointers {