Merge pull request #3 from k1LoW/support-proto3-optional

sorcererxw · Mar 22, 2023 · 3689db4 · 3689db4
2 parents 83631c4 + adbc182
commit 3689db4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -6,6 +6,7 @@ import (
 
 	"github.com/golang/mock/mockgen/model"
 	"google.golang.org/protobuf/compiler/protogen"
+	"google.golang.org/protobuf/types/pluginpb"
 )
 
 type methodType int
@@ -385,6 +386,7 @@ func baseServerStreamMethods() []*model.Method {
 
 func main() {
 	protogen.Options{}.Run(func(plugin *protogen.Plugin) error {
+		plugin.SupportedFeatures = uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL)
 		for path, file := range plugin.FilesByPath {
 			if !file.Generate {
 				continue