// GORM 自动迁移, 创建表 // 1.读取 etc/usercenter.yaml 文件, 获取配置,连接数据库 // 2.遍历 orm 包下的所有文件, 获取所有表名,自动迁移 // 1.读取 etc/usercenter.yaml 文件, 获取配置,连接数据库 package orm import ( "backend/usercenter/internal/config" "flag" "fmt" "github.com/zeromicro/go-zero/core/conf" "gorm.io/driver/mysql" "gorm.io/gorm" ) var configFile = flag.String("f", "etc/usercenter.yaml", "the config file") func AutoMigrate() { flag.Parse() var c config.Config conf.MustLoad(*configFile, &c) db, err := gorm.Open(mysql.Open(c.MySQL.DSN()), &gorm.Config{}) if err != nil { panic(err) } // 自动迁移 User 表 fmt.Println("迁移User表") if err := db.AutoMigrate(&User{}); err != nil { panic(err) } fmt.Println("迁移User表完成") // 迁移完成,打印信息 fmt.Println("全部迁移完成") }