Fix read timing and test case

This commit is contained in:
Masaaki Goshima 2020-12-07 11:40:03 +09:00
parent 59f5713178
commit 699ab766f5
3 changed files with 11 additions and 4 deletions

View File

@ -14,6 +14,7 @@ import (
"bytes"
"compress/gzip"
"fmt"
"io"
"io/ioutil"
"os"
"strings"
@ -373,7 +374,9 @@ func BenchmarkCodeDecoder(b *testing.B) {
buf.WriteByte('\n')
buf.WriteByte('\n')
if err := dec.Decode(&r); err != nil {
b.Fatal("Decode:", err)
if err != io.EOF {
b.Fatal("Decode:", err)
}
}
}
})
@ -390,7 +393,9 @@ func BenchmarkUnicodeDecoder(b *testing.B) {
b.ResetTimer()
for i := 0; i < b.N; i++ {
if err := dec.Decode(&out); err != nil {
b.Fatal("Decode:", err)
if err != io.EOF {
b.Fatal("Decode:", err)
}
}
r.Seek(0, 0)
}
@ -414,7 +419,9 @@ func BenchmarkDecoderStream(b *testing.B) {
}
x = nil
if err := dec.Decode(&x); err != nil || x != 1.0 {
b.Fatalf("Decode: %v after %d", err, i)
if err != io.EOF {
b.Fatalf("Decode: %v after %d", err, i)
}
}
}
}

View File

@ -61,7 +61,6 @@ const (
// read data from r beyond the JSON values requested.
func NewDecoder(r io.Reader) *Decoder {
s := newStream(r)
s.read()
return &Decoder{
s: s,
}

View File

@ -26,6 +26,7 @@ func newStream(r io.Reader) *stream {
return &stream{
r: r,
bufSize: initBufSize,
buf: []byte{nul},
}
}