Go语言之GORM框架——预加载，关联标签与多态关联，自定义数据类型与事务(完结篇)

package main
 
import (
    "fmt"
    "gorm.io/driver/sqlite"
    "gorm.io/gorm"
)
 
type Product struct {
    gorm.Model
    Code  string
    Price uint
    Tags  []Tag `gorm:"many2many:product_tags;"`
}
 
type Tag struct {
    ID string `gorm:"primaryKey"`
}
 
func main() {
    db, err := gorm.Open(sqlite.Open("test.db"), &gorm.Config{})
    if err != nil {
        panic("数据库连接失败")
    }
 
    // 初始化数据库
    db.AutoMigrate(&Product{}, &Tag{})
 
    // 创建标签
    db.Create(&Tag{ID: "awesome"})
    db.Create(&Tag{ID: "sale"})
 
    // 创建产品并关联标签
    product := Product{
        Code:  "1001",
        Price: 1000,
        Tags:  []Tag{{ID: "awesome"}, {ID: "sale"}},
    }
    db.Create(&product)
 
    // 预加载产品的标签
    var p1 Product
    db.Preload("Tags").First(&p1, "code = ?", "1001")
    fmt.Printf("产品 %s 的标签: %v\n", p1.Code, p1.Tags)
 
    // 开启事务
    tx := db.Begin()
    defer func() {
        if err := recover(); err != nil {
            tx.Rollback()
        }
    }()
 
    // 在事务内进行操作
    tx.Model(&product).Association("Tags").Append(Tag{ID: "new"})
    tx.Commit()
 
    // 再次预加载以确认变更
    var p2 Product
    db.Preload("Tags").First(&p2, "code = ?", "1001")
    fmt.Printf("产品 %s 的标签 (更新后): %v\n", p2.Code, p2.Tags)
}

这段代码首先展示了如何使用GORM进行数据库迁移和建立数据库连接。接着创建了两个简单的数据模型Product和Tag，并展示了如何预加载产品的标签。然后，演示了如何在事务中添加新的标签到产品，并在事务提交后再次预加载以确认变更。这个过程展示了如何在GORM中使用事务来管理数据的一致性和完整性。