1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
package main
import (
benchmarkWrapper "./tmp"
googleMessage1Proto2 "./tmp/datasets/google_message1/proto2"
googleMessage1Proto3 "./tmp/datasets/google_message1/proto3"
googleMessage2 "./tmp/datasets/google_message2"
googleMessage3 "./tmp/datasets/google_message3"
googleMessage4 "./tmp/datasets/google_message4"
"errors"
"flag"
"github.com/golang/protobuf/proto"
"io/ioutil"
"os"
"testing"
)
// Data is returned by the Load function.
type Data struct {
// marshaled is a slice of marshaled protocol
// buffers. 1:1 with unmarshaled.
marshaled [][]byte
// Unmarshaled is a slice of unmarshaled protocol
// buffers. 1:1 with marshaled.
unmarshaled []proto.Message
count int
}
var data *Data
var counter int
type GetDefaultInstanceFunction func() proto.Message
var getDefaultInstance GetDefaultInstanceFunction
// This is used to getDefaultInstance for a message type.
func generateGetDefaltInstanceFunction(dataset benchmarkWrapper.BenchmarkDataset) error {
switch dataset.MessageName {
case "benchmarks.proto3.GoogleMessage1":
getDefaultInstance = func() proto.Message { return &googleMessage1Proto3.GoogleMessage1{} }
return nil
case "benchmarks.proto2.GoogleMessage1":
getDefaultInstance = func() proto.Message { return &googleMessage1Proto2.GoogleMessage1{} }
return nil
case "benchmarks.proto2.GoogleMessage2":
getDefaultInstance = func() proto.Message { return &googleMessage2.GoogleMessage2{} }
return nil
case "benchmarks.google_message3.GoogleMessage3":
getDefaultInstance = func() proto.Message { return &googleMessage3.GoogleMessage3{} }
return nil
case "benchmarks.google_message4.GoogleMessage4":
getDefaultInstance = func() proto.Message { return &googleMessage4.GoogleMessage4{} }
return nil
default:
return errors.New("Unknown message type: " + dataset.MessageName)
}
}
func TestMain(m *testing.M) {
flag.Parse()
data = new(Data)
rawData, err := ioutil.ReadFile(flag.Arg(0))
if err != nil {
panic("Couldn't find file" + flag.Arg(0))
}
var dataset benchmarkWrapper.BenchmarkDataset
if err = proto.Unmarshal(rawData, &dataset); err != nil {
panic("The raw input data can't be parse into BenchmarkDataset message.")
}
generateGetDefaltInstanceFunction(dataset)
for _, payload := range dataset.Payload {
data.marshaled = append(data.marshaled, payload)
m := getDefaultInstance()
proto.Unmarshal(payload, m)
data.unmarshaled = append(data.unmarshaled, m)
}
data.count = len(data.unmarshaled)
os.Exit(m.Run())
}
func BenchmarkUnmarshal(b *testing.B) {
b.ReportAllocs()
for i := 0; i < b.N; i++ {
payload := data.marshaled[counter%data.count]
out := getDefaultInstance()
if err := proto.Unmarshal(payload, out); err != nil {
b.Fatalf("can't unmarshal message %d %v", counter%data.count, err)
}
counter++
}
}
func BenchmarkMarshal(b *testing.B) {
b.ReportAllocs()
for i := 0; i < b.N; i++ {
m := data.unmarshaled[counter%data.count]
if _, err := proto.Marshal(m); err != nil {
b.Fatalf("can't marshal message %d %+v: %v", counter%data.count, m, err)
}
counter++
}
}
func BenchmarkSize(b *testing.B) {
b.ReportAllocs()
for i := 0; i < b.N; i++ {
proto.Size(data.unmarshaled[counter%data.count])
counter++
}
}
func BenchmarkClone(b *testing.B) {
b.ReportAllocs()
for i := 0; i < b.N; i++ {
proto.Clone(data.unmarshaled[counter%data.count])
counter++
}
}
func BenchmarkMerge(b *testing.B) {
b.ReportAllocs()
for i := 0; i < b.N; i++ {
out := getDefaultInstance()
proto.Merge(out, data.unmarshaled[counter%data.count])
counter++
}
}
|
