通过 shell 生成针对 model 的 rpc 结构定义

Makefile

@@ -19,3 +19,5 @@ lint: ## lint go install github.com/golangci/golangci-lint/cmd/golangci-lint@lat
 
 help: ## Show this help
 	@egrep '^[a-zA-Z_-]+:.*?## .*$$' $(MAKEFILE_LIST) | sort | sed 's/Makefile://' | awk 'BEGIN {FS = ":.*##"; printf "\nUsage:\n  make \033[36m<target>\033[0m\n\nTargets:\n"} /^[a-z0-9A-Z_-]+:.*?##/ { printf "  \033[36m%-30s\033[0m %s\n", $$1, $$2 }'
+
+# make build && ./dist/cli_darwin_arm64/codegen gen -i ../example/example.proto -d add-rpc -m user -a add:,admin

README.md

@@ -39,6 +39,8 @@
 	- go.mod
 	- go.sum
 	- .editorconfig
+    - orm_gen.go(gorm 配套的接口协议的生成)
+    - table_gen.go(表名的生成)
 	- state
 		- table.go(数据库相关的内容)
         - cache.go(缓存相关的内容)
@@ -59,7 +61,7 @@
 
 4. 执行命令
    ```bash
-	codegen gen pb -i <proto文件路径>
+	codegen gen pb <action> -i <proto文件路径>
    ```
 
 ### 注意事项

cli/gen_add_rpc.go

@@ -0,0 +1,94 @@
+package cli
+
+import (
+	"errors"
+	"github.com/lazygophers/codegen/codegen"
+	"github.com/lazygophers/log"
+	"github.com/lazygophers/utils/stringx"
+	"github.com/pterm/pterm"
+	"github.com/spf13/cobra"
+	"strings"
+)
+
+var genAddRpcCmd = &cobra.Command{
+	Use:   "add-rpc",
+	Short: "add rpc to proto with model",
+	RunE: func(cmd *cobra.Command, args []string) (err error) {
+		opt := codegen.NewAddRpcOption()
+
+		if v, err := cmd.Flags().GetString("default-role"); err == nil {
+			opt.DefaultRole = v
+		}
+
+		if v, err := cmd.Flags().GetString("gen-to"); err == nil {
+			opt.GenTo = v
+		}
+
+		var msg *codegen.PbMessage
+		if v, err := cmd.Flags().GetString("model"); err == nil && v != "" {
+			// 先看一下加了 Model 的 时候存在
+			{
+				model := stringx.ToSnake(v)
+				model = strings.TrimPrefix(model, "model_")
+				model = "model_" + model
+				model = stringx.ToCamel(model)
+
+				msg = pb.GetMessage(model)
+				if msg != nil {
+					opt.Model = model
+				}
+			}
+
+			// 直接找名字的
+			if msg == nil {
+				msg = pb.GetMessage(v)
+				if msg != nil {
+					opt.Model = v
+				}
+			}
+
+			if msg == nil {
+				log.Errorf("not found model:%v", v)
+				pterm.Error.Printfln("not found model:%v", v)
+				return errors.New("not found model")
+			}
+		} else {
+			log.Errorf("missed argument `model`")
+			pterm.Error.Printfln("missing argument `model`")
+			return errors.New("missing argument `model`")
+		}
+
+		if v, err := cmd.Flags().GetString("action"); err == nil && v != "" {
+			opt.ParseActions(v)
+		} else {
+			log.Errorf("missed argument `action`")
+			pterm.Error.Printfln("missing argument `action`")
+			return errors.New("missing argument `action`")
+		}
+
+		if v, err := cmd.Flags().GetString("list-option"); err == nil && v != "" {
+			opt.ParseListOption(v, msg)
+		}
+
+		err = codegen.GenerateAddRpc(pb, msg, opt)
+		if err != nil {
+			log.Errorf("err:%v", err)
+			return err
+		}
+
+		return err
+	},
+}
+
+func init() {
+	genAddRpcCmd.Flags().StringP("model", "m", "", "model name，must be specified")
+	genAddRpcCmd.Flags().StringP("gen-to", "t", "", "generate go source code path, will be used in gen go source code")
+
+	genAddRpcCmd.Flags().StringP("action", "a", "", "action for adding, segmente by ';',\nsupports: add/set/get/list/del/update")
+
+	genAddRpcCmd.Flags().StringP("list-option", "l", "", "list options, segmente by ';'")
+
+	genAddRpcCmd.Flags().String("default-role", "", "default role")
+
+	genCmd.AddCommand(genAddRpcCmd)
+}

cli/gen_state.go

@@ -36,5 +36,5 @@ func runGenState(cmd *cobra.Command, args []string) (err error) {
 }
 
 func init() {
-	rootCmd.AddCommand(genStateCmd)
+	genCmd.AddCommand(genStateCmd)
 }

cli/gen_table.go

@@ -14,6 +14,18 @@ var genTableCmd = &cobra.Command{
 }
 
 func runGenTable(cmd *cobra.Command, args []string) (err error) {
+	err = codegen.GenerateOrm(pb)
+	if err != nil {
+		log.Errorf("err:%v", err)
+		return err
+	}
+
+	err = codegen.GenerateTableName(pb)
+	if err != nil {
+		log.Errorf("err:%v", err)
+		return err
+	}
+
 	err = codegen.GenerateTable(pb)
 	if err != nil {
 		log.Errorf("err:%v", err)