使用泛型设计gorm扩展字段
大型项目经常需要在标准的用户表或员工表加些扩展字段，如果你有一套标准的用户管理系统的话，要么每个项目复制过来改一遍或者添加十几个固定的自定义字段Def1~Def10， 现在可以用在gorm定义模型时使用泛型来解决这个问题
gorm embeded嵌入字段
使用gorm:"embedded"定义嵌入字段
type User struct {
gorm.Model
Username string
Password string
Extend UserExtendField `gorm:"embedded"` // 添加embed tag
}
type UserExtendField struct {
OpenID string
UnionID string
}
上面的model等价于这个, 两个生成的表结构是一致的
type User struct {
gorm.Model
Username string
Password string
OpenID string
UnionID string
}
上面两个模型都会生成同样的ddl sql, 如下
create table if not users
(
id bigint unsigned auto_increment
primary key,
created_at datetime(3) null,
updated_at datetime(3) null,
deleted_at datetime(3) null,
username longtext null,
password longtext null,
org_id longtext null,
open_id longtext null,
union_id longtext null
)
使用泛型定义可扩展的模型
type User[ExtendField any] struct {
gorm.Model
Username string
Password string
Extend ExtendField `gorm:"embedded"`
}
// 需要自定义表名，否则gorm识别到的泛型的表名是不符合数据库标准的，建表会错误
func (User[Extend, Item]) TableName() string {
return "ma_users"
}
// 定义扩展字段结构
type ExtendUserField struct {
OrgID string
OpenID string
UnionID string
}
使用方法
// 生成数据库表结构
db.AutoMigrate(User[ExtendUserField])
// 查找
var list []User[ExtendUserField]
db.model(&User[ExtendUserField]).Find(&list)
可扩展的子表
type User[ExtendField any,Item any] struct {
gorm.Model
Username string
Password string
Extend ExtendField `gorm:"embedded"`
Items Item `gorm:"foreignKey:user_id"` // 这里需要写外键
}
// 需要自定义表名，否则gorm识别到的泛型的表名是不符合数据库标准的，建表会错误
func (User[Extend, Item]) TableName() string {
return "ma_users"
}
type UserItem struct {
gorm.Model
UserID uint
Product string
}
// 定义扩展字段结构
type ExtendUserField struct {
OrgID string
OpenID string
UnionID string
}
使用方法
// 查找
var list []User[ExtendUserField, UserItem]
db.model(&User[ExtendUserField, UserItem]).Find(&list)
使用类型别名简化泛型类型
使用类型别名， 可以把上面的泛型代码简化
type MyUser = User[ExUserField, UserItem]
var list []MyUser
db.model(&MyUser).Find(&list)