package main import ( "bufio" "fmt" "io/fs" "log" "os" "os/exec" "path/filepath" "strings" ) var ( basePath, _ = os.Getwd() protoFolder = filepath.Join(basePath, "rts-sim-testing-message", "protos") protocPath = filepath.Join(basePath, "rts-sim-testing-message", "protoc-23.1", "bin", "win64", "protoc") modulePrefix = "joylink.club/rts-sim-testing-service" ) func main() { //先安装以下插件 //go install google.golang.org/protobuf/cmd/protoc-gen-go@latest protoFiles := getProtoFiles() // 编译proto文件为Go文件 if err := compileProto(protoFiles); err != nil { log.Fatalf("编译proto文件失败:%v", err) } } // 获取指定文件夹下的所有proto文件的绝对路径 func getProtoFiles() []string { var protoFiles []string err := filepath.WalkDir(protoFolder, func(path string, d fs.DirEntry, err error) error { if !d.IsDir() { protoFiles = append(protoFiles, path) } return err }) if err != nil { log.Fatal("获取proto文件列表失败:", err) } return protoFiles } // 编译proto文件为Go文件 func compileProto(protoFiles []string) error { for _, fileName := range protoFiles { file, err := os.Open(fileName) if err != nil { return err } scanner := bufio.NewScanner(file) var outPath string for scanner.Scan() { text := scanner.Text() if !strings.HasPrefix(text, "option go_package") { continue } start := strings.Index(text, modulePrefix) if start < 0 { break } start += len(modulePrefix) dir := "." + text[start:len(text)-2] err := os.MkdirAll(dir, fs.ModeDir) if err != nil { panic(fmt.Sprintf("创建目录 %s 失败:%v", dir, err)) } outPath = "paths=source_relative:" + dir break } if outPath == "" { outPath = "./" } cmd := exec.Command(protocPath, "-I="+protoFolder, "--go_out="+outPath, fileName) fmt.Println(cmd.String()) cmd.Stdout = os.Stdout cmd.Stderr = os.Stderr if err := cmd.Run(); err != nil { return err } } return nil }