mirror of
https://github.com/protocolbuffers/protobuf-go.git
synced 2024-12-28 18:25:46 +00:00
ef75becaa8
Fix a bug where using Message.Range alone does not clear memory references for empty, but allocated lists and maps. Thus, we iterate over every known field and explicitly call Clear. Change-Id: I9c1847d4056cf66f3199947150f3140d0783444a Reviewed-on: https://go-review.googlesource.com/c/protobuf/+/183197 Reviewed-by: Damien Neil <dneil@google.com>
32 lines
767 B
Go
32 lines
767 B
Go
// Copyright 2019 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 proto
|
|
|
|
import "google.golang.org/protobuf/reflect/protoreflect"
|
|
|
|
// Reset clears every field in the message.
|
|
func Reset(m Message) {
|
|
// TODO: Document memory aliasing guarantees.
|
|
// TODO: Add fast-path for reset?
|
|
resetMessage(m.ProtoReflect())
|
|
}
|
|
|
|
func resetMessage(m protoreflect.Message) {
|
|
// Clear all known fields.
|
|
fds := m.Descriptor().Fields()
|
|
for i := 0; i < fds.Len(); i++ {
|
|
m.Clear(fds.Get(i))
|
|
}
|
|
|
|
// Clear extension fields.
|
|
m.Range(func(fd protoreflect.FieldDescriptor, _ protoreflect.Value) bool {
|
|
m.Clear(fd)
|
|
return true
|
|
})
|
|
|
|
// Clear unknown fields.
|
|
m.SetUnknown(nil)
|
|
}
|