GORM中使用虚拟字段
作者：matrix 被围观: 12 次 发布时间：2023-11-30 分类：Golang | 无评论 »
使用gorm时，可能需要处理虚拟字段（不在数据库中实际存在的字段）的情况。可以使用结构体tag标签来支持
User结构体模型
代码语言：javascript复制type User struct {
ID uint `gorm:"primaryKey;not null"` // 主键ID
// 虚拟字段
Isvip int `gorm:"-;default:0"` // 是否vip 1是 0否
}说明：
IsVip字段被标记为 gorm:"-" ，表示虚拟字段。GORM在进行数据库操作（如查询、插入、更新等）时，将不会考虑此字段。同时，可以使用default 标签为其指定默认值。
自定义获取器
自定义一个Get方法 例如，下面的GetIsVip方法会基于用户的VIP状态来返回相应的值：
代码语言：javascript复制func (u *User) GetIsVip() int {
if u.Vip != nil && u.Vip.IsActive == 1 {
return 1
}
return 0
}应用获取器
在查询User对象时，GORM提供了 AfterFind 方法来自动执行特定逻辑。这在处理虚拟字段时很有用：
// 查询数据时自动赋值字段
func (u *User) AfterFind(tx *gorm.DB) (err error) {
if u.Vip == nil {
//TIPS：Association方法手动触发模型关联。如果使用Preload会再次查询User主表，不推荐
// tx.Preload("Vip").First(&u, u.ID) //不推荐
tx.Model(u).Association("Vip").Find(&u.Vip)
}
u.Isvip = u.GetIsVip() // 手动触发虚拟字段计算
return
}说明：
首先检查VIP信息是否已加载。如果未加载，则使用Association方法手动触发加载。之后，我们使用前面定义的GetIsVip方法来计算并设置Isvip字段的值。
注意
使用AfterFind可能会覆盖Isvip字段的默认值（如default:0 ）
