前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >学习gorm系列三:深入理解Find函数的使用

学习gorm系列三:深入理解Find函数的使用

作者头像
Go学堂
发布2023-10-09 12:46:09
9510
发布2023-10-09 12:46:09
举报
文章被收录于专栏:Go工具箱Go工具箱

大家好,我是渔夫子。

今天我们来深入理解下在gorm中,Find函数的底层逻辑。

Find函数是用来从数据库中查询的。我们通常的用法是给Find函数指定一个定义的Model对象,然后再指定对应的查询条件,这样就能查询数据了。

我们通过一个示例来查看下。首先,我们有一个数据表,并导入两条数据。如下:

代码语言:javascript
复制
CREATE TABLE `m_tests` (
  `id` bigint(20) NOT NULL AUTO_INCREMENT,
  `name` varchar(100) NOT NULL DEFAULT '',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=7 DEFAULT CHARSET=utf8mb4;

INSERT INTO test01.m_test (id,name) VALUES (1,'John'), (2,'Jack');

然后定义一个和表对应的Model结构体:

代码语言:javascript
复制
type MTest struct {
 Id   int64
 Name string `gorm:"DEFAULT:'John'"`
}

打开一个数据库连接:

代码语言:javascript
复制
  dsn := "user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local"
  db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})

通过gorm的Find函数进行查询:

代码语言:javascript
复制
var row MTest
db.Find(&row)

这样,我们就能将id=1的数据从数据库中查出并赋值到row中。这也是我们在研发中常用的做法。

但Find函数并不仅仅是如此。

深入剖析Find函数

我们先来看Find函数的原型,如下:

代码语言:javascript
复制
func (db *DB) Find(dest interface{}, conds ...interface{}) (tx *DB)

第一个参数名是dest,通过参数名可知,dest是用来接收从数据库中查询出来的数据的。从数据类型来看,是interface{},即可以是任意的数据类型。

用map来接收查询结果

在使用map接收查询结果时,就必须要明确地指定要查询的表了。如下:

代码语言:javascript
复制
var row map[string]interface{}
err := db.Find(&row).Error
fmt.Printf("debug-err:%+v\n", err)

大家看,直接给Find函数一个map类型的变量时,程序是会报错的:

代码语言:javascript
复制
unsupported data type: &map[]: Table not set, please set it like: db.Model(&user) or db.Table("users")

是说没有指定具体要查询哪张表。

所以,当接收查询结果的变量是map类型时,则需要明确地指定是哪张表。如下:

代码语言:javascript
复制
var row map[string]interface{}
// 通过model指定表
err := db.Model(MTest{}).Find(&row).Error
// 通过Table函数指定具体表名
err := db.Table("m_tests").Find(&row).Error
fmt.Printf("debug-err:%+v\n", err)

这样,我们就能把查询到的其中的一行数据储到row变量中。

那这又是为什么Find的第一个参数是对应Model类型的变量时就可以不指定表名,而是map类型时就需要明确指定表名呢?

原因在于,gorm在执行具体的查询的时候,是从指定的Model对象中来解析表的。下面是gorm中db相关的数据结构。如下:

大家看,在Statement结构中,实际上有两个字段,一个是Dest,即Find查询函数中的第一个参数,用来接收查询结果用的。第二个字段是Model,这个是和具体的数据表对应的Model结构。

在具体执行查询语句时,程序会判断Model字段是否为nil,如果是nil,则将Dest赋值给Model。然后再解析Model以便解析出对应的表。当然,在解析Model时会判断该Model的类型是否是结构体类型,以及该结构体是否有对应的表。否则,就会跑出类型不支持,没有指定具体表的错误。

用数组来接收查询结果

上面的示例中,结果都是只有一条数据。当然,也可以给Find函数指定一个数组来接收多个查询结果。如下:

代码语言:javascript
复制
var rows []MTest
err := db.Find(&rows).Error

通过指定数组,就能把表里的数据全都查询出来并用rows来接收。

在一开始的时候,我们是通过指定了一个非切片类型的变量来接收查询结果的,但只接收到了一行数据。我们通过将两者的sql语句打印出来看看有什么区别。

代码语言:javascript
复制
 var row MTest

 row_sql := db.ToSQL(func(tx *gorm.DB) *gorm.DB {
  return tx.Find(&row)
 })

 fmt.Printf("接收的sql语句:%s\n", row_sql) //SELECT * FROM `m_tests`

 var rows []MTest
 rows_sql := db.ToSQL(func(tx *gorm.DB) *gorm.DB {
  return tx.Find(&rows)
 })

 fmt.Printf("切片接收的sql语句:%s\n", rows_sql) //SELECT * FROM `m_tests`

有没有发现两个sql语句是一样的。

这里要注意:Find函数如果在没有指定Where条件的情况下,都是全表扫描,最后根据接收参数是切片还是单条数组来提取扫描的结果而已。

总结

通过本文,我们了解到Find函数的第一个参数是接收查询结果的参数,而并不是通过该参数指定的数据表。当没有显式的指定Model时,gorm的查询会自动地将Dest参数值赋值给Model。然后,查询函数会从Model解析表名。如果从Model中解析不到对应的表名,就会报错。最后,我们分析了Find函数查询一行和多行数据的区别。其本质上是扫描符合条件的所有数据,最后根据是否是切片类型来返回数据而已。

特别说明:你的关注,是我写下去的最大动力。

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2023-10-07,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 Go学堂 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 深入剖析Find函数
    • 用map来接收查询结果
      • 用数组来接收查询结果
      • 总结
      领券
      问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档