允许指定 gen to 自动给 model

lazygophers · Jan 1, 2025 · 5474d74 · 5474d74
1 parent bcad582
commit 5474d74
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cli/gen/gen_add_rpc.go b/cli/gen/gen_add_rpc.go
@@ -26,7 +26,11 @@ var addRpcCmd = &cobra.Command{
 		}
 
 		var msg *codegen.PbMessage
-		if v, err := cmd.Flags().GetString("model"); err == nil && v != "" {
+		if v, err := cmd.Flags().GetString("model"); err == nil && (v != "" || opt.GenTo != "") {
+			if v == "" {
+				v = opt.GenTo
+			}
+
 			// 先看一下加了 Model 的 时候存在
 			{
 				model := stringx.ToSnake(v)