You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
90 lines
1.3 KiB
Go
90 lines
1.3 KiB
Go
package gogen
|
|
|
|
import (
|
|
"github.com/tal-tech/go-zero/tools/goctl/rpc/ctx"
|
|
"github.com/tal-tech/go-zero/tools/goctl/rpc/parser"
|
|
)
|
|
|
|
const (
|
|
dirTarget = "dirTarget"
|
|
dirConfig = "config"
|
|
dirEtc = "etc"
|
|
dirSvc = "svc"
|
|
dirShared = "shared"
|
|
dirHandler = "handler"
|
|
dirLogic = "logic"
|
|
dirPb = "pb"
|
|
dirInternal = "internal"
|
|
fileConfig = "config.go"
|
|
fileServiceContext = "servicecontext.go"
|
|
)
|
|
|
|
type (
|
|
defaultRpcGenerator struct {
|
|
dirM map[string]string
|
|
Ctx *ctx.RpcContext
|
|
ast *parser.PbAst
|
|
}
|
|
)
|
|
|
|
func NewDefaultRpcGenerator(ctx *ctx.RpcContext) *defaultRpcGenerator {
|
|
return &defaultRpcGenerator{
|
|
Ctx: ctx,
|
|
}
|
|
}
|
|
|
|
func (g *defaultRpcGenerator) Generate() (err error) {
|
|
g.Ctx.Info("generating code...")
|
|
defer func() {
|
|
if err == nil {
|
|
g.Ctx.Success("Done.")
|
|
}
|
|
}()
|
|
err = g.createDir()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = g.genEtc()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = g.genPb()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = g.genConfig()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = g.genSvc()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = g.genLogic()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = g.genRemoteHandler()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = g.genMain()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = g.genShared()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
return nil
|
|
}
|