Function BenchmarkHintFileWrite

hintfile_bench_test.go:74–126

BenchmarkHintFileWrite benchmarks writing hint entries to file

(b *testing.B)

// BenchmarkHintFileWrite benchmarks writing hint entries to file
func BenchmarkHintFileWrite(b *testing.B) {
	tmpDir, err := os.MkdirTemp("", "hintfile-bench-write")
	if err != nil {
		b.Fatalf("Failed to create temp dir: %v", err)
	}
	defer func() {
		if err := os.RemoveAll(tmpDir); err != nil {
			b.Errorf("Failed to remove temp dir: %v", err)
		}
	}()

	hintPath := tmpDir + "/test.hint"

	entries := make([]*HintEntry, 1000)
	for i := 0; i < 1000; i++ {
		entries[i] = &HintEntry{
			BucketId:  1,
			KeySize:   10,
			ValueSize: 100,
			Timestamp: uint64(time.Now().UnixMilli()),
			TTL:       3600,
			Flag:      DataSetFlag,
			Status:    Committed,
			Ds:        DataStructureBTree,
			DataPos:   uint64(i * 10),
			FileID:    1,
			Key:       []byte(fmt.Sprintf("key%d", i)),
		}
	}

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		writer := &HintFileWriter{}
		err := writer.Create(hintPath)
		if err != nil {
			b.Fatalf("Failed to create hint file: %v", err)
		}

		for _, entry := range entries {
			err := writer.Write(entry)
			if err != nil {
				b.Fatalf("Failed to write hint entry: %v", err)
			}
		}

		err = writer.Close()
		if err != nil {
			b.Fatalf("Failed to close hint file: %v", err)
		}

		_ = os.Remove(hintPath)
	}
}

// BenchmarkHintFileRead benchmarks reading hint entries from file
func BenchmarkHintFileRead(b *testing.B) {