Skip to content

Commit

Permalink
test: benchmem for std and sonic
Browse files Browse the repository at this point in the history
  • Loading branch information
@AsterDY
AsterDY committed Nov 8, 2023
1 parent 42f95bf commit 35e07e8
Showing 1 changed file with 143 additions and 114 deletions.
257 changes: 143 additions & 114 deletions internal/decoder/stream_test.go
View file
Original file line number Diff line number Diff line change
Expand Up @@ -198,152 +198,181 @@ func TestBuffered(t *testing.T) {
assert.Equal(t, d1.InputOffset(), d2.InputOffset()-1)
}

func BenchmarkDecodeStream_Std(b *testing.B) {
b.Run("single", func (b *testing.B) {
var str = _Single_JSON
var r1 = bytes.NewBufferString(str)
dc := json.NewDecoder(r1)
for i:=0; i<b.N; i++ {
var v1 map[string]interface{}
e := dc.Decode(&v1)
if e != nil {
b.Fatal(e)
}
r1.WriteString(str[1:])
}
})

b.Run("double", func (b *testing.B) {
var str = _Double_JSON
for i:=0; i<b.N; i++ {
var r1 = strings.NewReader(str)
var v1 map[string]interface{}
dc := json.NewDecoder(r1)
_ = dc.Decode(&v1)
if dc.More() {
_ = dc.Decode(&v1)
}
}
})
type DC interface{
Decode(interface{}) error
More() bool
}

b.Run("4x", func (b *testing.B) {
var str = _Double_JSON + strings.Repeat(" ", int(DefaultBufferSize-10)) + _Double_JSON
b.ResetTimer()
for i:=0; i<b.N; i++ {
var r1 = strings.NewReader(str)
var v1 map[string]interface{}
dc := json.NewDecoder(r1)
for dc.More() {
func benchDecodeStream(b *testing.B, dcFunc func(io.Reader)DC ) {
b.Run("single", func (b *testing.B) {
b.Run("new", func(b *testing.B) {
var str = _Single_JSON
for i:=0; i<b.N; i++ {
var r1 = bytes.NewBufferString(str)
dc := dcFunc(r1)
var v1 map[string]interface{}
e := dc.Decode(&v1)
if e != nil {
b.Fatal(e)
}
}
}
})

b.Run("halt", func (b *testing.B) {
var str = _Double_JSON
for i:=0; i<b.N; i++ {
var r1 = NewHaltReader(str, testHalts())
var v1 map[string]interface{}
dc := json.NewDecoder(r1)
_ = dc.Decode(&v1)
}
})

b.Run("small", func (b *testing.B) {
var str = _SMALL_JSON
for i:=0; i<b.N; i++ {
})
b.Run("reuse", func(b *testing.B) {
var str = _Single_JSON
var r1 = strings.NewReader(str)
var v1 interface{}
dc := json.NewDecoder(r1)
for dc.More() {
dc := dcFunc(r1)
for i:=0; i<b.N; i++ {
var v1 map[string]interface{}
e := dc.Decode(&v1)
if e != nil {
b.Fatal(e)
}
if dc.More() {
r1.Reset(str[1:])
}
}
}
})
}

// func BenchmarkDecodeError_Sonic(b *testing.B) {
// var str = `\b测试1234`
// for i:=0; i<b.N; i++ {
// var v1 map[string]interface{}
// _ = NewDecoder(str).Decode(&v1)
// }
// }

func BenchmarkDecodeStream_Sonic(b *testing.B) {
b.Run("single", func (b *testing.B) {
var str = _Single_JSON
var r1 = bytes.NewBufferString(str)
dc := NewStreamDecoder(r1)
for i:=0; i<b.N; i++ {
var v1 map[string]interface{}
e := dc.Decode(&v1)
if e != nil {
b.Fatal(e)
}
r1.WriteString(str[1:])
}
})
})

b.Run("double", func (b *testing.B) {
var str = _Double_JSON
for i:=0; i<b.N; i++ {
b.Run("new", func(b *testing.B) {
var str = _Double_JSON
for i:=0; i<b.N; i++ {
var r1 = strings.NewReader(str)
dc := dcFunc(r1)
var v1 map[string]interface{}
for dc.More() {
if e := dc.Decode(&v1); e != nil {
b.Fatal(e)
}
}
}
})
b.Run("reuse", func(b *testing.B) {
var str = _Double_JSON
var r1 = strings.NewReader(str)
var v1 map[string]interface{}
dc := NewStreamDecoder(r1)
_ = dc.Decode(&v1)
if dc.More() {
_ = dc.Decode(&v1)
dc := dcFunc(r1)
pdc, ok := dc.(*StreamDecoder)
for i:=0; i<b.N; i++ {
var v1 map[string]interface{}
for dc.More() {
e := dc.Decode(&v1)
if e != nil {
b.Fatal(e)
}
}
if ok {
pdc.err = nil
}
r1.Reset(str)
}
}
})
})

b.Run("4x", func (b *testing.B) {
var str = _Double_JSON + strings.Repeat(" ", int(DefaultBufferSize-10)) + _Double_JSON
b.ResetTimer()
for i:=0; i<b.N; i++ {
// println("loop")
b.Run("new", func(b *testing.B) {
var str = _Double_JSON + strings.Repeat(" ", int(DefaultBufferSize-10)) + _Double_JSON
b.ResetTimer()
for i:=0; i<b.N; i++ {
// println("loop")
var r1 = strings.NewReader(str)
dc := dcFunc(r1)
var v1 map[string]interface{}
for dc.More() {
e := dc.Decode(&v1)
if e != nil {
b.Fatal(e)
}
}
}
})
b.Run("reuse", func (b *testing.B) {
var str = _Double_JSON + strings.Repeat(" ", int(DefaultBufferSize-10)) + _Double_JSON
var r1 = strings.NewReader(str)
var v1 map[string]interface{}
dc := NewStreamDecoder(r1)
for dc.More() {
e := dc.Decode(&v1)
if e != nil {
b.Fatal(e)
dc := dcFunc(r1)
pdc, ok := dc.(*StreamDecoder)
b.ResetTimer()
for i:=0; i<b.N; i++ {
var v1 map[string]interface{}
for dc.More() {
e := dc.Decode(&v1)
if e != nil {
b.Fatal(e)
}
}
if ok {
pdc.err = nil
}
r1.Reset(str)
}
}
})
})

b.Run("halt", func (b *testing.B) {
var str = _Double_JSON
for i:=0; i<b.N; i++ {
b.Run("new", func(b *testing.B) {
var str = _Double_JSON
for i:=0; i<b.N; i++ {
var r1 = NewHaltReader(str, testHalts())
dc := dcFunc(r1)
var v1 map[string]interface{}
if e := dc.Decode(&v1); e != nil {
b.Fatal(e)
}
}
})
b.Run("reuse", func(b *testing.B) {
var str = _Double_JSON
var r1 = NewHaltReader(str, testHalts())
var v1 map[string]interface{}
dc := NewStreamDecoder(r1)
_ = dc.Decode(&v1)
}
dc := dcFunc(r1)
pdc, ok := dc.(*StreamDecoder)
for i:=0; i<b.N; i++ {
var v1 map[string]interface{}
e := dc.Decode(&v1)
if e != nil {
b.Fatal(e)
}
if ok {
pdc.err = nil
}
r1.Reset(str)
}
})
})

b.Run("small", func (b *testing.B) {
var str = _SMALL_JSON
for i:=0; i<b.N; i++ {
// println("one loop")
b.Run("new", func(b *testing.B) {
var str = _SMALL_JSON
for i:=0; i<b.N; i++ {
// println("one loop")
var r1 = strings.NewReader(str)
dc := dcFunc(r1)
var v1 interface{}
for dc.More() {
e := dc.Decode(&v1)
if e != nil {
b.Fatal(e)
}
}
}
})
b.Run("reuse", func(b *testing.B) {
var str = _SMALL_JSON
var r1 = strings.NewReader(str)
var v1 interface{}
dc := NewStreamDecoder(r1)
for dc.More() {
e := dc.Decode(&v1)
if e != nil {
b.Fatal(e)
dc := dcFunc(r1)
pdc, ok := dc.(*StreamDecoder)
for i:=0; i<b.N; i++ {
var v1 interface{}
for dc.More() {
e := dc.Decode(&v1)
if e != nil {
b.Fatal(e)
}
}
if ok {
pdc.err = nil
}
r1.Reset(str)
}
}
})
})
}

0 comments on commit 35e07e8

Please sign in to comment.