io.go

package test

import (
	"bytes"
	"crypto/rand"
	"io"
	"testing"
)

func generateRandomFile() io.Reader {
	buf := &bytes.Buffer{}
	buf.Grow(10 * 1024 * 1024)
	_, _ = io.CopyN(buf, io.LimitReader(rand.Reader, 10*1024*1024), 10*1024*1024)
	return buf
}

func BenchmarkReadAll(b *testing.B) {
	randomFileBuffer := generateRandomFile()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		b, _ := io.ReadAll(randomFileBuffer)
		_ = b
	}
}

func BenchmarkCopy(b *testing.B) {
	randomFileBuffer := generateRandomFile()
	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		buf := &bytes.Buffer{}
		io.Copy(buf, randomFileBuffer)
		b := buf.Bytes()
		_ = b
	}
}