|
@@ -0,0 +1,30 @@
|
|
|
|
+# 封装的 gorm 生成器函数
|
|
|
|
+generate_gorm_models() {
|
|
|
|
+ local ip=$1
|
|
|
|
+ local port=$2
|
|
|
|
+ local user=$3
|
|
|
|
+ local password=$4
|
|
|
|
+ local db=$5
|
|
|
|
+ local tables=$6
|
|
|
|
+ local target_dir="/tmp" # 默认输出目录为 /tmp
|
|
|
|
+
|
|
|
|
+ # 检查是否有必要参数
|
|
|
|
+ if [ -z "$ip" ] || [ -z "$port" ] || [ -z "$user" ] || [ -z "$password" ] || [ -z "$db" ] || [ -z "$tables" ]; then
|
|
|
|
+ echo "Usage: generate_gorm_models <ip> <port> <user> <password> <db> <tables>"
|
|
|
|
+ return 1
|
|
|
|
+ fi
|
|
|
|
+
|
|
|
|
+ # 生成 Gorm 模型
|
|
|
|
+ gorm-plus generate --json --host "$ip" --u "$user" --p "$password" --port "$port" --target "$target_dir" --d "$db" --tables "$tables"
|
|
|
|
+
|
|
|
|
+ # 检查是否生成成功
|
|
|
|
+ if [ $? -eq 0 ]; then
|
|
|
|
+ echo "Models generated successfully in $target_dir"
|
|
|
|
+ cat "$target_dir/tmp.go"
|
|
|
|
+ else
|
|
|
|
+ echo "Failed to generate models."
|
|
|
|
+ fi
|
|
|
|
+}
|
|
|
|
+
|
|
|
|
+# 示例调用方式,前提:go get github.com/joeyzhouy/gorm-plus go install github.com/joeyzhouy/gorm-plus
|
|
|
|
+# generate_gorm_models "127.0.0.1" "3306" "root" "root" "test" "user,product"
|