Function BenchmarkZSTDReader

internal/compressions/reader_zstd_test.go:56–106

(b *testing.B)

}

func BenchmarkZSTDReader(b *testing.B) {
	var randomData = func() []byte {
		var b = strings.Builder{}
		for i := 0; i < 1024; i++ {
			b.WriteString(types.String(rands.Int64() % 10))
		}
		return []byte(b.String())
	}

	var buf = &bytes.Buffer{}
	writer, err := compressions.NewZSTDWriter(buf, 5)
	if err != nil {
		b.Fatal(err)
	}
	_, err = writer.Write(randomData())
	if err != nil {
		b.Fatal(err)
	}
	err = writer.Close()
	if err != nil {
		b.Fatal(err)
	}

	b.ResetTimer()

	for i := 0; i < b.N; i++ {
		var newBytes = make([]byte, buf.Len())
		copy(newBytes, buf.Bytes())
		reader, err := compressions.NewZSTDReader(bytes.NewReader(newBytes))
		if err != nil {
			b.Fatal(err)
		}
		var data = make([]byte, 4096)
		for {
			n, err := reader.Read(data)
			if n > 0 {
				_ = data[:n]
			}
			if err != nil {
				if err == io.EOF {
					break
				}
				b.Fatal(err)
			}
		}
		err = reader.Close()
		if err != nil {
			b.Fatal(err)
		}
	}
}