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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
|
package utils
import (
"github.com/golang/protobuf/proto"
"github.com/golang/protobuf/protoc-gen-go/descriptor"
"github.com/golang/protobuf/protoc-gen-go/plugin"
"errors"
"io/ioutil"
"path/filepath"
)
// CreateGenRequest creates a codegen request from a `FileDescriptorSet`
func CreateGenRequest(fds *descriptor.FileDescriptorSet, filesToGen ...string) *plugin_go.CodeGeneratorRequest {
req := new(plugin_go.CodeGeneratorRequest)
req.ProtoFile = fds.GetFile()
for _, f := range req.GetProtoFile() {
if InStringSlice(filesToGen, f.GetName()) {
req.FileToGenerate = append(req.FileToGenerate, f.GetName())
}
}
return req
}
// FilesToGenerate iterates through the proto files in the request and returns only the ones that were requested on the
// command line. Only these protos should be generated by a codegen plugin.
func FilesToGenerate(req *plugin_go.CodeGeneratorRequest) []*descriptor.FileDescriptorProto {
protos := make([]*descriptor.FileDescriptorProto, 0)
OUTERLOOP:
for _, name := range req.GetFileToGenerate() {
for _, f := range req.GetProtoFile() {
if f.GetName() == name {
protos = append(protos, f)
continue OUTERLOOP
}
}
}
return protos
}
// LoadDescriptorSet loads a `FileDescriptorSet` from a file on disk. Such a file can be generated using the
// `--descriptor_set_out` flag with `protoc`.
//
// Example:
// protoc --descriptor_set_out=fileset.pb --include_imports --include_source_info ./booking.proto ./todo.proto
func LoadDescriptorSet(pathSegments ...string) (*descriptor.FileDescriptorSet, error) {
f, err := ioutil.ReadFile(filepath.Join(pathSegments...))
if err != nil {
return nil, err
}
set := new(descriptor.FileDescriptorSet)
if err = proto.Unmarshal(f, set); err != nil {
return nil, err
}
return set, nil
}
// FindDescriptor finds the named descriptor in the given set. Only base names are searched. The first match is
// returned, on `nil` if not found
func FindDescriptor(set *descriptor.FileDescriptorSet, name string) *descriptor.FileDescriptorProto {
for _, pf := range set.GetFile() {
if filepath.Base(pf.GetName()) == name {
return pf
}
}
return nil
}
// LoadDescriptor loads file descriptor protos from a file on disk, and returns the named proto descriptor. This is
// useful mostly for testing purposes.
func LoadDescriptor(name string, pathSegments ...string) (*descriptor.FileDescriptorProto, error) {
set, err := LoadDescriptorSet(pathSegments...)
if err != nil {
return nil, err
}
if pf := FindDescriptor(set, name); pf != nil {
return pf, nil
}
return nil, errors.New("FileDescriptor not found")
}
|
