Merge pull request #179 from sevennt/feature/go-errors-add-flag

feat: add -file flag to protoc-gen-go-errors
gotomicro · Sep 6, 2021 · b6e2ece · b6e2ece
2 parents 38b4edd + 8b4b31f
commit b6e2ece
Show file tree

Hide file tree

Showing 5 changed files with 43 additions and 38 deletions.
diff --git a/cmd/protoc-gen-go-errors/main.go b/cmd/protoc-gen-go-errors/main.go
@@ -25,6 +25,7 @@ package main
 import (
 	"flag"
 	"fmt"
+	"path/filepath"
 
 	"google.golang.org/protobuf/compiler/protogen"
 	"google.golang.org/protobuf/types/pluginpb"
@@ -41,12 +42,16 @@ func main() {
 		return
 	}
 	var flags flag.FlagSet
+	file := flags.String("file", "", "specified proto files (default value: errors.proto)")
 	protogen.Options{ParamFunc: flags.Set}.Run(func(gen *protogen.Plugin) error {
 		gen.SupportedFeatures = uint64(pluginpb.CodeGeneratorResponse_FEATURE_PROTO3_OPTIONAL)
 		for _, f := range gen.Files {
 			if !f.Generate {
 				continue
 			}
+			if filepath.Base(f.Proto.GetName()) != *file {
+				continue
+			}
 			generateFile(gen, f)
 		}
 		return nil

diff --git a/examples/grpc/errors/error.sh b/examples/grpc/errors/error.sh
@@ -4,4 +4,4 @@
 rm -rf ./protoc-gen-go-errors
 go build -o ./protoc-gen-go-errors ../../../cmd/protoc-gen-go-errors
 export PATH=$PATH:$(pwd)/
-protoc --proto_path=. --go_out=paths=source_relative:. --go-errors_out=paths=source_relative:. ./myerr.proto
+protoc --proto_path=. --go_out=paths=source_relative:. --go-errors_out=paths=source_relative,file=errors.proto:. ./errors.proto
diff --git a/examples/grpc/errors/myerr.pb.go → examples/grpc/errors/errors.pb.go b/examples/grpc/errors/myerr.pb.go → examples/grpc/errors/errors.pb.go
diff --git a/examples/grpc/errors/myerr.proto → examples/grpc/errors/errors.proto b/examples/grpc/errors/myerr.proto → examples/grpc/errors/errors.proto
diff --git a/examples/grpc/errors/myerr_errors.pb.go → examples/grpc/errors/errors_errors.pb.go b/examples/grpc/errors/myerr_errors.pb.go → examples/grpc/errors/errors_errors.pb.go