87 lines
2.0 KiB
Go
87 lines
2.0 KiB
Go
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
|
||
}
|