mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2025-02-21 18:40:03 +00:00
encoding: add simple benchmarks to compare V1 vs V2
name old time/op new time/op delta TextEncode-4 7.70ms ± 2% 18.57ms ± 3% +141.06% (p=0.000 n=10+9) TextDecode-4 9.62ms ± 6% 252.85ms ±10% +2529.10% (p=0.000 n=10+10) JSONEncode-4 30.3ms ± 0% 10.7ms ± 7% -64.80% (p=0.000 n=9+9) JSONDecode-4 54.1ms ± 0% 880.1ms ± 3% +1526.65% (p=0.000 n=9+10) name old alloc/op new alloc/op delta TextEncode-4 3.98MB ± 0% 28.90MB ± 0% +626.36% (p=0.000 n=10+10) TextDecode-4 2.74MB ± 0% 1158.24MB ± 0% +42114.16% (p=0.000 n=10+8) JSONEncode-4 10.3MB ± 0% 3.9MB ± 0% -61.74% (p=0.000 n=10+9) JSONDecode-4 19.0MB ± 0% 3349.9MB ± 0% +17534.57% (p=0.000 n=10+10) name old allocs/op new allocs/op delta TextEncode-4 43.9k ± 0% 73.5k ± 0% +67.53% (p=0.000 n=10+10) TextDecode-4 56.2k ± 0% 277.1k ± 0% +393.31% (p=0.000 n=8+8) JSONEncode-4 465k ± 0% 64k ± 0% -86.30% (p=0.000 n=10+10) JSONDecode-4 289k ± 0% 247k ± 0% -14.58% (p=0.000 n=10+10) Change-Id: I593a52445b6356eec1488236d5f1f1a8c4b62cb3 Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/172137 Reviewed-by: Damien Neil <dneil@google.com> Reviewed-by: Joe Tsai <thebrokentoaster@gmail.com>
This commit is contained in:
Herbie Ong
parent
a2dd228004
commit
8eba9eee0b
213
encoding/bench_test.go
Normal file
213
encoding/bench_test.go
Normal file
@ -0,0 +1,213 @@
|
||||
// Copyright 2018 The Go Authors. All rights reserved.
|
||||
// Use of this source code is governed by a BSD-style
|
||||
// license that can be found in the LICENSE file.
|
||||
|
||||
package encoding_test
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
jsonpbV1 "github.com/golang/protobuf/jsonpb"
|
||||
protoV1 "github.com/golang/protobuf/proto"
|
||||
"github.com/golang/protobuf/v2/encoding/jsonpb"
|
||||
"github.com/golang/protobuf/v2/encoding/textpb"
|
||||
pref "github.com/golang/protobuf/v2/reflect/protoreflect"
|
||||
|
||||
tpb "github.com/golang/protobuf/v2/internal/testprotos/test"
|
||||
)
|
||||
|
||||
// The results of these microbenchmarks are unlikely to correspond well
|
||||
// to real world peformance. They are mainly useful as a quick check to
|
||||
// detect unexpected regressions and for profiling specific cases.
|
||||
|
||||
var benchV1 = flag.Bool("v1", false, "benchmark the v1 implementation")
|
||||
|
||||
const (
|
||||
boolValue = true
|
||||
intValue = 1 << 30
|
||||
floatValue = 3.14159265
|
||||
strValue = "hello world"
|
||||
|
||||
maxRecurseLevel = 3
|
||||
)
|
||||
|
||||
func makeProto() *tpb.TestAllTypes {
|
||||
m := &tpb.TestAllTypes{}
|
||||
fillMessage(m.ProtoReflect(), 0)
|
||||
return m
|
||||
}
|
||||
|
||||
func fillMessage(m pref.Message, level int) {
|
||||
if level > maxRecurseLevel {
|
||||
return
|
||||
}
|
||||
|
||||
knownFields := m.KnownFields()
|
||||
fieldDescs := m.Type().Fields()
|
||||
for i := 0; i < fieldDescs.Len(); i++ {
|
||||
fd := fieldDescs.Get(i)
|
||||
num := fd.Number()
|
||||
if cardinality := fd.Cardinality(); cardinality == pref.Repeated {
|
||||
if !fd.IsMap() {
|
||||
setList(knownFields.Get(num).List(), fd, level)
|
||||
} else {
|
||||
setMap(knownFields.Get(num).Map(), fd, level)
|
||||
}
|
||||
} else {
|
||||
setScalarField(knownFields, fd, level)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func setScalarField(knownFields pref.KnownFields, fd pref.FieldDescriptor, level int) {
|
||||
num := fd.Number()
|
||||
switch fd.Kind() {
|
||||
case pref.MessageKind, pref.GroupKind:
|
||||
m := knownFields.NewMessage(num)
|
||||
fillMessage(m, level+1)
|
||||
knownFields.Set(num, pref.ValueOf(m))
|
||||
default:
|
||||
knownFields.Set(num, scalarField(fd.Kind()))
|
||||
}
|
||||
}
|
||||
|
||||
func scalarField(kind pref.Kind) pref.Value {
|
||||
switch kind {
|
||||
case pref.BoolKind:
|
||||
return pref.ValueOf(boolValue)
|
||||
|
||||
case pref.Int32Kind, pref.Sint32Kind, pref.Sfixed32Kind:
|
||||
return pref.ValueOf(int32(intValue))
|
||||
|
||||
case pref.Int64Kind, pref.Sint64Kind, pref.Sfixed64Kind:
|
||||
return pref.ValueOf(int64(intValue))
|
||||
|
||||
case pref.Uint32Kind, pref.Fixed32Kind:
|
||||
return pref.ValueOf(uint32(intValue))
|
||||
|
||||
case pref.Uint64Kind, pref.Fixed64Kind:
|
||||
return pref.ValueOf(uint64(intValue))
|
||||
|
||||
case pref.FloatKind:
|
||||
return pref.ValueOf(float32(floatValue))
|
||||
|
||||
case pref.DoubleKind:
|
||||
return pref.ValueOf(float64(floatValue))
|
||||
|
||||
case pref.BytesKind:
|
||||
return pref.ValueOf([]byte(strValue))
|
||||
|
||||
case pref.StringKind:
|
||||
return pref.ValueOf(strValue)
|
||||
|
||||
case pref.EnumKind:
|
||||
return pref.ValueOf(pref.EnumNumber(42))
|
||||
}
|
||||
|
||||
panic(fmt.Sprintf("FieldDescriptor.Kind %v is not valid", kind))
|
||||
}
|
||||
|
||||
func setList(list pref.List, fd pref.FieldDescriptor, level int) {
|
||||
switch fd.Kind() {
|
||||
case pref.MessageKind, pref.GroupKind:
|
||||
for i := 0; i < 10; i++ {
|
||||
m := list.NewMessage()
|
||||
fillMessage(m, level+1)
|
||||
list.Append(pref.ValueOf(m))
|
||||
}
|
||||
default:
|
||||
for i := 0; i < 100; i++ {
|
||||
list.Append(scalarField(fd.Kind()))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func setMap(mmap pref.Map, fd pref.FieldDescriptor, level int) {
|
||||
fields := fd.MessageType().Fields()
|
||||
keyDesc := fields.ByNumber(1)
|
||||
valDesc := fields.ByNumber(2)
|
||||
|
||||
pkey := scalarField(keyDesc.Kind())
|
||||
switch kind := valDesc.Kind(); kind {
|
||||
case pref.MessageKind, pref.GroupKind:
|
||||
m := mmap.NewMessage()
|
||||
fillMessage(m, level+1)
|
||||
mmap.Set(pkey.MapKey(), pref.ValueOf(m))
|
||||
default:
|
||||
mmap.Set(pkey.MapKey(), scalarField(kind))
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkTextEncode(b *testing.B) {
|
||||
m := makeProto()
|
||||
for i := 0; i < b.N; i++ {
|
||||
if *benchV1 {
|
||||
protoV1.MarshalTextString(m)
|
||||
} else {
|
||||
_, err := textpb.MarshalOptions{Indent: " "}.Marshal(m)
|
||||
if err != nil {
|
||||
b.Fatal(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkTextDecode(b *testing.B) {
|
||||
m := makeProto()
|
||||
in, err := textpb.MarshalOptions{Indent: " "}.Marshal(m)
|
||||
if err != nil {
|
||||
b.Fatal(err)
|
||||
}
|
||||
|
||||
for i := 0; i < b.N; i++ {
|
||||
m := &tpb.TestAllTypes{}
|
||||
var err error
|
||||
if *benchV1 {
|
||||
err = protoV1.UnmarshalText(string(in), m)
|
||||
} else {
|
||||
err = textpb.Unmarshal(m, in)
|
||||
}
|
||||
if err != nil {
|
||||
b.Fatal(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkJSONEncode(b *testing.B) {
|
||||
m := makeProto()
|
||||
for i := 0; i < b.N; i++ {
|
||||
var err error
|
||||
if *benchV1 {
|
||||
jsm := &jsonpbV1.Marshaler{Indent: " "}
|
||||
_, err = jsm.MarshalToString(m)
|
||||
} else {
|
||||
_, err = jsonpb.MarshalOptions{Indent: " "}.Marshal(m)
|
||||
}
|
||||
if err != nil {
|
||||
b.Fatal(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func BenchmarkJSONDecode(b *testing.B) {
|
||||
m := makeProto()
|
||||
out, err := jsonpb.MarshalOptions{Indent: " "}.Marshal(m)
|
||||
if err != nil {
|
||||
b.Fatal(err)
|
||||
}
|
||||
|
||||
for i := 0; i < b.N; i++ {
|
||||
m := &tpb.TestAllTypes{}
|
||||
var err error
|
||||
if *benchV1 {
|
||||
err = jsonpbV1.UnmarshalString(string(out), m)
|
||||
} else {
|
||||
err = jsonpb.Unmarshal(m, out)
|
||||
}
|
||||
if err != nil {
|
||||
b.Fatal(err)
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
x
Reference in New Issue
Block a user