go_config.sh 1.0 KB

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