reactor rpc (#179)

* reactor rpc generation * update flag * update command * update command * update unit test * delete test file * optimize code * update doc * update gen pb * rename target dir * update mysql data type convert rule * add done flag * optimize req/reply parameter * optimize req/reply parameter * remove waste code * remove duplicate parameter * format code * format code * optimize naming * reactor rpcv2 to rpc * remove new line * format code * rename underline to snake * reactor getParentPackage * remove debug log * reactor background
2020-11-05 14:12:47 +08:00
parent c9ec22d5f4
commit e76f44a35b
95 changed files with 2708 additions and 3301 deletions
--- a/tools/goctl/rpc/generator/genserver.go
+++ b/tools/goctl/rpc/generator/genserver.go
@@ -0,0 +1,102 @@
+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
+}