【新增】车库门PSL模型构建及相关ecs逻辑;
【改动】车库门ecs逻辑
This commit is contained in:
parent
b748c7ff3e
commit
9c420a826c
86
main.go
86
main.go
@ -1,86 +0,0 @@
|
|||||||
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/bj-rtsts-server"
|
|
||||||
)
|
|
||||||
|
|
||||||
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
|
|
||||||
}
|
|
Loading…
Reference in New Issue
Block a user