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.
127 lines
3.4 KiB
Go
127 lines
3.4 KiB
Go
package cli
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/zeromicro/go-zero/tools/goctl/rpc/generator"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util/console"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
|
|
)
|
|
|
|
var (
|
|
// VarStringOutput describes the output.
|
|
VarStringOutput string
|
|
// VarStringHome describes the goctl home.
|
|
VarStringHome string
|
|
// VarStringRemote describes the remote git repository.
|
|
VarStringRemote string
|
|
// VarStringBranch describes the git branch.
|
|
VarStringBranch string
|
|
// VarStringSliceGoOut describes the go output.
|
|
VarStringSliceGoOut []string
|
|
// VarStringSliceGoGRPCOut describes the grpc output.
|
|
VarStringSliceGoGRPCOut []string
|
|
// VarStringSlicePlugin describes the protoc plugin.
|
|
VarStringSlicePlugin []string
|
|
// VarStringSliceProtoPath describes the proto path.
|
|
VarStringSliceProtoPath []string
|
|
// VarStringSliceGoOpt describes the go options.
|
|
VarStringSliceGoOpt []string
|
|
// VarStringSliceGoGRPCOpt describes the grpc options.
|
|
VarStringSliceGoGRPCOpt []string
|
|
// VarStringStyle describes the style of output files.
|
|
VarStringStyle string
|
|
// VarStringZRPCOut describes the zRPC output.
|
|
VarStringZRPCOut string
|
|
// VarBoolIdea describes whether idea or not
|
|
VarBoolIdea bool
|
|
// VarBoolVerbose describes whether verbose.
|
|
VarBoolVerbose bool
|
|
)
|
|
|
|
// RPCNew is to generate rpc greet service, this greet service can speed
|
|
// up your understanding of the zrpc service structure
|
|
func RPCNew(_ *cobra.Command, args []string) error {
|
|
rpcname := args[0]
|
|
ext := filepath.Ext(rpcname)
|
|
if len(ext) > 0 {
|
|
return fmt.Errorf("unexpected ext: %s", ext)
|
|
}
|
|
style := VarStringStyle
|
|
home := VarStringHome
|
|
remote := VarStringRemote
|
|
branch := VarStringBranch
|
|
verbose := VarBoolVerbose
|
|
if len(remote) > 0 {
|
|
repo, _ := util.CloneIntoGitHome(remote, branch)
|
|
if len(repo) > 0 {
|
|
home = repo
|
|
}
|
|
}
|
|
if len(home) > 0 {
|
|
pathx.RegisterGoctlHome(home)
|
|
}
|
|
|
|
protoName := rpcname + ".proto"
|
|
filename := filepath.Join(".", rpcname, protoName)
|
|
src, err := filepath.Abs(filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = generator.ProtoTmpl(src)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var ctx generator.ZRpcContext
|
|
ctx.Src = src
|
|
ctx.GoOutput = filepath.Dir(src)
|
|
ctx.GrpcOutput = filepath.Dir(src)
|
|
ctx.IsGooglePlugin = true
|
|
ctx.Output = filepath.Dir(src)
|
|
ctx.ProtocCmd = fmt.Sprintf("protoc -I=%s %s --go_out=%s --go-grpc_out=%s", filepath.Dir(src), filepath.Base(src), filepath.Dir(src), filepath.Dir(src))
|
|
|
|
grpcOptList := VarStringSliceGoGRPCOpt
|
|
if len(grpcOptList) > 0 {
|
|
ctx.ProtocCmd += " --go-grpc_opt=" + strings.Join(grpcOptList, ",")
|
|
}
|
|
|
|
goOptList := VarStringSliceGoOpt
|
|
if len(goOptList) > 0 {
|
|
ctx.ProtocCmd += " --go_opt=" + strings.Join(goOptList, ",")
|
|
}
|
|
|
|
g := generator.NewGenerator(style, verbose)
|
|
return g.Generate(&ctx)
|
|
}
|
|
|
|
// RPCTemplate is the entry for generate rpc template
|
|
func RPCTemplate(_ *cobra.Command, _ []string) error {
|
|
console.Warning("deprecated: goctl rpc template -o is deprecated and will be removed in the future, use goctl rpc -o instead")
|
|
protoFile := VarStringOutput
|
|
home := VarStringHome
|
|
remote := VarStringRemote
|
|
branch := VarStringBranch
|
|
if len(remote) > 0 {
|
|
repo, _ := util.CloneIntoGitHome(remote, branch)
|
|
if len(repo) > 0 {
|
|
home = repo
|
|
}
|
|
}
|
|
if len(home) > 0 {
|
|
pathx.RegisterGoctlHome(home)
|
|
}
|
|
|
|
if len(protoFile) == 0 {
|
|
return errors.New("missing -o")
|
|
}
|
|
|
|
return generator.ProtoTmpl(protoFile)
|
|
}
|