persister: add optional support (#85)

Co-authored-by: Dmitriy Kunitskiy <[email protected]>
lyft · May 4, 2021 · 8ed22a1 · 8ed22a1
1 parent faa4d12
commit 8ed22a1
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/persister.go b/persister.go
@@ -8,6 +8,7 @@ import (
 	"github.com/golang/protobuf/proto"
 	plugin_go "github.com/golang/protobuf/protoc-gen-go/plugin"
 	"github.com/spf13/afero"
+	"google.golang.org/protobuf/types/pluginpb"
 )
 
 type persister interface {
@@ -33,6 +34,9 @@ func (p *stdPersister) AddPostProcessor(proc ...PostProcessor) { p.procs = appen
 func (p *stdPersister) Persist(arts ...Artifact) *plugin_go.CodeGeneratorResponse {
 	resp := new(plugin_go.CodeGeneratorResponse)
 
+	supportedFeatures := uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL)
+	resp.SupportedFeatures = &supportedFeatures
+
 	for _, a := range arts {
 		switch a := a.(type) {
 		case GeneratorFile: