Hunter0x7c7
2022-08-11 b8230139fb40edea387617b6accd8371e37eda58
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
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/v5/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
}