117 lines
3.0 KiB
Go
117 lines
3.0 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
|
|
"git.magicany.cc/black1552/gin-base/cmd/gendao"
|
|
"git.magicany.cc/black1552/gin-base/config"
|
|
_ "git.magicany.cc/black1552/gin-base/database/drivers"
|
|
)
|
|
|
|
func main() {
|
|
ctx := context.Background()
|
|
|
|
// 加载配置
|
|
cfg := config.GetAllConfig()
|
|
if cfg == nil {
|
|
fmt.Println("❌ 错误: 配置为空")
|
|
os.Exit(1)
|
|
}
|
|
|
|
// 检查数据库配置
|
|
dbConfigMap, ok := cfg["database"].(map[string]any)
|
|
if !ok || len(dbConfigMap) == 0 {
|
|
fmt.Println("❌ 错误: 未找到数据库配置")
|
|
os.Exit(1)
|
|
}
|
|
|
|
// 获取默认数据库配置
|
|
defaultDbConfig, ok := dbConfigMap["default"].(map[string]any)
|
|
if !ok {
|
|
fmt.Println("❌ 错误: 未找到 default 数据库配置")
|
|
os.Exit(1)
|
|
}
|
|
|
|
// 提取配置值
|
|
host := getStringValue(defaultDbConfig, "host", "127.0.0.1")
|
|
port := getStringValue(defaultDbConfig, "port", "3306")
|
|
name := getStringValue(defaultDbConfig, "name", "test")
|
|
user := getStringValue(defaultDbConfig, "user", "root")
|
|
pass := getStringValue(defaultDbConfig, "pass", "")
|
|
dbType := getStringValue(defaultDbConfig, "type", "mysql")
|
|
|
|
fmt.Println("=== Gin-Base DAO 代码生成工具 ===")
|
|
fmt.Printf("📊 数据库: %s\n", name)
|
|
fmt.Printf("🔧 类型: %s\n", dbType)
|
|
fmt.Printf("🌐 主机: %s:%s\n\n", host, port)
|
|
|
|
// 构建数据库连接字符串
|
|
link := fmt.Sprintf("mysql:%s:%s@tcp(%s:%s)/%s?charset=utf8&parseTime=true&loc=Local",
|
|
user, pass, host, port, name,
|
|
)
|
|
|
|
// 准备表名参数
|
|
tablesArg := ""
|
|
if len(os.Args) > 1 {
|
|
tablesArg = joinStrings(os.Args[1:], ",")
|
|
fmt.Printf("📋 指定生成表: %s\n\n", tablesArg)
|
|
} else {
|
|
fmt.Println("💡 提示: 可以通过命令行参数指定要生成的表")
|
|
fmt.Println(" 例如: gin-dao-gen users orders\n")
|
|
}
|
|
|
|
// 调用 gendao 的 Dao 函数生成代码
|
|
genDao := gendao.CGenDao{}
|
|
input := gendao.CGenDaoInput{
|
|
Link: link,
|
|
Tables: tablesArg,
|
|
Path: "./",
|
|
DaoPath: "dao",
|
|
DoPath: "model/do",
|
|
EntityPath: "model/entity",
|
|
TablePath: "model/table",
|
|
Group: "default",
|
|
JsonCase: "CamelLower",
|
|
DescriptionTag: true,
|
|
GenTable: true,
|
|
}
|
|
|
|
fmt.Println("🔨 开始生成代码...")
|
|
_, err := genDao.Dao(ctx, input)
|
|
if err != nil {
|
|
fmt.Printf("\n❌ 代码生成失败: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
|
|
fmt.Println("\n🎉 代码生成完成！")
|
|
fmt.Println("📁 生成的文件位于:")
|
|
fmt.Println(" - ./dao/")
|
|
fmt.Println(" - ./model/do/")
|
|
fmt.Println(" - ./model/entity/")
|
|
fmt.Println(" - ./model/table/")
|
|
}
|
|
|
|
// 辅助函数：从 map 中获取字符串值
|
|
func getStringValue(m map[string]any, key string, defaultValue string) string {
|
|
if val, ok := m[key]; ok {
|
|
if str, ok := val.(string); ok {
|
|
return str
|
|
}
|
|
}
|
|
return defaultValue
|
|
}
|
|
|
|
// 辅助函数：连接字符串数组
|
|
func joinStrings(strs []string, sep string) string {
|
|
if len(strs) == 0 {
|
|
return ""
|
|
}
|
|
result := strs[0]
|
|
for i := 1; i < len(strs); i++ {
|
|
result += sep + strs[i]
|
|
}
|
|
return result
|
|
}
|