Browse Source

Fix generate .pb.go under Windows

RPRX 5 years ago
parent
commit
3d378fc280
1 changed files with 7 additions and 2 deletions
  1. 7 2
      infra/vprotogen/main.go

+ 7 - 2
infra/vprotogen/main.go

@@ -22,6 +22,11 @@ func main() {
 	GOBIN := common.GetGOBIN()
 	GOBIN := common.GetGOBIN()
 	protoc := core.ProtocMap[runtime.GOOS]
 	protoc := core.ProtocMap[runtime.GOOS]
 
 
+	EXE := ""
+	if runtime.GOOS == "windows" {
+		EXE = ".exe"
+	}
+
 	protoFilesMap := make(map[string][]string)
 	protoFilesMap := make(map[string][]string)
 	walkErr := filepath.Walk("./", func(path string, info os.FileInfo, err error) error {
 	walkErr := filepath.Walk("./", func(path string, info os.FileInfo, err error) error {
 		if err != nil {
 		if err != nil {
@@ -50,9 +55,9 @@ func main() {
 		for _, relProtoFile := range files {
 		for _, relProtoFile := range files {
 			var args []string
 			var args []string
 			if core.ProtoFilesUsingProtocGenGoFast[relProtoFile] {
 			if core.ProtoFilesUsingProtocGenGoFast[relProtoFile] {
-				args = []string{"--gofast_out", pwd, "--plugin", "protoc-gen-gofast=" + GOBIN + "/protoc-gen-gofast"}
+				args = []string{"--gofast_out", pwd, "--plugin", "protoc-gen-gofast=" + GOBIN + "/protoc-gen-gofast" + EXE}
 			} else {
 			} else {
-				args = []string{"--go_out", pwd, "--go-grpc_out", pwd, "--plugin", "protoc-gen-go=" + GOBIN + "/protoc-gen-go", "--plugin", "protoc-gen-go-grpc=" + GOBIN + "/protoc-gen-go-grpc"}
+				args = []string{"--go_out", pwd, "--go-grpc_out", pwd, "--plugin", "protoc-gen-go=" + GOBIN + "/protoc-gen-go" + EXE, "--plugin", "protoc-gen-go-grpc=" + GOBIN + "/protoc-gen-go-grpc" + EXE}
 			}
 			}
 			args = append(args, relProtoFile)
 			args = append(args, relProtoFile)
 			cmd := exec.Command(protoc, args...)
 			cmd := exec.Command(protoc, args...)