大家好,我是渔夫子。
欢迎关注 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相关的数据结构。如下:
大家看,在Statement结构中,实际上有两个字段,一个是Dest,即Find查询函数中的第一个参数,用来接收查询结果用的。第二个字段是Model,这个是和具体的数据表对应的Model结构。
在具体执行查询语句时,程序会判断Model字段是否为nil,如果是nil,则将Dest赋值给Model。然后再解析Model以便解析出对应的表。 当然,在解析Model时会判断该Model的类型是否是结构体类型,以及该结构体是否有对应的表。否则,就会跑出类型不支持,没有指定具体表的错误。
用数组来接收查询结果
上面的示例中,结果都是只有一条数据。当然,也可以给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函数查询一行和多行数据的区别。其本质上是扫描符合条件的所有数据,最后根据是否是切片类型来返回数据而已。