test: add benchmarks to share encoding and decoding (#83)

internal/test/factory.go

@@ -76,6 +76,18 @@ func GenerateBlobTxs(numTxs, blobsPerPfb, blobSize int) [][]byte {
 	return txs
 }
 
+func GenerateBlobs(blobSizes ...int) []*shares.Blob {
+	blobs := make([]*shares.Blob, len(blobSizes))
+	var err error
+	for i, size := range blobSizes {
+		blobs[i], err = shares.NewBlob(namespace.RandomBlobNamespace(), RandomBytes(size), shares.ShareVersionZero, nil)
+		if err != nil {
+			panic(err)
+		}
+	}
+	return blobs
+}
+
 const mockPFBExtraBytes = 329
 
 func MockPFB(blobSizes []uint32) []byte {

shares/share_benchmark_test.go

@@ -0,0 +1,57 @@
+package shares_test
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/celestiaorg/go-square/internal/test"
+	"github.com/celestiaorg/go-square/shares"
+)
+
+func BenchmarkBlobsToShares(b *testing.B) {
+	sizes := []int{256, 256 * 8, 256 * 64}
+	numBlobs := []int{1, 8, 64}
+	for _, size := range sizes {
+		for _, numBlobs := range numBlobs {
+			b.Run(fmt.Sprintf("ShareEncoding%dBlobs%dBytes", numBlobs, size), func(b *testing.B) {
+				b.ReportAllocs()
+				blobs := test.GenerateBlobs(test.Repeat(size, numBlobs)...)
+
+				b.ResetTimer()
+				for i := 0; i < b.N; i++ {
+					// Convert blob to shares
+					_, err := shares.SplitBlobs(blobs...)
+					if err != nil {
+						b.Fatal("Failed to split blob into shares:", err)
+					}
+				}
+			})
+		}
+	}
+}
+
+func BenchmarkSharesToBlobs(b *testing.B) {
+	sizes := []int{256, 256 * 8, 256 * 64}
+	numBlobs := []int{1, 8, 64}
+	for _, size := range sizes {
+		for _, numBlobs := range numBlobs {
+			b.Run(fmt.Sprintf("ShareDecoding%dBlobs%dBytes", numBlobs, size), func(b *testing.B) {
+				b.ReportAllocs()
+				blobs := test.GenerateBlobs(test.Repeat(size, numBlobs)...)
+				s, err := shares.SplitBlobs(blobs...)
+				if err != nil {
+					b.Fatal("Failed to split blob into shares:", err)
+				}
+
+				b.ResetTimer()
+				for i := 0; i < b.N; i++ {
+					// Convert shares back to blob
+					_, err := shares.ParseBlobs(s)
+					if err != nil {
+						b.Fatal("Failed to reconstruct blob from shares:", err)
+					}
+				}
+			})
+		}
+	}
+}