gorm学习系列三:真正掌握Find函数的用法

2023年 10月 8日 85.4k 0

大家好,我是渔夫子。

欢迎关注 Go学堂 送《go常见错误100例案例解析》pdf文档

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

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

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

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结构体:

type MTest struct {
	Id   int64
	Name string `gorm:"DEFAULT:'John'"`
}

打开一个数据库连接:

  dsn := "user:pass@tcp(127.0.0.1:3306)/dbname?&parseTime=True&loc=Local"
  db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})

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

var row MTest
db.Find(&row)

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

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

深入剖析Find函数

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

func (db *DB) Find(dest interface{}, conds ...interface{}) (tx *DB)

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

用map来接收查询结果

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

var row map[string]interface{}
err := db.Find(&row).Error
fmt.Printf("debug-err:%+vn", err)

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

unsupported data type: &map[]: Table not set, please set it like: db.Model(&user) or db.Table("users")

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

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

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:%+vn", err)

这样,我们就能把id=1的数据查询并存储到row变量中。

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

原因在于,gorm在执行具体的查询的时候,是从指定的Model对象中来解析表的。下面是gorm中db相关的数据结构。如下:
image.png
大家看,在Statement结构中,实际上有两个字段,一个是Dest,即Find查询函数中的第一个参数,用来接收查询结果用的。第二个字段是Model,这个是和具体的数据表对应的Model结构。

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

用数组来接收查询结果

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

var rows []MTest
err := db.Find(&rows).Error

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

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

	var row MTest

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

	fmt.Printf("接收的sql语句:%sn", 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语句:%sn", rows_sql) //SELECT * FROM `m_tests`

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

这里要注意:Find函数如果在没有指定Where条件的情况下,都是全表扫描。无论是用切片接受数据还是用非切片接收一条数据。

总结

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

相关文章

JavaScript2024新功能:Object.groupBy、正则表达式v标志
PHP trim 函数对多字节字符的使用和限制
新函数 json_validate() 、randomizer 类扩展…20 个PHP 8.3 新特性全面解析
使用HTMX为WordPress增效:如何在不使用复杂框架的情况下增强平台功能
为React 19做准备:WordPress 6.6用户指南
如何删除WordPress中的所有评论

发布评论