How to make multiple models auto migrate in gorm

在GORM中实现多个模型的自动迁移是一个简单而高效的过程，主要涉及使用AutoMigrate方法。这个方法会自动检测模型结构的变化，并更新数据库的表结构以匹配模型。以下是具体的实现步骤和示例：

步骤1: 定义模型

首先，你需要定义你的数据模型，每个模型都对应数据库中的一个表。例如，假设我们有两个模型：User和Product。

go
package main

import (
    "gorm.io/gorm"
)

type User struct {
    gorm.Model
    Name string
    Email string
    Age int
}

type Product struct {
    gorm.Model
    Code string
    Price uint
}

步骤2: 数据库连接

接下来，需要建立到数据库的连接。这里以SQLite为例：

go
package main

import (
    "gorm.io/driver/sqlite"
    "gorm.io/gorm"
)

func setupDB() *gorm.DB {
    db, err := gorm.Open(sqlite.Open("gorm.db"), &gorm.Config{})
    if err != nil {
        panic("failed to connect database")
    }
    return db
}

步骤3: 进行自动迁移

最后，使用AutoMigrate方法对所有模型进行自动迁移。你可以一次性迁移多个模型。

go
func main() {
    db := setupDB()

    // 迁移 schema
    err := db.AutoMigrate(&User{}, &Product{})
    if err != nil {
        panic("migration failed")
    }
}

在上面的代码中，AutoMigrate将会检查User和Product模型的结构，并在数据库中创建或修改相应的表。如果表已经存在，GORM会检查表结构是否需要更新（比如添加新的列或者修改列类型等），以确保表结构与模型一致。

示例

假设你初次运行迁移时，数据库是空的，GORM会为User和Product模型创建新的表。如果你后续在User模型中添加了一个新的字段，比如：

go
type User struct {
    gorm.Model
    Name string
    Email string
    Age int
    Address string // 新增字段
}

再次运行相同的迁移代码时，GORM会自动在users表中添加address列，而不会影响已有数据。

结论

使用GORM的AutoMigrate方法可以非常方便地同步Go应用中的模型结构与数据库表结构。这样的自动迁移机制减少了手动维护数据库结构的需要，使开发过程更加高效和准确。当然，对于生产环境，还是建议更加谨慎地处理数据库迁移，可能需要更复杂的迁移策略和备份方案。