rts-sim-testing-service/protobuf/main/main.go

package main

import (
	"bufio"
	"fmt"
	"io/fs"
	"log"
	"os"
	"os/exec"
	"path/filepath"
	"strings"
)

var (
	basePath, _  = os.Getwd()
	protoFolder  = filepath.Join(basePath, "bj-rtss-message", "protos")
	protocPath   = filepath.Join(basePath, "bj-rtss-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
}