How to make multiple models auto migrate in gorm

Implementing automatic migration for multiple models in GORM is straightforward and efficient, primarily utilizing the AutoMigrate method. This method automatically detects changes in the model structure and updates the database table structure to align with the model. Below are the specific implementation steps and examples:

Step 1: Define Models

First, define your data models, each corresponding to a table in the database. For instance, consider two models: User and 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
}

Step 2: Database Connection

Next, set up a connection to the database. For this example, we use 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
}

Step 3: Perform AutoMigrate

Finally, utilize the AutoMigrate method to automatically migrate all models. You can migrate multiple models simultaneously.

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

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

In the above code, AutoMigrate inspects the structure of the User and Product models and creates or modifies the corresponding database tables. If the tables already exist, GORM checks whether the table structure requires updates (e.g., adding new columns or modifying column types) to maintain consistency with the models.

Example

When you first run the migration with an empty database, GORM creates new tables for the User and Product models. If you subsequently add a new field to the User model, such as:

go
type User struct {
    gorm.Model
    Name string
    Email string
    Age int
    Address string // New field
}

Running the same migration code again will automatically add the address column to the users table without impacting existing data.

Conclusion

Utilizing GORM's AutoMigrate method facilitates the synchronization of model structures in Go applications with database table structures. This automatic migration mechanism minimizes the need for manual database structure maintenance, enhancing development efficiency and accuracy. However, for production environments, it is advisable to handle database migrations with greater caution, potentially implementing more sophisticated migration strategies and backup procedures.