Fix text/bench_test.go

This commit is contained in:
Peter Bourgon 2015-04-07 10:48:27 +02:00
parent 0d94dc38c3
commit 20696dd841
1 changed files with 4 additions and 4 deletions

View File

@ -21,7 +21,7 @@ import (
"testing" "testing"
dto "github.com/prometheus/client_model/go" dto "github.com/prometheus/client_model/go"
"github.com/matttproud/golang_protobuf_extensions/ext" "github.com/matttproud/golang_protobuf_extensions/pbutil"
) )
// Benchmarks to show how much penalty text format parsing actually inflicts. // Benchmarks to show how much penalty text format parsing actually inflicts.
@ -101,7 +101,7 @@ func BenchmarkParseProto(b *testing.B) {
in := bytes.NewReader(data) in := bytes.NewReader(data)
for { for {
family.Reset() family.Reset()
if _, err := ext.ReadDelimited(in, family); err != nil { if _, err := pbutil.ReadDelimited(in, family); err != nil {
if err == io.EOF { if err == io.EOF {
break break
} }
@ -129,7 +129,7 @@ func BenchmarkParseProtoGzip(b *testing.B) {
} }
for { for {
family.Reset() family.Reset()
if _, err := ext.ReadDelimited(in, family); err != nil { if _, err := pbutil.ReadDelimited(in, family); err != nil {
if err == io.EOF { if err == io.EOF {
break break
} }
@ -156,7 +156,7 @@ func BenchmarkParseProtoMap(b *testing.B) {
in := bytes.NewReader(data) in := bytes.NewReader(data)
for { for {
family := &dto.MetricFamily{} family := &dto.MetricFamily{}
if _, err := ext.ReadDelimited(in, family); err != nil { if _, err := pbutil.ReadDelimited(in, family); err != nil {
if err == io.EOF { if err == io.EOF {
break break
} }