如何在Go中使用Gorm实现单表继承？尝试discriminator无效

Gorm官方并没有原生支持单表继承（Single Table Inheritance），你尝试的gorm:"discriminator"标签其实是用于多态关联场景的，并非单表继承，所以不会生效。下面是一种手动实现单表继承的可行方案：

实现步骤 #

1. 定义基础模型与鉴别器字段 #

首先创建包含所有共用字段的基础模型，同时添加一个用于区分不同子类型的鉴别器字段（比如Type）：

import "gorm.io/gorm"

type BaseModel struct {
	gorm.Model        // 包含ID、CreatedAt、UpdatedAt、DeletedAt
	Type     string   `gorm:"column:type;not null;index"` // 鉴别器字段，标记子类型
	Name     string   `gorm:"column:name;not null"`       // 共用示例字段
}

2. 定义子模型（嵌入基础模型） #

子模型通过嵌入BaseModel来继承共用字段，同时添加自身独有的字段：

// 用户模型
type User struct {
	BaseModel
	Email    string `gorm:"column:email;unique"` // 用户独有的字段
}

// 管理员模型
type Admin struct {
	BaseModel
	Privilege int `gorm:"column:privilege_level"` // 管理员独有的字段
}

3. 自动设置鉴别器字段 #

利用Gorm的BeforeCreate钩子，在创建子模型实例时自动填充Type字段，避免手动设置：

// User的创建钩子
func (u *User) BeforeCreate(tx *gorm.DB) error {
	u.Type = "user"
	return nil
}

// Admin的创建钩子
func (a *Admin) BeforeCreate(tx *gorm.DB) error {
	a.Type = "admin"
	return nil
}

4. 处理CRUD操作 #

创建实例

直接调用Gorm的Create方法即可，钩子会自动设置Type：

// 创建用户
user := &User{Name: "Alice", Email: "alice@example.com"}
db.Create(user)

// 创建管理员
admin := &Admin{Name: "Bob", Privilege: 10}
db.Create(admin)

查询实例

查询时通过Type字段过滤对应的子类型：

// 查询所有用户
var users []User
db.Where("type = ?", "user").Find(&users)

// 查询单个管理员
var admin Admin
db.Where("type = ? AND id = ?", "admin", 1).First(&admin)

动态查询（根据ID获取对应子类型）

如果需要根据ID自动识别子类型并返回对应结构体，可以先查询基础模型的Type，再转换为对应子模型：

import "fmt"

func GetModelByID(db *gorm.DB, id uint) (interface{}, error) {
	var base BaseModel
	if err := db.First(&base, id).Error; err != nil {
		return nil, err
	}

	switch base.Type {
	case "user":
		var user User
		if err := db.First(&user, id).Error; err != nil {
			return nil, err
		}
		return user, nil
	case "admin":
		var admin Admin
		if err := db.First(&admin, id).Error; err != nil {
			return nil, err
		}
		return admin, nil
	default:
		return nil, fmt.Errorf("未知模型类型: %s", base.Type)
	}
}

注意事项 #

• 数据库表迁移：可以使用db.AutoMigrate(&User{}, &Admin{})让Gorm自动创建包含所有字段的表，或者直接基于BaseModel扩展字段。
• 字段冲突：确保所有子模型的字段名不冲突，否则会导致数据库字段覆盖。
• 软删除：如果基础模型包含gorm.Model的软删除字段，所有子模型都会继承该特性。

内容的提问来源于stack exchange，提问作者RedOne KaiTo