Gorm多态及相关标签

大杯无糖

已于 2023-01-18 17:24:06 修改

阅读量295

点赞数 1

分类专栏： Gorm 文章标签： golang

于 2022-10-31 11:00:57 首次发布

本文链接：https://blog.csdn.net/sunriseYJP/article/details/127610828

版权

Gorm 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

先来看以下代码


//课件
type LessonWare struct {
	gorm.Model        //课件创建是键在gorm.Model中
	Name       string `json:"name" gorm:"comment:课件名字"`
	UserID     uint   `json:"user_id" gorm:"comment:课件创建人id"`
	LessonID   uint   `json:"lesson_id" gorm:"comment:课件所属课程ID"`
}
//课程
type Lesson struct {
	gorm.Model
	Name string `json:"name" gorm:"comment:课程名字"`
    LessonWare LessonWare
}
//用户
type User struct {
	gorm.Model //课件创建是键在gorm.Model中
	Name       string
    LessonWare LessonWare
}

其中，课程拥有课件，用户拥有课件，所以我们的Lesson和User结构体中，就要有LessonWare结构体，同时LessonWare结构体中也要打上两者的标签。如果除了Lesson和User拥有LessonWare，还有LessonWareType课件分类也拥有课件，其他等等，那么就需要在LessonWare结构体中放入好多不同的ID，所以多态的使用就出现了，看下方代码

//课件
type LessonWare struct {
	gorm.Model        //课件创建是键在gorm.Model中
	Name       string `json:"name" gorm:"comment:课件名字"`
	OwnerType  string
	OwnerID    uint
}
type Lesson struct {
	gorm.Model
	Name       string     `json:"name" gorm:"comment:课程名字"`
	LessonWare LessonWare `gorm:"polymorphic:Owner"`
}
type User struct {
	gorm.Model //课件创建是键在gorm.Model中
	Name       string
	LessonWare LessonWare `gorm:"polymorphic:Owner"`
}

func polymorphic()  {
	GLOAB_DB.AutoMigrate(&LessonWare{},&Lesson{},&User{})
	GLOAB_DB.Create(&User{Name: "鸿飞",LessonWare: LessonWare{
		Name: "鸿飞一号课件",
	}})
	GLOAB_DB.Create(&Lesson{Name: "鸿飞课程",LessonWare:LessonWare{
		Name:"鸿飞二号课件",
	}})
}

数据库生成数据显示如下

可以看到，我们使用多态后，只用在属于多个结构体的LessonWare结构体定义一个OwnerType和OwnerID即可，这就是gorm多态的应用场景。

下方的一些标签都是服务于多态，使多态更加方便我们使用。

PolymorphicValue标签

我们可以看到我们上面的数据上面，owner_type字段一个是users，一个是lessons，PolymorphicValue标签就是让我们自定义owner_type字段。看下方代码

//课件
type LessonWare struct {
	gorm.Model        //课件创建是键在gorm.Model中
	Name       string `json:"name" gorm:"comment:课件名字"`
	OwnerType  string
	OwnerID    uint
}
type Lesson struct {
	gorm.Model
	Name       string     `json:"name" gorm:"comment:课程名字"`
	LessonWare LessonWare `gorm:"polymorphic:Owner;polymorphicValue:lesson"`
}
type User struct {
	gorm.Model //课件创建是键在gorm.Model中
	Name       string
	LessonWare LessonWare `gorm:"polymorphic:Owner;polymorphicValue:user"`
}

func polymorphic()  {
	GLOAB_DB.AutoMigrate(&LessonWare{},&Lesson{},&User{})
	GLOAB_DB.Create(&User{Name: "鸿飞",LessonWare: LessonWare{
		Name: "鸿飞一号课件",
	}})
	GLOAB_DB.Create(&Lesson{Name: "鸿飞课程",LessonWare:LessonWare{
		Name:"鸿飞二号课件",
	}})
}

注意我们添加了PolymorphicValue标签，下方是数据库数据