.github
alerting
client
config
controller
core
docs
example
jsonpath
k8s
k8stest
metric
pattern
security
storage
util
vendor
cloud.google.com
github.com
go.etcd.io
golang.org
google.golang.org
appengine
protobuf
encoding
internal
descfmt
descopts
detrand
encoding
errors
fieldsort
filedesc
filetype
flags
genid
impl
api_export.go
checkinit.go
codec_extension.go
codec_field.go
codec_gen.go
codec_map.go
codec_map_go111.go
codec_map_go112.go
codec_message.go
codec_messageset.go
codec_reflect.go
codec_tables.go
codec_unsafe.go
convert.go
convert_list.go
convert_map.go
decode.go
encode.go
enum.go
extension.go
legacy_enum.go
legacy_export.go
legacy_extension.go
legacy_file.go
legacy_message.go
merge.go
merge_gen.go
message.go
message_reflect.go
message_reflect_field.go
message_reflect_gen.go
pointer_reflect.go
pointer_unsafe.go
validate.go
weak.go
mapsort
pragma
set
strs
version
proto
reflect
runtime
types
AUTHORS
CONTRIBUTORS
LICENSE
PATENTS
gopkg.in
k8s.io
sigs.k8s.io
modules.txt
watchdog
web
.dockerignore
.gitattributes
.gitignore
Dockerfile
LICENSE.md
Makefile
README.md
config.yaml
go.mod
go.sum
main.go
75 lines
1.9 KiB
Go
75 lines
1.9 KiB
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 impl
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
pref "google.golang.org/protobuf/reflect/protoreflect"
|
|
"google.golang.org/protobuf/reflect/protoregistry"
|
|
)
|
|
|
|
// weakFields adds methods to the exported WeakFields type for internal use.
|
|
//
|
|
// The exported type is an alias to an unnamed type, so methods can't be
|
|
// defined directly on it.
|
|
type weakFields WeakFields
|
|
|
|
func (w weakFields) get(num pref.FieldNumber) (pref.ProtoMessage, bool) {
|
|
m, ok := w[int32(num)]
|
|
return m, ok
|
|
}
|
|
|
|
func (w *weakFields) set(num pref.FieldNumber, m pref.ProtoMessage) {
|
|
if *w == nil {
|
|
*w = make(weakFields)
|
|
}
|
|
(*w)[int32(num)] = m
|
|
}
|
|
|
|
func (w *weakFields) clear(num pref.FieldNumber) {
|
|
delete(*w, int32(num))
|
|
}
|
|
|
|
func (Export) HasWeak(w WeakFields, num pref.FieldNumber) bool {
|
|
_, ok := w[int32(num)]
|
|
return ok
|
|
}
|
|
|
|
func (Export) ClearWeak(w *WeakFields, num pref.FieldNumber) {
|
|
delete(*w, int32(num))
|
|
}
|
|
|
|
func (Export) GetWeak(w WeakFields, num pref.FieldNumber, name pref.FullName) pref.ProtoMessage {
|
|
if m, ok := w[int32(num)]; ok {
|
|
return m
|
|
}
|
|
mt, _ := protoregistry.GlobalTypes.FindMessageByName(name)
|
|
if mt == nil {
|
|
panic(fmt.Sprintf("message %v for weak field is not linked in", name))
|
|
}
|
|
return mt.Zero().Interface()
|
|
}
|
|
|
|
func (Export) SetWeak(w *WeakFields, num pref.FieldNumber, name pref.FullName, m pref.ProtoMessage) {
|
|
if m != nil {
|
|
mt, _ := protoregistry.GlobalTypes.FindMessageByName(name)
|
|
if mt == nil {
|
|
panic(fmt.Sprintf("message %v for weak field is not linked in", name))
|
|
}
|
|
if mt != m.ProtoReflect().Type() {
|
|
panic(fmt.Sprintf("invalid message type for weak field: got %T, want %T", m, mt.Zero().Interface()))
|
|
}
|
|
}
|
|
if m == nil || !m.ProtoReflect().IsValid() {
|
|
delete(*w, int32(num))
|
|
return
|
|
}
|
|
if *w == nil {
|
|
*w = make(weakFields)
|
|
}
|
|
(*w)[int32(num)] = m
|
|
}
|