为fusenapi兼容准备

This commit is contained in:
huangsimin@fusen.cn 2023-12-08 18:09:54 +08:00
parent 0a820d9c03
commit 702248cc8f
2 changed files with 15 additions and 22 deletions

View File

@ -54,10 +54,10 @@ func main() {
if len(args) == 0 || args[0] == "service" { if len(args) == 0 || args[0] == "service" {
ServiceMain() ServiceMain()
} else if args[0] == "gateway" { } else if args[0] == "client" {
GatewayMain() GenClientMain()
} else { } else {
log.Println("Invalid argument. Usage: go run main.go [service|gateway]") log.Println("Invalid argument. Usage: go run main.go [service|client]")
} }
} }
@ -104,35 +104,22 @@ go run -gcflags="-N" proto/goutils/proto_build/main.go`
// log.Println("Found proto folder at:", workerSpaceDir, ServiceNames, projectLastName, projectName) // log.Println("Found proto folder at:", workerSpaceDir, ServiceNames, projectLastName, projectName)
} }
func GatewayMain() { func GenClientMain() {
log.Println("项目目录:", workspaceDir) log.Println("项目目录:", workspaceDir)
checkProtoMessageName(workspaceDir + "/" + "proto/service") checkProtoMessageName(workspaceDir + "/" + "proto/service")
packageName := "service" packageName := "service"
ServerCodePath := "server"
genDir := "gen/go" genDir := "gen/go"
ServiceName, projectName, projectLastName := getServiceNameAndProjectName(ServerCodePath)
if ServiceName == "" {
err := os.MkdirAll("server", 0755)
if err != nil {
log.Println(err)
}
createFileWithPermNotExists("server/service_config.ini", func(f io.Writer) error {
return tpl.ExecuteTemplate(f, "service_config.tpl", nil)
})
panic("必须填写server/service_config.ini文件的项目名称与proto对应名称")
}
defer createFileNotExists("protoc.sh", 0755, func(f io.Writer) error { defer createFileNotExists("protoc.sh", 0755, func(f io.Writer) error {
shellstr := `#! /bin/bash shellstr := `#! /bin/bash
go run -gcflags="-N" proto/goutils/proto_build/main.go gateway` go run -gcflags="-N" proto/goutils/proto_build/main.go client`
f.Write([]byte(shellstr)) f.Write([]byte(shellstr))
return nil return nil
}) })
// genDir := fmt.Sprintf("gen/go/%s", PackageName)
serviceProtoDir := fmt.Sprintf("proto/%s", packageName) serviceProtoDir := fmt.Sprintf("proto/%s", packageName)
err := os.MkdirAll(genDir+"/"+packageName, 0755) err := os.MkdirAll(genDir+"/"+packageName, 0755)
@ -140,10 +127,16 @@ go run -gcflags="-N" proto/goutils/proto_build/main.go gateway`
log.Println(err) log.Println(err)
} }
projectName := ""
projectLastName := ""
ServiceName := ""
CheckGomodBasicPackage()
ExecProtoc(workspaceDir, serviceProtoDir, genDir, packageName, projectName) ExecProtoc(workspaceDir, serviceProtoDir, genDir, packageName, projectName)
ExecCreateAutoGrpc(genDir, packageName)
ExecCreateConfig(ServiceName, projectLastName) ExecCreateConfig(ServiceName, projectLastName)
ExecCreateGatewayAutoGrpc(genDir, packageName, projectLastName) ExecCreateAutoGrpc(genDir, packageName)
ExecCreateAutoLogic(workspaceDir, ServiceName, genDir, packageName, projectLastName)
} }
func ExecCreateConfig(ServiceName, ProjectName string) { func ExecCreateConfig(ServiceName, ProjectName string) {

View File

@ -9,5 +9,5 @@ func TestMain(t *testing.T) {
} }
func TestGateway(t *testing.T) { func TestGateway(t *testing.T) {
GatewayMain() GenClientMain()
} }