fix tests, verify bench command

Author: kocubinski

v2/cmd/gen/gen.go

@@ -144,8 +144,8 @@ func treeCommand() *cobra.Command {
 	cmd := &cobra.Command{
 		Use:   "tree",
 		Short: "build and save a Tree to disk, taking generated changesets as input",
-		RunE: func(cmd *cobra.Command, args []string) error {
-			multiTree := iavl.NewMultiTree(iavl.NewTestLogger(), dbPath, iavl.TreeOptions{StateStorage: true})
+		RunE: func(_ *cobra.Command, _ []string) error {
+			multiTree := iavl.NewMultiTree(iavl.NewDebugLogger(), dbPath, iavl.DefaultTreeOptions())
 			defer func(mt *iavl.MultiTree) {
 				err := mt.Close()
 				if err != nil {

v2/cmd/root.go

@@ -1,6 +1,7 @@
 package main
 
 import (
+	"github.com/cosmos/iavl/v2/cmd/bench"
 	"github.com/cosmos/iavl/v2/cmd/gen"
 	"github.com/cosmos/iavl/v2/cmd/rollback"
 	"github.com/cosmos/iavl/v2/cmd/scan"
@@ -19,6 +20,7 @@ func RootCommand() (*cobra.Command, error) {
 		rollback.Command(),
 		scan.Command(),
 		latestCommand(),
+		bench.Command(),
 	)
 	return cmd, nil
 }

v2/logger.go

@@ -1,6 +1,9 @@
 package iavl
 
-import "log/slog"
+import (
+	"log/slog"
+	"os"
+)
 
 // Logger defines basic logger that IAVL expects.
 // It is a subset of the cosmossdk.io/core/log.Logger interface.
@@ -39,6 +42,10 @@ func NewTestLogger() Logger {
 	return &testLogger{}
 }
 
+func NewDebugLogger() Logger {
+	return slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{Level: slog.LevelDebug}))
+}
+
 type testLogger struct{}
 
 func (l *testLogger) Info(msg string, keys ...any) {

v2/multitree.go

@@ -129,6 +129,7 @@ func (mt *MultiTree) newTree(dbPath string) (*Tree, error) {
 	sql, err := NewSqliteDb(pool, defaultSqliteDbOptions(SqliteDbOptions{
 		Path:    dbPath,
 		Metrics: opts.MetricsProxy,
+		Logger:  mt.logger,
 	}))
 	if err != nil {
 		return nil, err

v2/tree_test.go

@@ -4,8 +4,6 @@ import (
 	"crypto/sha256"
 	"encoding/hex"
 	"fmt"
-	"log/slog"
-	"os"
 	"sort"
 	"testing"
 	"time"
@@ -313,8 +311,7 @@ func Test_Prune_Logic(t *testing.T) {
 
 	pool := NewNodePool()
 	tmpDir := t.TempDir()
-	logger := slog.New(slog.NewTextHandler(os.Stdout, &slog.HandlerOptions{Level: slog.LevelDebug}))
-	sql, err := NewSqliteDb(pool, SqliteDbOptions{Path: tmpDir, ShardTrees: false, Logger: logger})
+	sql, err := NewSqliteDb(pool, SqliteDbOptions{Path: tmpDir, ShardTrees: false, Logger: NewDebugLogger()})
 	require.NoError(t, err)
 	treeOpts := DefaultTreeOptions()
 	treeOpts.CheckpointInterval = 100