Golang Gorm实现自定义多态模型关联查询
更新时间：2024年11月04日 09:07:03 作者：erternalKing
GORM 是一个流行的开源 ORM (Object-Relational Mapping) 库,专为 Go 语言设计,它简化了与 SQL 数据库的交互,GORM 封装了数据库操作,使得开发者能够通过简单的链式调用来执行 CRUD,本文给大家介绍了Golang Gorm实现自定义多态模型关联查询,需要的朋友可以参考下
一、表结构
CREATE TABLE `orders` ( `id` int unsigned NOT NULL AUTO_INCREMENT, `order_no` varchar(32) NOT NULL DEFAULT '', `orderable_id` int unsigned NOT NULL DEFAULT '0', `orderable_type` char(30) NOT NULL DEFAULT '', `status` tinyint unsigned NOT NULL DEFAULT '0', `created_at` datetime NOT NULL, `updated_at` datetime NOT NULL, `deleted_at` datetime DEFAULT NULL, PRIMARY KEY (`id`) ) ENGINE=InnoDB AUTO_INCREMENT=6 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci; CREATE TABLE `phone` ( `id` int unsigned NOT NULL AUTO_INCREMENT, `phone_name` varchar(50) NOT NULL DEFAULT '', `phone_model` varchar(30) NOT NULL DEFAULT '', `status` tinyint unsigned NOT NULL DEFAULT '0', `created_at` datetime NOT NULL, `updated_at` datetime NOT NULL, `deleted_at` datetime DEFAULT NULL, PRIMARY KEY (`id`) ) ENGINE=InnoDB AUTO_INCREMENT=1003 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci; CREATE TABLE `cars` ( `id` int unsigned NOT NULL AUTO_INCREMENT, `car_name` varchar(50) NOT NULL DEFAULT '', `car_model` varchar(30) NOT NULL DEFAULT '', `status` tinyint unsigned NOT NULL DEFAULT '0', `created_at` datetime NOT NULL, `updated_at` datetime NOT NULL, `deleted_at` datetime DEFAULT NULL, PRIMARY KEY (`id`) ) ENGINE=InnoDB AUTO_INCREMENT=1003 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_0900_ai_ci;
二、接口定义
// LoaderAble 接口定义数据加载行为
type LoaderAble interface {
LoadAble(IDs []int) (map[int]any, error)
}
// LoadAbleItem 接口定义了可加载项的通用方法
type LoadAbleItem interface {
GetAbleType() string // 获取类型键值
GetAbleID() int // 获取ID
SetLoadedAbleData(data any) // 设置加载的数据
}三、模型定义并实现接口
type Order struct {
Id int `json:"id"`
OrderNo string `json:"order_no"`
OrderableId int `json:"orderable_id"`
OrderableType string `json:"orderable_type"`
Orderable any `json:"orderable" gorm:"-"`
Status uint8 `json:"status"`
CreatedAt *time.Time `json:"created_at"`
UpdatedAt *time.Time `json:"updated_at"`
DeletedAt gorm.DeletedAt `json:"deleted_at"`
}
func (tb *Order) TableName() string {
return "orders"
}
func (tb *Order) GetAbleType() string {
return tb.OrderableType
}
func (tb *Order) GetAbleID() int {
return tb.OrderableId
}
func (tb *Order) SetLoadedAbleData(data any) {
tb.Orderable = data
}
//--------------------------------------分割线--------------------------------------
type Car struct {
Id int `json:"id"`
CarName string `json:"car_name"`
CarModel string `json:"car_model"`
Status uint8 `json:"status"`
CreatedAt *time.Time `json:"created_at"`
UpdatedAt *time.Time `json:"updated_at"`
DeletedAt gorm.DeletedAt `json:"deleted_at"`
}
func (tb *Car) TableName() string {
return "cars"
}
// CarLoaderAble 实现 Loader 接口
type CarLoaderAble struct{}
// LoadAble 具体实现加载多态关联逻辑
// IDs 多态关联类型ID(主要参数)
func (loader *CarLoaderAble) LoadAble(IDs []int) (resultMap map[int]any, err error) {
IDsLen := len(IDs)
if IDsLen == 0 {
return
}
car := make([]*Car, 0, IDsLen)
err = mysql.DefaultMysql.Db.Debug().Where("id IN (?) AND status = ?", IDs, 1).Find(&car).Error
if err != nil {
return
}
resultMap = make(map[int]any, IDsLen)
for _, item := range car {
resultMap[item.Id] = item
}
return
}
//--------------------------------------分割线--------------------------------------
type Phone struct {
Id int `json:"id"`
PhoneName string `json:"phone_name"`
PhoneModel string `json:"phone_model"`
Status uint8 `json:"status" gorm:"column:status"`
StatusNew uint8 `json:"status_new" gorm:"-"`
CreatedAt *time.Time `json:"created_at"`
UpdatedAt *time.Time `json:"updated_at"`
DeletedAt gorm.DeletedAt `json:"deleted_at"`
}
func (tb *Phone) TableName() string {
return "phone"
}
func (tb *Phone) AfterFind(tx *gorm.DB) (err error) {
tb.StatusNew = tb.Status
return
}
// PhoneLoaderAble 实现 Loader 接口
type PhoneLoaderAble struct{}
// LoadAble 具体实现加载多态关联逻辑
// IDs 多态关联类型ID(主要参数)
func (loader *PhoneLoaderAble) LoadAble(IDs []int) (resultMap map[int]any, err error) {
IDsLen := len(IDs)
if IDsLen == 0 {
return
}
phone := make([]*Phone, 0, IDsLen)
err = mysql.DefaultMysql.Db.Debug().Where("id IN (?) AND status = ?", IDs, 1).Find(&phone).Error
if err != nil {
return
}
resultMap = make(map[int]any, IDsLen)
for _, item := range phone {
resultMap[item.Id] = item
}
return
}四、创建loader预加载器
// LoaderAbleFactory 用于管理不同类型的加载器
type LoaderAbleFactory struct {
loaders map[string]LoaderAble
}
func NewLoaderAbleFactory() *LoaderAbleFactory {
return &LoaderAbleFactory{
loaders: make(map[string]LoaderAble),
}
}
func (f *LoaderAbleFactory) RegisterLoader(typeName string, loader LoaderAble) {
f.loaders[typeName] = loader
}五、注册loader预加载器服务
var (
loaderAbleFactory *LoaderAbleFactory
)
// init 选择在项目启动初始化时进行全局加载
func init() {
loaderAbleFactory = NewLoaderAbleFactory()
loaderAbleFactory.RegisterLoader("phone", &PhoneLoaderAble{})
loaderAbleFactory.RegisterLoader("car", &CarLoaderAble{})
log.Println("多态模型关系注册成功...")
}六、实现LoadAble通用的加载函数
// LoadAble 通用的加载函数，可以处理任何实现了 LoadableItem 接口的切片
func LoadAble[T LoadAbleItem](items []T) error {
if len(items) == 0 {
return nil
}
// 按类型分组收集ID
typeIDsMap := make(map[string][]int)
for _, item := range items {
typeKey := item.GetAbleType()
typeIDsMap[typeKey] = append(typeIDsMap[typeKey], item.GetAbleID())
}
// 使用对应的loader加载数据
typeDataMap := make(map[string]map[int]any)
for typeName, ids := range typeIDsMap {
loader, ok := loaderAbleFactory.loaders[typeName]
if !ok {
continue
}
resultMap, err := loader.LoadAble(ids)
if err != nil {
return err
}
typeDataMap[typeName] = resultMap
}
// 填充数据
for _, item := range items {
if dataMap, ok := typeDataMap[item.GetAbleType()]; ok {
if data, exists := dataMap[item.GetAbleID()]; exists {
item.SetLoadedAbleData(data)
}
}
}
return nil
}七、调试
- 准备数据
--orders表 INSERT INTO `orders` (`id`, `order_no`, `orderable_id`, `orderable_type`, `status`, `created_at`, `updated_at`, `deleted_at`) VALUES (1, '202411010001', 1002, 'car', 1, '2024-11-01 12:03:03', '2024-11-01 12:03:06', NULL); INSERT INTO `orders` (`id`, `order_no`, `orderable_id`, `orderable_type`, `status`, `created_at`, `updated_at`, `deleted_at`) VALUES (2, '202411010002', 1001, 'phone', 1, '2024-11-01 12:03:03', '2024-11-01 12:03:06', NULL); INSERT INTO `orders` (`id`, `order_no`, `orderable_id`, `orderable_type`, `status`, `created_at`, `updated_at`, `deleted_at`) VALUES (3, '202411010003', 1000, 'car', 1, '2024-11-01 12:03:03', '2024-11-01 12:03:06', NULL); INSERT INTO `orders` (`id`, `order_no`, `orderable_id`, `orderable_type`, `status`, `created_at`, `updated_at`, `deleted_at`) VALUES (4, '202411010004', 1001, 'car', 1, '2024-11-01 12:03:03', '2024-11-01 12:03:06', NULL); INSERT INTO `orders` (`id`, `order_no`, `orderable_id`, `orderable_type`, `status`, `created_at`, `updated_at`, `deleted_at`) VALUES (5, '202411010005', 1002, 'phone', 1, '2024-11-01 12:03:03', '2024-11-01 12:03:06', NULL); --phone表 INSERT INTO `phone` (`id`, `phone_name`, `phone_model`, `status`, `created_at`, `updated_at`, `deleted_at`) VALUES (1000, 'XiaoMi', '2S', 2, '2024-11-01 11:59:37', '2024-11-01 11:59:40', NULL); INSERT INTO `phone` (`id`, `phone_name`, `phone_model`, `status`, `created_at`, `updated_at`, `deleted_at`) VALUES (1001, 'HUAWEI', 'mate60', 1, '2024-11-01 11:59:54', '2024-11-01 11:59:57', NULL); INSERT INTO `phone` (`id`, `phone_name`, `phone_model`, `status`, `created_at`, `updated_at`, `deleted_at`) VALUES (1002, 'Apple', 'iPhone 12 Pro Max', 1, '2024-11-01 12:00:26', '2024-11-01 12:00:28', NULL); --cars表 INSERT INTO `cars` (`id`, `car_name`, `car_model`, `status`, `created_at`, `updated_at`, `deleted_at`) VALUES (1000, '奥迪', 'A6L', 1, '2024-11-01 11:57:53', '2024-11-01 11:57:55', NULL); INSERT INTO `cars` (`id`, `car_name`, `car_model`, `status`, `created_at`, `updated_at`, `deleted_at`) VALUES (1001, '宝马', '5系', 1, '2024-11-01 11:58:12', '2024-11-01 11:58:15', NULL); INSERT INTO `cars` (`id`, `car_name`, `car_model`, `status`, `created_at`, `updated_at`, `deleted_at`) VALUES (1002, '奔驰', 'E300', 1, '2024-11-01 11:58:53', '2024-11-01 11:58:56', NULL);
- 编写代码
// Gin框架+Gorm为例
api.GET("/orders", controller.GetOrderList)
// GetOrderList 获取订单列表接口
func GetOrderList(c *gin.Context) {
orders := make([]*model.Order, 0)
err := mysql.DefaultMysql.Db.Debug().Find(&orders).Error
if err != nil {
c.JSON(200, gin.H{"error": err.Error()})
return
}
err = model.LoadAble(orders)
if err != nil {
c.JSON(200, gin.H{"error": err.Error()})
return
}
c.JSON(200, gin.H{"data": orders})
}- 发起请求
curl '127.0.0.1:16888/api/orders'
{
"data": [
{
"id": 1,
"order_no": "202411010001",
"orderable_id": 1002,
"orderable_type": "car",
"orderable": {
"id": 1002,
"car_name": "奔驰",
"car_model": "E300",
"status": 1,
"created_at": "2024-11-01T11:58:53+08:00",
"updated_at": "2024-11-01T11:58:56+08:00",
"deleted_at": null
},
"status": 1,
"created_at": "2024-11-01T12:03:03+08:00",
"updated_at": "2024-11-01T12:03:06+08:00",
"deleted_at": null
},
{
"id": 2,
"order_no": "202411010002",
"orderable_id": 1001,
"orderable_type": "phone",
"orderable": {
"id": 1001,
"phone_name": "HUAWEI",
"phone_model": "mate60",
"status": 1,
"status_new": 1,
"created_at": "2024-11-01T11:59:54+08:00",
"updated_at": "2024-11-01T11:59:57+08:00",
"deleted_at": null
},
"status": 1,
"created_at": "2024-11-01T12:03:03+08:00",
"updated_at": "2024-11-01T12:03:06+08:00",
"deleted_at": null
},
{
"id": 3,
"order_no": "202411010003",
"orderable_id": 1000,
"orderable_type": "car",
"orderable": {
"id": 1000,
"car_name": "奥迪",
"car_model": "A6L",
"status": 1,
"created_at": "2024-11-01T11:57:53+08:00",
"updated_at": "2024-11-01T11:57:55+08:00",
"deleted_at": null
},
"status": 1,
"created_at": "2024-11-01T12:03:03+08:00",
"updated_at": "2024-11-01T12:03:06+08:00",
"deleted_at": null
},
{
"id": 4,
"order_no": "202411010004",
"orderable_id": 1001,
"orderable_type": "car",
"orderable": {
"id": 1001,
"car_name": "宝马",
"car_model": "5系",
"status": 1,
"created_at": "2024-11-01T11:58:12+08:00",
"updated_at": "2024-11-01T11:58:15+08:00",
"deleted_at": null
},
"status": 1,
"created_at": "2024-11-01T12:03:03+08:00",
"updated_at": "2024-11-01T12:03:06+08:00",
"deleted_at": null
},
{
"id": 5,
"order_no": "202411010005",
"orderable_id": 1002,
"orderable_type": "phone",
"orderable": {
"id": 1002,
"phone_name": "Apple",
"phone_model": "iPhone 12 Pro Max",
"status": 1,
"status_new": 1,
"created_at": "2024-11-01T12:00:26+08:00",
"updated_at": "2024-11-01T12:00:28+08:00",
"deleted_at": null
},
"status": 1,
"created_at": "2024-11-01T12:03:03+08:00",
"updated_at": "2024-11-01T12:03:06+08:00",
"deleted_at": null
}
]
}以上就是Golang Gorm实现自定义多态模型关联查询的详细内容，更多关于Golang Gorm关联查询的资料请关注脚本之家其它相关文章！
相关文章
这篇文章主要介绍了go不同业务环境变量的设置方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2025-07-07
context是Go语言中用于在多个goroutine之间传递取消信号、超时控制和上下文信息的重要机制,通过合理使用context,开发者可以更高效地管理并发任务,感兴趣的可以了解一下2025-07-07
Go 语言中，有一个名为database/sql的标准库，提供了统一的编程接口，使开发人员能够以一种通用的方式与各种关系型数据库进行交互,本文就来和大家讲讲它的具体操作吧2023-06-06
在Go语言中,有一种从其他语言带来的常见模式：预防性接口,虽然这种模式在 Java 等语言中很有价值,但在Go中往往会成为反模式,本文我们就来深入探讨一下原因2025-01-01
在Go语言开发中,跨平台编译是一个非常重要的功能,本文将详细介绍如何使用Go语言将项目编译为Linux平台的可执行文件,特别是针对国内开发者的优化配置,需要的朋友可以参考下2025-08-08
Golang有自动垃圾回收机制,但是仍然可能会出现内存泄漏的情况,本文主要介绍了Golang内存泄露场景与定位方式的实现,具有一定的参考价值,感兴趣的可以了解一下2024-04-04
这篇文章主要介绍了Go标准库http与fasthttp服务端性能比较,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-06-06
Go 语言的内存分配机制是理解和优化 Go 程序性能的关键,在 Go 中,内存管理是自动进行的,这得益于 Go 的垃圾回收机制,了解内存如何分配和回收,可以帮助我们写出更高性能的代码,本文将深入讲解下 Go 内存分配机制,需要的朋友可以参考下2023-12-12
这篇文章主要介绍了go 中的 fmt 占位符,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友参考下吧2024-01-01
Go Module是Go语言的官方依赖管理解决方案,其提供了一种简单、可靠的方式来管理项目的依赖关系,本文主要介绍了Go Module依赖管理的实现,感兴趣的可以了解一下2024-06-06
最新评论