123456789101112131415161718192021222324252627282930 |
- # 封装的 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"
|