mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-06-10 09:50:43 +00:00
28 lines
919 B
Go
28 lines
919 B
Go
package protoext
|
|
|
|
import (
|
|
"github.com/golang/protobuf/proto"
|
|
"google.golang.org/protobuf/reflect/protoreflect"
|
|
"google.golang.org/protobuf/types/descriptorpb"
|
|
)
|
|
|
|
//go:generate go run github.com/v2fly/v2ray-core/v4/common/errors/errorgen
|
|
|
|
func GetMessageOptions(msgDesc protoreflect.MessageDescriptor) (*MessageOpt, error) {
|
|
msgOpt := msgDesc.Options().(*descriptorpb.MessageOptions)
|
|
msgOptRet, err := proto.GetExtension(msgOpt, E_MessageOpt)
|
|
if err != nil {
|
|
return nil, newError("unable to parse extension from message").Base(err)
|
|
}
|
|
return msgOptRet.(*MessageOpt), nil
|
|
}
|
|
|
|
func GetFieldOptions(fieldDesc protoreflect.FieldDescriptor) (*FieldOpt, error) {
|
|
fieldOpt := fieldDesc.Options().(*descriptorpb.FieldOptions)
|
|
msgOptRet, err := proto.GetExtension(fieldOpt, E_FieldOpt)
|
|
if err != nil {
|
|
return nil, newError("unable to parse extension from message").Base(err)
|
|
}
|
|
return msgOptRet.(*FieldOpt), nil
|
|
}
|