quickly generating rpc demo service (#60)

* add execute files

* add protoc-osx

* add rpc generation

* add rpc generation

* add: rpc template generation

* add README.md

* format error

* reactor templatex.go

* update project.go & README.md

* add: quickly generate rpc service

tools/goctl/rpc/gen/gen.go

@@ -42,6 +42,11 @@ func (g *defaultRpcGenerator) Generate() (err error) {
 		return
 	}
 
+	err = g.initGoMod()
+	if err != nil {
+		return
+	}
+
 	err = g.genEtc()
 	if err != nil {
 		return
@@ -82,5 +87,5 @@ func (g *defaultRpcGenerator) Generate() (err error) {
 		return
 	}
 
-	return nil
+	return
 }

tools/goctl/rpc/gen/gencall.go

@@ -113,10 +113,7 @@ func (g *defaultRpcGenerator) genCall() error {
 	}
 
 	service := file.Service[0]
-	callPath, err := filepath.Abs(service.Name.Lower())
-	if err != nil {
-		return err
-	}
+	callPath := filepath.Join(g.dirM[dirTarget], service.Name.Lower())
 
 	if err = util.MkdirIfNotExist(callPath); err != nil {
 		return err
@@ -152,7 +149,7 @@ func (g *defaultRpcGenerator) genCall() error {
 	}
 
 	mockFile := filepath.Join(callPath, fmt.Sprintf("%s_mock.go", service.Name.Lower()))
-	os.Remove(mockFile)
+	_ = os.Remove(mockFile)
 	err = util.With("shared").GoFmt(true).Parse(callTemplateText).SaveTo(map[string]interface{}{
 		"name":        service.Name.Lower(),
 		"head":        head,
@@ -167,9 +164,9 @@ func (g *defaultRpcGenerator) genCall() error {
 		return err
 	}
 	// if mockgen is already installed, it will generate code of gomock for shared files
-	_, err = exec.LookPath("mockgen")
-	if mockGenInstalled {
-		execx.Run(fmt.Sprintf("go generate %s", filename), "")
+	// Deprecated: it will be removed
+	if mockGenInstalled && g.Ctx.IsInGoEnv {
+		_, _ = execx.Run(fmt.Sprintf("go generate %s", filename), "")
 	}
 
 	return nil

tools/goctl/rpc/gen/gomod.go

@@ -0,0 +1,22 @@
+package gen
+
+import (
+	"fmt"
+
+	"github.com/tal-tech/go-zero/core/logx"
+	"github.com/tal-tech/go-zero/tools/goctl/rpc/execx"
+)
+
+func (g *defaultRpcGenerator) initGoMod() error {
+	if !g.Ctx.IsInGoEnv {
+		projectDir := g.dirM[dirTarget]
+		cmd := fmt.Sprintf("go mod init %s", g.Ctx.ProjectName.Source())
+		output, err := execx.Run(fmt.Sprintf(cmd), projectDir)
+		if err != nil {
+			logx.Error(err)
+			return err
+		}
+		g.Ctx.Info(output)
+	}
+	return nil
+}

tools/goctl/rpc/gen/template.go

@@ -1,26 +1,28 @@
 package gen
 
 import (
+	"path/filepath"
+	"strings"
+
 	"github.com/tal-tech/go-zero/tools/goctl/util"
 	"github.com/tal-tech/go-zero/tools/goctl/util/console"
+	"github.com/tal-tech/go-zero/tools/goctl/util/stringx"
 )
 
 const rpcTemplateText = `syntax = "proto3";
 
-package remote;
+package {{.package}};
 
 message Request {
-  string username = 1;
-  string password = 2;
+  string ping = 1;
 }
 
 message Response {
-  string name = 1;
-  string gender = 2;
+  string pong = 1;
 }
 
-service User {
-  rpc Login(Request) returns(Response);
+service {{.serviceName}} {
+  rpc Ping(Request) returns(Response);
 }
 `
 
@@ -36,8 +38,15 @@ func NewRpcTemplate(out string, idea bool) *rpcTemplate {
 	}
 }
 
-func (r *rpcTemplate) MustGenerate() {
-	err := util.With("t").Parse(rpcTemplateText).SaveTo(nil, r.out, false)
+func (r *rpcTemplate) MustGenerate(showState bool) {
+	protoFilename := filepath.Base(r.out)
+	serviceName := stringx.From(strings.TrimSuffix(protoFilename, filepath.Ext(protoFilename)))
+	err := util.With("t").Parse(rpcTemplateText).SaveTo(map[string]string{
+		"package":     serviceName.UnTitle(),
+		"serviceName": serviceName.Title(),
+	}, r.out, false)
 	r.Must(err)
-	r.Success("Done.")
+	if showState {
+		r.Success("Done.")
+	}
 }