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.
61 lines
1.3 KiB
Go
61 lines
1.3 KiB
Go
package command
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/tal-tech/go-zero/tools/goctl/rpc/ctx"
|
|
"github.com/tal-tech/go-zero/tools/goctl/rpc/gen"
|
|
"github.com/tal-tech/go-zero/tools/goctl/util"
|
|
"github.com/urfave/cli"
|
|
)
|
|
|
|
func Rpc(c *cli.Context) error {
|
|
rpcCtx := ctx.MustCreateRpcContextFromCli(c)
|
|
generator := gen.NewDefaultRpcGenerator(rpcCtx)
|
|
rpcCtx.Must(generator.Generate())
|
|
return nil
|
|
}
|
|
|
|
func RpcTemplate(c *cli.Context) error {
|
|
out := c.String("out")
|
|
idea := c.Bool("idea")
|
|
generator := gen.NewRpcTemplate(out, idea)
|
|
generator.MustGenerate(true)
|
|
return nil
|
|
}
|
|
|
|
func RpcNew(c *cli.Context) error {
|
|
idea := c.Bool("idea")
|
|
arg := c.Args().First()
|
|
if len(arg) == 0 {
|
|
arg = "greet"
|
|
}
|
|
abs, err := filepath.Abs(arg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
_, err = os.Stat(abs)
|
|
if err != nil {
|
|
if !os.IsNotExist(err) {
|
|
return err
|
|
}
|
|
err = util.MkdirIfNotExist(abs)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
dir := filepath.Base(filepath.Clean(abs))
|
|
|
|
protoSrc := filepath.Join(abs, fmt.Sprintf("%v.proto", dir))
|
|
templateGenerator := gen.NewRpcTemplate(protoSrc, idea)
|
|
templateGenerator.MustGenerate(false)
|
|
|
|
rpcCtx := ctx.MustCreateRpcContext(protoSrc, "", "", idea)
|
|
generator := gen.NewDefaultRpcGenerator(rpcCtx)
|
|
rpcCtx.Must(generator.Generate())
|
|
return nil
|
|
}
|