Add benches for the merkle package

merkle/merkle.go

@@ -0,0 +1,45 @@
+package merkle
+
+import (
+	"context"
+
+	"github.com/ava-labs/avalanchego/database"
+	"github.com/ava-labs/avalanchego/database/memdb"
+	"github.com/ava-labs/avalanchego/trace"
+	"github.com/ava-labs/avalanchego/x/merkledb"
+	"github.com/ava-labs/hypersdk/utils"
+	"github.com/ava-labs/avalanchego/ids"
+)
+
+// Generate merkle root for a set of items
+func GenerateMerkleRoot(ctx context.Context, config merkledb.Config, tracer trace.Tracer, merkleItems [][]byte, consumeBytes bool) (ids.ID, merkledb.MerkleDB, error) {
+	batchOps := make([]database.BatchOp, 0, len(merkleItems))
+
+	for _, item := range merkleItems {
+		key := utils.ToID(item)
+		batchOps = append(batchOps, database.BatchOp{
+			Key:   key[:],
+			Value: item,
+		})
+	}
+
+	db, err := merkledb.New(ctx, memdb.New(), config)
+	if err != nil {
+		return ids.Empty, nil, err
+	}
+
+	view, err := db.NewView(ctx, merkledb.ViewChanges{BatchOps: batchOps, ConsumeBytes: consumeBytes})
+	if err != nil {
+		return ids.Empty, nil, err
+	}
+	if err := view.CommitToDB(ctx); err != nil {
+		return ids.Empty, nil, err
+	}
+
+	root, err := db.GetMerkleRoot(ctx)
+	if err != nil {
+		return ids.Empty, nil, err
+	}
+
+	return root, db, nil
+}

merkle/merkle_test.go

@@ -0,0 +1,61 @@
+package merkle
+
+import (
+	"testing"
+
+	"context"
+	"crypto/rand"
+	"strconv"
+
+	"github.com/ava-labs/avalanchego/ids"
+	"github.com/ava-labs/avalanchego/trace"
+	"github.com/ava-labs/avalanchego/utils/units"
+	"github.com/ava-labs/avalanchego/x/merkledb"
+)
+
+var resRoot ids.ID
+var resDb merkledb.MerkleDB
+var resErr error
+
+func BenchmarkMerkleTxRoot(b *testing.B) {
+	for _, size := range []int{10, 100, 1000, 10000} {
+    ctx := context.TODO()
+    tracer := trace.Noop
+    merkleItems := make([][]byte, 0, size)
+    for i := 0; i < size; i++ {
+      item := make([]byte, 32)
+      _, err := rand.Read(item)
+      if err != nil {
+        b.Fatal(err)
+      }
+      merkleItems = append(merkleItems, item)
+    }
+
+    var root ids.ID
+    var db merkledb.MerkleDB
+    var err error
+
+    defaultConfig := merkledb.Config{
+      BranchFactor:              merkledb.BranchFactor16,
+      HistoryLength:             100,
+      IntermediateNodeCacheSize: units.MiB,
+      ValueNodeCacheSize:        units.MiB,
+      Tracer:                    tracer,
+    }
+
+		b.Run(strconv.Itoa(size), func(b *testing.B) {
+			for n := 0; n < b.N; n++ {
+				for i := 0; i < size; i++ {
+          root, db, err = GenerateMerkleRoot(ctx, defaultConfig, tracer, merkleItems, false)
+				}
+			}
+		})
+
+    // avoid compiler optimizations to cancel out the bench
+    resRoot = root
+    resDb = db
+    resErr = err
+	}
+
+  b.ReportAllocs()
+}