mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2025-03-25 16:43:37 +00:00
8a744307e3
messageInfo() looks like this:
func (ms *messageState) messageInfo() *MessageInfo {
mi := ms.LoadMessageInfo()
if mi == nil {
panic("invalid nil message info; this suggests memory corruption due to a race or shallow copy on the message struct")
}
return mi
}
func (ms *messageState) LoadMessageInfo() *MessageInfo {
return (*MessageInfo)(atomic.LoadPointer((*unsafe.Pointer)(unsafe.Pointer(&ms.atomicMessageInfo))))
}
Which is an atomic load and a predictable branch. On x86, this 64-bit
load is just a MOV. On other platforms, like ARM64, there's actual
atomics involved (LDAR).
Meaning, it's cheap, but not free. Eliminate redundant copies of this
(Common Subexpression Elimination).
The newly added benchmarks improve by (geomean) 2.5%:
$ benchstat pre post | head -10
goarch: amd64
cpu: AMD Ryzen Threadripper PRO 3995WX 64-Cores
│ pre │ post │
│ sec/op │ sec/op vs base │
Extension/Has/None-12 106.4n ± 2% 104.0n ± 2% -2.21% (p=0.020 n=10)
Extension/Has/Set-12 116.4n ± 1% 114.4n ± 2% -1.76% (p=0.017 n=10)
Extension/Get/None-12 184.2n ± 1% 181.0n ± 1% -1.68% (p=0.003 n=10)
Extension/Get/Set-12 144.5n ± 3% 140.7n ± 2% -2.63% (p=0.041 n=10)
Extension/Set-12 227.2n ± 2% 218.6n ± 2% -3.81% (p=0.000 n=10)
geomean 149.6n 145.9n -2.42%
I didn't test on ARM64, but the difference should be larger due to the
reduced atomics.
Change-Id: I8eebeb6f753425b743368a7f5c7be4d48537e5c3
Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/575036
Reviewed-by: Michael Stapelberg <stapelberg@google.com>
LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com>
Reviewed-by: Damien Neil <dneil@google.com>
Commit-Queue: Nicolas Hillegeer <aktau@google.com>
Auto-Submit: Nicolas Hillegeer <aktau@google.com>
272 lines
7.6 KiB
Go
272 lines
7.6 KiB
Go
// 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.
|
|
|
|
// Code generated by generate-types. DO NOT EDIT.
|
|
|
|
package impl
|
|
|
|
import (
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
"google.golang.org/protobuf/runtime/protoiface"
|
|
)
|
|
|
|
func (m *messageState) Descriptor() protoreflect.MessageDescriptor {
|
|
return m.messageInfo().Desc
|
|
}
|
|
func (m *messageState) Type() protoreflect.MessageType {
|
|
return m.messageInfo()
|
|
}
|
|
func (m *messageState) New() protoreflect.Message {
|
|
return m.messageInfo().New()
|
|
}
|
|
func (m *messageState) Interface() protoreflect.ProtoMessage {
|
|
return m.protoUnwrap().(protoreflect.ProtoMessage)
|
|
}
|
|
func (m *messageState) protoUnwrap() interface{} {
|
|
return m.pointer().AsIfaceOf(m.messageInfo().GoReflectType.Elem())
|
|
}
|
|
func (m *messageState) ProtoMethods() *protoiface.Methods {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
return &mi.methods
|
|
}
|
|
|
|
// ProtoMessageInfo is a pseudo-internal API for allowing the v1 code
|
|
// to be able to retrieve a v2 MessageInfo struct.
|
|
//
|
|
// WARNING: This method is exempt from the compatibility promise and
|
|
// may be removed in the future without warning.
|
|
func (m *messageState) ProtoMessageInfo() *MessageInfo {
|
|
return m.messageInfo()
|
|
}
|
|
|
|
func (m *messageState) Range(f func(protoreflect.FieldDescriptor, protoreflect.Value) bool) {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
for _, ri := range mi.rangeInfos {
|
|
switch ri := ri.(type) {
|
|
case *fieldInfo:
|
|
if ri.has(m.pointer()) {
|
|
if !f(ri.fieldDesc, ri.get(m.pointer())) {
|
|
return
|
|
}
|
|
}
|
|
case *oneofInfo:
|
|
if n := ri.which(m.pointer()); n > 0 {
|
|
fi := mi.fields[n]
|
|
if !f(fi.fieldDesc, fi.get(m.pointer())) {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
mi.extensionMap(m.pointer()).Range(f)
|
|
}
|
|
func (m *messageState) Has(fd protoreflect.FieldDescriptor) bool {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if fi, xt := mi.checkField(fd); fi != nil {
|
|
return fi.has(m.pointer())
|
|
} else {
|
|
return mi.extensionMap(m.pointer()).Has(xt)
|
|
}
|
|
}
|
|
func (m *messageState) Clear(fd protoreflect.FieldDescriptor) {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if fi, xt := mi.checkField(fd); fi != nil {
|
|
fi.clear(m.pointer())
|
|
} else {
|
|
mi.extensionMap(m.pointer()).Clear(xt)
|
|
}
|
|
}
|
|
func (m *messageState) Get(fd protoreflect.FieldDescriptor) protoreflect.Value {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if fi, xt := mi.checkField(fd); fi != nil {
|
|
return fi.get(m.pointer())
|
|
} else {
|
|
return mi.extensionMap(m.pointer()).Get(xt)
|
|
}
|
|
}
|
|
func (m *messageState) Set(fd protoreflect.FieldDescriptor, v protoreflect.Value) {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if fi, xt := mi.checkField(fd); fi != nil {
|
|
fi.set(m.pointer(), v)
|
|
} else {
|
|
mi.extensionMap(m.pointer()).Set(xt, v)
|
|
}
|
|
}
|
|
func (m *messageState) Mutable(fd protoreflect.FieldDescriptor) protoreflect.Value {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if fi, xt := mi.checkField(fd); fi != nil {
|
|
return fi.mutable(m.pointer())
|
|
} else {
|
|
return mi.extensionMap(m.pointer()).Mutable(xt)
|
|
}
|
|
}
|
|
func (m *messageState) NewField(fd protoreflect.FieldDescriptor) protoreflect.Value {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if fi, xt := mi.checkField(fd); fi != nil {
|
|
return fi.newField()
|
|
} else {
|
|
return xt.New()
|
|
}
|
|
}
|
|
func (m *messageState) WhichOneof(od protoreflect.OneofDescriptor) protoreflect.FieldDescriptor {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if oi := mi.oneofs[od.Name()]; oi != nil && oi.oneofDesc == od {
|
|
return od.Fields().ByNumber(oi.which(m.pointer()))
|
|
}
|
|
panic("invalid oneof descriptor " + string(od.FullName()) + " for message " + string(m.Descriptor().FullName()))
|
|
}
|
|
func (m *messageState) GetUnknown() protoreflect.RawFields {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
return mi.getUnknown(m.pointer())
|
|
}
|
|
func (m *messageState) SetUnknown(b protoreflect.RawFields) {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
mi.setUnknown(m.pointer(), b)
|
|
}
|
|
func (m *messageState) IsValid() bool {
|
|
return !m.pointer().IsNil()
|
|
}
|
|
|
|
func (m *messageReflectWrapper) Descriptor() protoreflect.MessageDescriptor {
|
|
return m.messageInfo().Desc
|
|
}
|
|
func (m *messageReflectWrapper) Type() protoreflect.MessageType {
|
|
return m.messageInfo()
|
|
}
|
|
func (m *messageReflectWrapper) New() protoreflect.Message {
|
|
return m.messageInfo().New()
|
|
}
|
|
func (m *messageReflectWrapper) Interface() protoreflect.ProtoMessage {
|
|
if m, ok := m.protoUnwrap().(protoreflect.ProtoMessage); ok {
|
|
return m
|
|
}
|
|
return (*messageIfaceWrapper)(m)
|
|
}
|
|
func (m *messageReflectWrapper) protoUnwrap() interface{} {
|
|
return m.pointer().AsIfaceOf(m.messageInfo().GoReflectType.Elem())
|
|
}
|
|
func (m *messageReflectWrapper) ProtoMethods() *protoiface.Methods {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
return &mi.methods
|
|
}
|
|
|
|
// ProtoMessageInfo is a pseudo-internal API for allowing the v1 code
|
|
// to be able to retrieve a v2 MessageInfo struct.
|
|
//
|
|
// WARNING: This method is exempt from the compatibility promise and
|
|
// may be removed in the future without warning.
|
|
func (m *messageReflectWrapper) ProtoMessageInfo() *MessageInfo {
|
|
return m.messageInfo()
|
|
}
|
|
|
|
func (m *messageReflectWrapper) Range(f func(protoreflect.FieldDescriptor, protoreflect.Value) bool) {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
for _, ri := range mi.rangeInfos {
|
|
switch ri := ri.(type) {
|
|
case *fieldInfo:
|
|
if ri.has(m.pointer()) {
|
|
if !f(ri.fieldDesc, ri.get(m.pointer())) {
|
|
return
|
|
}
|
|
}
|
|
case *oneofInfo:
|
|
if n := ri.which(m.pointer()); n > 0 {
|
|
fi := mi.fields[n]
|
|
if !f(fi.fieldDesc, fi.get(m.pointer())) {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
mi.extensionMap(m.pointer()).Range(f)
|
|
}
|
|
func (m *messageReflectWrapper) Has(fd protoreflect.FieldDescriptor) bool {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if fi, xt := mi.checkField(fd); fi != nil {
|
|
return fi.has(m.pointer())
|
|
} else {
|
|
return mi.extensionMap(m.pointer()).Has(xt)
|
|
}
|
|
}
|
|
func (m *messageReflectWrapper) Clear(fd protoreflect.FieldDescriptor) {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if fi, xt := mi.checkField(fd); fi != nil {
|
|
fi.clear(m.pointer())
|
|
} else {
|
|
mi.extensionMap(m.pointer()).Clear(xt)
|
|
}
|
|
}
|
|
func (m *messageReflectWrapper) Get(fd protoreflect.FieldDescriptor) protoreflect.Value {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if fi, xt := mi.checkField(fd); fi != nil {
|
|
return fi.get(m.pointer())
|
|
} else {
|
|
return mi.extensionMap(m.pointer()).Get(xt)
|
|
}
|
|
}
|
|
func (m *messageReflectWrapper) Set(fd protoreflect.FieldDescriptor, v protoreflect.Value) {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if fi, xt := mi.checkField(fd); fi != nil {
|
|
fi.set(m.pointer(), v)
|
|
} else {
|
|
mi.extensionMap(m.pointer()).Set(xt, v)
|
|
}
|
|
}
|
|
func (m *messageReflectWrapper) Mutable(fd protoreflect.FieldDescriptor) protoreflect.Value {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if fi, xt := mi.checkField(fd); fi != nil {
|
|
return fi.mutable(m.pointer())
|
|
} else {
|
|
return mi.extensionMap(m.pointer()).Mutable(xt)
|
|
}
|
|
}
|
|
func (m *messageReflectWrapper) NewField(fd protoreflect.FieldDescriptor) protoreflect.Value {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if fi, xt := mi.checkField(fd); fi != nil {
|
|
return fi.newField()
|
|
} else {
|
|
return xt.New()
|
|
}
|
|
}
|
|
func (m *messageReflectWrapper) WhichOneof(od protoreflect.OneofDescriptor) protoreflect.FieldDescriptor {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
if oi := mi.oneofs[od.Name()]; oi != nil && oi.oneofDesc == od {
|
|
return od.Fields().ByNumber(oi.which(m.pointer()))
|
|
}
|
|
panic("invalid oneof descriptor " + string(od.FullName()) + " for message " + string(m.Descriptor().FullName()))
|
|
}
|
|
func (m *messageReflectWrapper) GetUnknown() protoreflect.RawFields {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
return mi.getUnknown(m.pointer())
|
|
}
|
|
func (m *messageReflectWrapper) SetUnknown(b protoreflect.RawFields) {
|
|
mi := m.messageInfo()
|
|
mi.init()
|
|
mi.setUnknown(m.pointer(), b)
|
|
}
|
|
func (m *messageReflectWrapper) IsValid() bool {
|
|
return !m.pointer().IsNil()
|
|
}
|