mirror/protobuf-go
1
0
Fork 0
mirror of https://github.com/protocolbuffers/protobuf-go.git synced 2025-03-08 19:14:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

internal/msgfmt: adjust handling of well-known types

Browse Source 
Identify well-known types by message name instead of the .proto file
that they belong to in case more are added (however unlikely).

Change-Id: I77b8880c106960b0d9a1cb304e2990e1bcddeb39
Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/235478
Reviewed-by: Herbie Ong <herbie@google.com>
This commit is contained in:
Joe Tsai 2020-05-27 16:28:31 -07:00 committed by Joe Tsai
parent e0b77db13b
commit b2f4e6269c
1 changed files with 13 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
internal/msgfmt/format.go
View File

@ -104,10 +104,10 @@ var protocmpMessageType = reflect.TypeOf(map[string]interface{}(nil))
func appendKnownMessage(b []byte, m protoreflect.Message) []byte {
md := m.Descriptor()
fds := md.Fields()
switch genid.WhichFile(md.FullName()) {
case genid.Any_file:
switch md.FullName() {
case genid.Any_message_fullname:
var msgVal protoreflect.Message
url := m.Get(fds.ByName(genid.Any_TypeUrl_field_name)).String()
url := m.Get(fds.ByNumber(genid.Any_TypeUrl_field_number)).String()
if v := reflect.ValueOf(m); v.Type().ConvertibleTo(protocmpMessageType) {
// For protocmp.Message, directly obtain the sub-message value
// which is stored in structured form, rather than as raw bytes.
@ -118,7 +118,7 @@ func appendKnownMessage(b []byte, m protoreflect.Message) []byte {
}
msgVal = v.ProtoReflect()
} else {
val := m.Get(fds.ByName(genid.Any_Value_field_name)).Bytes()
val := m.Get(fds.ByNumber(genid.Any_Value_field_number)).Bytes()
mt, err := protoregistry.GlobalTypes.FindMessageByURL(url)
if err != nil {
return nil
@ -137,9 +137,9 @@ func appendKnownMessage(b []byte, m protoreflect.Message) []byte {
b = append(b, '}')
return b
case genid.Timestamp_file:
secs := m.Get(fds.ByName(genid.Timestamp_Seconds_field_name)).Int()
nanos := m.Get(fds.ByName(genid.Timestamp_Nanos_field_name)).Int()
case genid.Timestamp_message_fullname:
secs := m.Get(fds.ByNumber(genid.Timestamp_Seconds_field_number)).Int()
nanos := m.Get(fds.ByNumber(genid.Timestamp_Nanos_field_number)).Int()
if nanos < 0 || nanos >= 1e9 {
return nil
}
@ -150,9 +150,9 @@ func appendKnownMessage(b []byte, m protoreflect.Message) []byte {
x = strings.TrimSuffix(x, ".000")
return append(b, x+"Z"...)
case genid.Duration_file:
secs := m.Get(fds.ByName(genid.Duration_Seconds_field_name)).Int()
nanos := m.Get(fds.ByName(genid.Duration_Nanos_field_name)).Int()
case genid.Duration_message_fullname:
secs := m.Get(fds.ByNumber(genid.Duration_Seconds_field_number)).Int()
nanos := m.Get(fds.ByNumber(genid.Duration_Nanos_field_number)).Int()
if nanos <= -1e9 || nanos >= 1e9 || (secs > 0 && nanos < 0) || (secs < 0 && nanos > 0) {
return nil
}
@ -161,9 +161,10 @@ func appendKnownMessage(b []byte, m protoreflect.Message) []byte {
x = strings.TrimSuffix(x, "000")
x = strings.TrimSuffix(x, ".000")
return append(b, x+"s"...)
}
case genid.Wrappers_file:
fd := fds.ByName(genid.WrapperValue_Value_field_name)
if genid.WhichFile(md.FullName()) == genid.Wrappers_file {
fd := fds.ByNumber(genid.WrapperValue_Value_field_number)
return appendValue(b, m.Get(fd), fd)
}