Gorm 预加载及输出处理(一)- 预加载应用

2019/04/10 10:10
阅读数 92

单条关联查询

先创建两个关联模型:

// 用户模型
type User struct {
    gorm.Model
    Username string    `gorm:"type:varchar(20);not null;unique"`
    Email    string    `gorm:"type:varchar(64);not null;unique"`
    Role     string    `gorm:"type:varchar(32);not null"`
    Active   uint8     `gorm:"type:tinyint unsigned;default:1"`
    Profile  Profile   `gorm:"foreignkey:UserID;association_autoupdate:false"`
}

// 用户信息模型
type Profile struct {
    gorm.Model
    UserID   uint      `gorm:"type:int unsigned;not null;unique"`
    Username string    `gorm:"type:varchar(20);not null;unique"`
    Nickname string    `gorm:"type:varchar(64);not null;unique"`
    Phone    string    `gorm:"type:varchar(32)"`
    Gender   uint8     `gorm:"type:tinyint unsigned;default:0"`
    Birthday time.Time `gorm:"type:datetime;default:null"`
    Sign     string    `gorm:"type:varchar(255)"`
    Avatar   string    `gorm:"type:text"`
}

直接查询单条 User 记录

var user User

DB.Debug().First(&user)

这里省略 JSON 序列化输出的过程,直接看输出,类似这样:

{
    "ID": 1,
    "CreatedAt": "2020-03-11T18:26:13+08:00",
    "UpdatedAt": "2020-03-11T18:28:41+08:00",
    "DeletedAt": null,
    "Username": "test",
    "Email": "text@demo.dev",
    "Role": "member",
    "Active": 1,
    "Profile": {
        "ID": 0,
        "CreatedAt": "001-01-01T00:00:00Z",
        "UpdatedAt": "001-01-01T00:00:00Z",
        "DeletedAt": null,
        "UserID": 0,
        "Username": "",
        "Nickname": "",
        "Phone": "",
        "Gender": 0,
        "Birthday": "0001-01-01T00:00:00Z",
        "Sign": "",
        "Avatar": ""
    }
}

可以看到,虽然有输出 Profile 字段,但是里面的字段值全为零值,也就是说直接查询 User 并不会默认把关联的 Profile 一同查询出来。

可能有童鞋要问了,没查询 Profile 为什么还会输出空值的 Profile 字段呢?这是因为 JSON 序列化是按照模型的定义自动处理,User 模型中定义了 Profile 字段,如进行关联查询且能查到结果,那么就赋值给 Profile 字段,否则 Profile 依然序列化输出,只不过 Profile 里面的字段全都为空值。

接下来看下如何使用关联查询获取完整的 User 数据:

// 方式一:手动查询关联数据
var user User
// 第一步,查询用户
DB.Debug().First(&user)
// 第二步,查询关联的用户信息
// 注意,Related 方法第二个参数为 Profile 的外键
DB.Debug().Model(&user).Related(&user.Profile, "UserID")

// 方式二:也可以使用预加载方式查询关联数据
DB.Debug().Model(&user).Preload("Profile").First(&user)

可以看到,使用预加载方式语法更简练,实际上底层还是2个查询,只不过 gorm 帮我们封装好了,现在可以获取到完整的数据,类似这样:

{
    "ID": 1,
    "CreatedAt": "2020-03-11T18:26:13+08:00",
    "UpdatedAt": "2020-03-11T18:28:41+08:00",
    "DeletedAt": null,
    "Username": "test",
    "Email": "text@demo.dev",
    "Role": "member",
    "Active": 1,
    "Profile": {
        "ID": 1,
        "CreatedAt": "2020-03-11T18:26:13+08:00",
        "UpdatedAt": "2020-03-11T18:26:13+08:00",
        "DeletedAt": null,
        "UserID": 1,
        "Username": "test",
        "Nickname": "test",
        "Phone": "",
        "Gender": 0,
        "Birthday": "0001-01-01T00:00:00Z",
        "Sign": "",
        "Avatar": ""
    }
}

列表关联查询

列表的关联查询和单条关联查询类似,不过手动进行列表的关联查询很繁琐,得先查出 User 列表,然后再查询一次 Profile 列表获取对应的数据,最后整合两部分数据。直接使用预加载就很简单了,代码如下:

var users []User

// 使用预加载查询
DB.Debug().Model(&User{}).Preload("Profile").Find(&users)

输出如下:

[
    {
        "ID": 1,
        "CreatedAt": "2020-03-11T18:26:13+08:00",
        "UpdatedAt": "2020-03-11T18:28:41+08:00",
        "DeletedAt": null,
        "Username": "test",
        "Email": "text@demo.dev",
        "Role": "member",
        "Active": 1,
        "Profile": {
            "ID": 1,
            "CreatedAt": "2020-03-11T18:26:13+08:00",
            "UpdatedAt": "2020-03-11T18:26:13+08:00",
            "DeletedAt": null,
            "UserID": 1,
            "Username": "test",
            "Nickname": "test",
            "Phone": "",
            "Gender": 0,
            "Birthday": "0001-01-01T00:00:00Z",
            "Sign": "",
            "Avatar": ""
        }
    },
    ...
]

gorm 底层使用这两条查询:

SELECT * FROM `user`  WHERE `user`.`deleted_at` IS NULL

SELECT * FROM `profile`  WHERE `profile`.`deleted_at` IS
 NULL AND ((`user_id` IN (1,2,3,4,5,6)))

gorm 还在内部把两条查询的数据都整合好了,使用相当方便。

这只是一个简单的预加载应用,更多应用可参考 Gorm官方文档-预加载

小结

预加载在单条关联查询中提供了更简洁的语法,在列表关联查询中不仅解决了关联查询 N + 1 的问题,还自动整合了数据,方便快捷。

到这里,一个简单的预加载查询就完成了,但是可以发现查询输出还有很多瑕疵,如:只查询 User 也会带上空值 Profile,字段名和模型定义的一样都是首字母大写,并且时间格式不友好。

这就衍生出了几个问题:

  • 如何自定义输出结构,只输出指定字段?
  • 如何自定义字段名,并去掉空值字段?
  • 如何自定义时间格式?

下一篇将介绍如何处理查询输出,解决上述问题。

如发现任何问题,欢迎指正,谢谢观看!


参考资料: Gorm官方文档

本文出处:https://www.cnblogs.com/zhenfengxun/ 本文链接:https://www.cnblogs.com/zhenfengxun/p/12486325.html

原文出处:https://www.cnblogs.com/zhenfengxun/p/12486325.html

展开阅读全文
打赏
0
0 收藏
分享
加载中
更多评论
打赏
0 评论
0 收藏
0
分享
返回顶部
顶部