mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2025-01-07 12:56:47 +00:00
89d49632e5
Add ExtensionField.{SetType,GetType} to hide the fact that the underlying
descriptor is actually an ExtensionDescV1.
Change-Id: I1d0595484ced0a88d2df0852a732fdf0fe9aa232
Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/180538
Reviewed-by: Damien Neil <dneil@google.com>
76 lines
2.4 KiB
Go
76 lines
2.4 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.
|
|
|
|
package impl_test
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
pimpl "google.golang.org/protobuf/internal/impl"
|
|
ptype "google.golang.org/protobuf/internal/prototype"
|
|
pref "google.golang.org/protobuf/reflect/protoreflect"
|
|
piface "google.golang.org/protobuf/runtime/protoiface"
|
|
|
|
proto2_20180125 "google.golang.org/protobuf/internal/testprotos/legacy/proto2.v1.0.0-20180125-92554152"
|
|
)
|
|
|
|
type legacyExtendedMessage struct {
|
|
XXX_unrecognized []byte
|
|
XXX_InternalExtensions map[int32]pimpl.ExtensionField
|
|
}
|
|
|
|
func (*legacyExtendedMessage) Reset() {}
|
|
func (*legacyExtendedMessage) String() string { return "" }
|
|
func (*legacyExtendedMessage) ProtoMessage() {}
|
|
func (*legacyExtendedMessage) ExtensionRangeArray() []piface.ExtensionRangeV1 {
|
|
return []piface.ExtensionRangeV1{{Start: 10000, End: 20000}}
|
|
}
|
|
|
|
func mustMakeExtensionType(x *ptype.StandaloneExtension, v interface{}) pref.ExtensionType {
|
|
xd, err := ptype.NewExtension(x)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return pimpl.LegacyExtensionTypeOf(xd, reflect.TypeOf(v))
|
|
}
|
|
|
|
var (
|
|
extParentDesc = pimpl.Export{}.MessageDescriptorOf((*legacyExtendedMessage)(nil))
|
|
extMessageV1Desc = pimpl.Export{}.MessageDescriptorOf((*proto2_20180125.Message_ChildMessage)(nil))
|
|
|
|
wantType = mustMakeExtensionType(&ptype.StandaloneExtension{
|
|
FullName: "fizz.buzz.optional_message_v1",
|
|
Number: 10007,
|
|
Cardinality: pref.Optional,
|
|
Kind: pref.MessageKind,
|
|
MessageType: extMessageV1Desc,
|
|
ExtendedType: extParentDesc,
|
|
}, (*proto2_20180125.Message_ChildMessage)(nil))
|
|
wantDesc = &piface.ExtensionDescV1{
|
|
ExtendedType: (*legacyExtendedMessage)(nil),
|
|
ExtensionType: (*proto2_20180125.Message_ChildMessage)(nil),
|
|
Field: 10007,
|
|
Name: "fizz.buzz.optional_message_v1",
|
|
Tag: "bytes,10007,opt,name=optional_message_v1",
|
|
}
|
|
)
|
|
|
|
func BenchmarkConvert(b *testing.B) {
|
|
b.ReportAllocs()
|
|
for i := 0; i < b.N; i++ {
|
|
xd := pimpl.Export{}.ExtensionDescFromType(wantType)
|
|
gotType := pimpl.Export{}.ExtensionTypeFromDesc(xd)
|
|
if gotType != wantType {
|
|
b.Fatalf("ExtensionType mismatch: got %p, want %p", gotType, wantType)
|
|
}
|
|
|
|
xt := pimpl.Export{}.ExtensionTypeFromDesc(wantDesc)
|
|
gotDesc := pimpl.Export{}.ExtensionDescFromType(xt)
|
|
if gotDesc != wantDesc {
|
|
b.Fatalf("ExtensionDesc mismatch: got %p, want %p", gotDesc, wantDesc)
|
|
}
|
|
}
|
|
}
|