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.
103 lines
2.7 KiB
Go
103 lines
2.7 KiB
Go
4 years ago
|
package generator
|
||
|
|
||
|
import (
|
||
|
"fmt"
|
||
|
"path/filepath"
|
||
|
"strings"
|
||
|
|
||
|
"github.com/tal-tech/go-zero/core/collection"
|
||
|
"github.com/tal-tech/go-zero/tools/goctl/rpc/parser"
|
||
|
"github.com/tal-tech/go-zero/tools/goctl/util"
|
||
|
"github.com/tal-tech/go-zero/tools/goctl/util/stringx"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
serverTemplate = `{{.head}}
|
||
|
|
||
|
package server
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
|
||
|
{{.imports}}
|
||
|
)
|
||
|
|
||
|
type {{.server}}Server struct {
|
||
|
svcCtx *svc.ServiceContext
|
||
|
}
|
||
|
|
||
|
func New{{.server}}Server(svcCtx *svc.ServiceContext) *{{.server}}Server {
|
||
|
return &{{.server}}Server{
|
||
|
svcCtx: svcCtx,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
{{.funcs}}
|
||
|
`
|
||
|
functionTemplate = `
|
||
|
{{if .hasComment}}{{.comment}}{{end}}
|
||
|
func (s *{{.server}}Server) {{.method}} (ctx context.Context, in {{.request}}) ({{.response}}, error) {
|
||
|
l := logic.New{{.logicName}}(ctx,s.svcCtx)
|
||
|
return l.{{.method}}(in)
|
||
|
}
|
||
|
`
|
||
|
)
|
||
|
|
||
|
func (g *defaultGenerator) GenServer(ctx DirContext, proto parser.Proto) error {
|
||
|
dir := ctx.GetServer()
|
||
|
logicImport := fmt.Sprintf(`"%v"`, ctx.GetLogic().Package)
|
||
|
svcImport := fmt.Sprintf(`"%v"`, ctx.GetSvc().Package)
|
||
|
pbImport := fmt.Sprintf(`"%v"`, ctx.GetPb().Package)
|
||
|
|
||
|
imports := collection.NewSet()
|
||
|
imports.AddStr(logicImport, svcImport, pbImport)
|
||
|
|
||
|
head := util.GetHead(proto.Name)
|
||
|
service := proto.Service
|
||
|
serverFile := filepath.Join(dir.Filename, formatFilename(service.Name+"_server")+".go")
|
||
|
funcList, err := g.genFunctions(proto.PbPackage, service)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
text, err := util.LoadTemplate(category, serverTemplateFile, serverTemplate)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
err = util.With("server").GoFmt(true).Parse(text).SaveTo(map[string]interface{}{
|
||
|
"head": head,
|
||
|
"server": stringx.From(service.Name).Title(),
|
||
|
"imports": strings.Join(imports.KeysStr(), util.NL),
|
||
|
"funcs": strings.Join(funcList, util.NL),
|
||
|
}, serverFile, true)
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
func (g *defaultGenerator) genFunctions(goPackage string, service parser.Service) ([]string, error) {
|
||
|
var functionList []string
|
||
|
for _, rpc := range service.RPC {
|
||
|
text, err := util.LoadTemplate(category, serverFuncTemplateFile, functionTemplate)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
comment := parser.GetComment(rpc.Doc())
|
||
|
buffer, err := util.With("func").Parse(text).Execute(map[string]interface{}{
|
||
|
"server": stringx.From(service.Name).Title(),
|
||
|
"logicName": fmt.Sprintf("%sLogic", stringx.From(rpc.Name).Title()),
|
||
|
"method": parser.CamelCase(rpc.Name),
|
||
|
"request": fmt.Sprintf("*%s.%s", goPackage, parser.CamelCase(rpc.RequestType)),
|
||
|
"response": fmt.Sprintf("*%s.%s", goPackage, parser.CamelCase(rpc.ReturnsType)),
|
||
|
"hasComment": len(comment) > 0,
|
||
|
"comment": comment,
|
||
|
})
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
|
||
|
functionList = append(functionList, buffer.String())
|
||
|
}
|
||
|
return functionList, nil
|
||
|
}
|