add benchmark
This commit is contained in:
@@ -65,6 +65,31 @@ func BenchmarkHandleFile(b *testing.B) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func BenchmarkHandleZipFile(b *testing.B) {
|
||||||
|
file, err := os.Open("testdata/example.zip.gz")
|
||||||
|
assert.Nil(b, err)
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
b.ResetTimer()
|
||||||
|
for i := 0; i < b.N; i++ {
|
||||||
|
sourceChan := make(chan *sources.Chunk, 1)
|
||||||
|
|
||||||
|
b.StartTimer()
|
||||||
|
go func() {
|
||||||
|
defer close(sourceChan)
|
||||||
|
err := HandleFile(context.Background(), file, &sources.Chunk{}, sources.ChanReporter{Ch: sourceChan})
|
||||||
|
assert.NoError(b, err)
|
||||||
|
}()
|
||||||
|
|
||||||
|
for range sourceChan {
|
||||||
|
}
|
||||||
|
b.StopTimer()
|
||||||
|
|
||||||
|
_, err = file.Seek(0, io.SeekStart)
|
||||||
|
assert.NoError(b, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestSkipArchive(t *testing.T) {
|
func TestSkipArchive(t *testing.T) {
|
||||||
file, err := os.Open("testdata/test.tgz")
|
file, err := os.Open("testdata/test.tgz")
|
||||||
assert.Nil(t, err)
|
assert.Nil(t, err)
|
||||||
|
|||||||
Reference in New Issue
Block a user