Function BenchmarkIterator_LargeValues

iterator_bench_test.go:399–454

BenchmarkIterator_LargeValues tests iteration with large values

(b *testing.B)

// BenchmarkIterator_LargeValues tests iteration with large values
func BenchmarkIterator_LargeValues(b *testing.B) {
	dir := b.TempDir()
	opts := DefaultOptions
	opts.Dir = dir
	opts.SegmentSize = 256 * 1024 * 1024
	opts.SyncEnable = false

	db, err := Open(opts)
	if err != nil {
		b.Fatalf("Failed to open database: %v", err)
	}
	defer func() { _ = db.Close() }()

	bucket := "large_bucket"
	largeValue := make([]byte, 64*1024) // 64KB values

	err = db.Update(func(tx *Tx) error {
		if err := tx.NewKVBucket(bucket); err != nil {
			return err
		}
		for i := 0; i < 1000; i++ {
			key := testutils.GetTestBytes(i)
			if err := tx.Put(bucket, key, largeValue, Persistent); err != nil {
				return err
			}
		}
		return nil
	})
	if err != nil {
		b.Fatalf("Failed to populate database: %v", err)
	}

	b.ResetTimer()
	b.ReportAllocs()

	for i := 0; i < b.N; i++ {
		err := db.View(func(tx *Tx) error {
			iterator := NewIterator(tx, bucket, IteratorOptions{Reverse: false})
			defer iterator.Release()

			for iterator.Valid() {
				_, err := iterator.Value()
				if err != nil {
					return err
				}
				if !iterator.Next() {
					break
				}
			}
			return nil
		})
		if err != nil {
			b.Fatalf("View failed: %v", err)
		}
	}
}