文章目录
- 初始数据表
- 一般查询
- Where条件
- 普通SQL查询
- 查询 Struct & Map
- Not 条件
- Or 条件
- 内联条件
- 额外查询选项
- FirstOrInit
- Attrs
- Assign
- FirstOrCreate
- Attrs
- Assign
初始数据表
一般查询
var p1, p2, p3, p4 connect.PersonInfo
var ps []connect.PersonInfo
// 把主键升序,获取表的第一条记录
connect.DB.First(&p1)
fmt.Printf("%#v\n", p1)
// 主键不排序,获取表的第一条记录
connect.DB.Take(&p2)
fmt.Printf("%#v\n", p2)
// 把主键降序,获取表的第一条记录
connect.DB.Last(&p3)
fmt.Printf("%#v\n", p3)
// 查询所有的记录
connect.DB.Find(&ps)
fmt.Printf("%#v\n", ps)
// 查询指定的某条记录(仅当主键为整型时可用)
connect.DB.First(&p4, 3)
fmt.Printf("%#v\n", p4)
Where条件
普通SQL查询
var p1 connect.PersonInfo
var ps1, ps2, ps3, ps4, ps5 []connect.PersonInfo
// 查询username为abc的第一个记录
connect.DB.Where("username = ?", "abc").First(&p1)
//SELECT * FROM `person_info` WHERE username = 'abc' ORDER BY `person_info`.`id` LIMIT 1
// 查询username不等于drg的所有记录
connect.DB.Where("username <> ?", "drg").Find(&ps1)
//SELECT * FROM `person_info` WHERE username <> 'drg'
// 查询username为abc或者xyz的所有记录
connect.DB.Where("username IN ?", []string{"abc", "xyz"}).Find(&ps2)
//SELECT * FROM `person_info` WHERE username IN ('abc','xyz')
// 查询username里面有字串"e"的所有记录
connect.DB.Where("username LIKE ?", "%e%").Find(&ps3)
//SELECT * FROM `person_info` WHERE username LIKE '%e%'
// 查询username里面有字串"e"的所有记录 并且 age大于20的记录
connect.DB.Where("username LIKE ? AND age >= ?", "%e%", "20").Find(&ps4)
//SELECT * FROM `person_info` WHERE username LIKE '%e%' AND age >= '20'
// 查询age再19到23之间的记录
connect.DB.Where("age BETWEEN ? AND ?", 19, 23).Find(&ps5)
//SELECT * FROM `person_info` WHERE age BETWEEN 19 AND 23
查询 Struct & Map
var p1 connect.PersonInfo
var ps1, ps2 []connect.PersonInfo
// Struct
connect.DB.Where(&connect.PersonInfo{UserName: "xyz", Age: 22}).First(&p1)
//SELECT * FROM `person_info` WHERE `person_info`.`username` = 'xyz'
//AND `person_info`.`age` = 22 ORDER BY `person_info`.`id` LIMIT 1
fmt.Printf("%#v\n", p1)
// Map
connect.DB.Where(map[string]interface{}{"username": "edg", "age": 24}).Find(&ps1)
//SELECT * FROM `person_info` WHERE `age` = 24 AND `username` = 'edg'
fmt.Printf("%#v\n", ps1)
// 主键的切片
connect.DB.Where([]int64{4, 5, 6}).Find(&ps2)
//SELECT * FROM `person_info` WHERE `person_info`.`id` IN (4,5,6)
fmt.Printf("%#v\n", ps2)
运行结果如下:
注意: 当通过结构体进行查询时,GORM将会只通过非零值字段查询,这意味着如果你的字段值为0
,''
,false
或者其他零值
时,将不会被用于构建查询条件,例如:
type User struct {
Id uint64
UserName string
PassWord string
Age int
}
var u connect.User
connect.DB.Where(&connect.User{UserName: "bing", Age: 0}).Find(&u)
上面代码的Age字段由于为0,会被忽略,导致Age字段不会参与查询,所以底层执行的SQL语句为SELECT * FROM User WHERE name = “bing”;可以使用指针或实现 Scanner/Valuer 接口来避免这个问题。具体步骤详见:GORM—创建-黎明的博客-CSDN,这篇博客的最后有详细的解决步骤。
Not 条件
var p1, p2, p3, p4, p5 connect.PersonInfo
var ps1 []connect.PersonInfo
connect.DB.Not("username", "dyg").First(&p1)
//SELECT * FROM `person_info` WHERE `username` <> 'dyg' ORDER BY `person_info`.`id` LIMIT 1
connect.DB.Not("username", []string{"abc", "xyz", "dyg"}).Find(&ps1)
//SELECT * FROM `person_info` WHERE `username` NOT IN ('abc','xyz','dyg')
connect.DB.Not([]int64{1, 2, 3}).First(&p2)
//SELECT * FROM `person_info` WHERE `person_info`.`id` NOT IN (1,2,3) ORDER BY `person_info`.`id` LIMIT 1
connect.DB.Not([]int64{}).First(&p3)
//SELECT * FROM `person_info` ORDER BY `person_info`.`id` LIMIT 1
connect.DB.Not("username = ?", "tes").First(&p4)
//SELECT * FROM `person_info` WHERE NOT username = 'tes' ORDER BY `person_info`.`id` LIMIT 1
connect.DB.Not(connect.PersonInfo{UserName: "edg"}).First(&p5)
//SELECT * FROM `person_info` WHERE `person_info`.`username` <> 'edg' ORDER BY `person_info`.`id` LIMIT 1
Or 条件
var p1, p2, p3 connect.PersonInfo
connect.DB.Where("age = ?", "23").Or("username = ?", "xyz").Find(&p1)
//SELECT * FROM `person_info` WHERE age = '23' OR username = 'xyz'
fmt.Printf("%#v\n", p1)
connect.DB.Where("username = 'tes'").Or(connect.PersonInfo{UserName: "abc"}).Find(&p2)
//SELECT * FROM `person_info` WHERE username = 'tes' OR `person_info`.`username` = 'abc'
fmt.Printf("%#v\n", p2)
connect.DB.Where("username = 'dyg'").Or(map[string]interface{}{"password": "12345"}).Find(&p3)
//SELECT * FROM `person_info` WHERE username = 'dyg' OR `password` = '12345'
fmt.Printf("%#v\n", p3)
运行结果如下:
内联条件
作用与Where查询类似,当内联条件与多个立即执行方法一起使用时, 内联条件不会传递给后面的立即执行方法。
var p1, p2 connect.PersonInfo
var ps1, ps2, ps3, ps4 []*connect.PersonInfo
// 根据主键获取记录 (只适用于整形主键)
connect.DB.First(&p1, 5)
//SELECT * FROM `person_info` WHERE `person_info`.`id` = 5 ORDER BY `person_info`.`id` LIMIT 1
// 根据主键获取记录, 如果它是一个非整形主键
connect.DB.First(&p2, "id = ?", "string_primary_key")
//SELECT * FROM `person_info` WHERE id = 'string_primary_key' ORDER BY `person_info`.`id` LIMIT 1
connect.DB.Find(&ps1, "username = ?", "xyz")
//SELECT * FROM `person_info` WHERE username = 'xyz'
connect.DB.Find(&ps2, "username <> ? AND age > ?", "abc", 20)
//SELECT * FROM `person_info` WHERE username <> 'abc' AND age > 20
connect.DB.Find(&ps3, connect.PersonInfo{Age: 20})
//SELECT * FROM `person_info` WHERE `person_info`.`age` = 20
connect.DB.Find(&ps4, map[string]interface{}{"age": 20})
//SELECT * FROM `person_info` WHERE `age` = 20
额外查询选项
在SQL查询时添加额外的 SQL 操作。例如在查询的时候,执行加锁、排序、分页等。下面的例子的作用是查询 id 为 10 的用户数据,并对查询结果加锁,防止其他事务修改这些数据。查询结果将存储到变量 p 中。
tx := connect.DB.Begin() //开启事务
defer func() {
if r := recover(); r != nil {
tx.Rollback()
}
}()
var p connect.PersonInfo
connect.DB.Set("gorm:query_option", "FOR UPDATE").First(&p, 6)
//SELECT * FROM `person_info` WHERE `person_info`.`id` = 6 ORDER BY `person_info`.`id` LIMIT 1
tx.Commit() //提交事务
fmt.Printf("%#v\n", p)
运行结果如下:
需要注意的是,FOR UPDATE 选项只在事务中生效,如果没有使用事务,那么这个选项将被忽略。如果要使用事务,可以使用 Begin 方法开启事务,然后在事务中执行查询操作。相应的,如果要执行排序操作可以使用如下写法:
//升序(ASC) 降序(DESC)
connect.DB.Set("gorm:query_option", "ORDER BY age ASC").Find(&p)
在实际开发中,通常需要动态设置排序的一些属性,可以根据具体情况动态构建选项值。例如,可以使用 fmt.Sprintf 函数将排序字段和排序方式作为参数传递给 Set 方法,动态构建排序条件。代码如下:
sortField := "age"
sortOrder := "ASC"
connect.DB.Set("gorm:query_option", fmt.Sprintf("ORDER BY %s %s", sortField, sortOrder)).Find(&p)
那如果要对查询结果进行分页操作呢?写法如下:
// 查询第 2 页的 10 条数据
connect.DB.Set("gorm:query_option", "LIMIT 10 OFFSET 10").Find(&p)
// 动态设置
pageSize := 10
pageNum := 2
offset := (pageNum - 1) * pageSize
connect.DB.Set("gorm:query_option", fmt.Sprintf("LIMIT %d OFFSET %d", pageSize, offset)).Find(&p)
//对于分页操作,GORM 框架提供了更为方便的方法,可以使用 Limit 和 Offset 方法直接设置分页条件
pageSize := 10
pageNum := 2
connect.DB.Limit(pageSize).Offset((pageNum - 1) * pageSize).Find(&p)
这三种写法根据实际应用场景来选择。
FirstOrInit
根据给定的条件初始化一个新的对象 (仅支持 struct 和 map 查询条件),从数据库中查询符合条件的记录,如果找到了,则返回该记录;如果没有找到,则返回该查询条件。示例代码如下:
var p1, p2, p3, p4 connect.PersonInfo
//没找到的情况
connect.DB.FirstOrInit(&p1, connect.PersonInfo{UserName: "xyg"})
// SELECT * FROM `person_info` WHERE `person_info`.`username` = 'xyg' ORDER BY `person_info`.`id` LIMIT 1
fmt.Printf("%#v\n", p1) //p1为空
//找到了的情况
connect.DB.FirstOrInit(&p2, connect.PersonInfo{UserName: "abc"})
//SELECT * FROM `person_info` WHERE `person_info`.`username` = 'abc' ORDER BY `person_info`.`id` LIMIT 1
fmt.Printf("%#v\n", p2)
connect.DB.Where(connect.PersonInfo{Age: 22}).FirstOrInit(&p3)
//SELECT * FROM `person_info` WHERE `person_info`.`age` = 22 ORDER BY `person_info`.`id` LIMIT 1
fmt.Printf("%#v\n", p3)
connect.DB.FirstOrInit(&p4, map[string]interface{}{"password": "0395"})
//SELECT * FROM `person_info` WHERE `password` = '0395' ORDER BY `person_info`.`id` LIMIT 1
fmt.Printf("%#v\n", p4)
运行结果如下:
Attrs
Attrs方法只是为模型对象设置属性值(可以理解为一个临时变量),但并不会将这些属性值保存到数据库中。
var p1, p2, p3 connect.PersonInfo
// 数据库中并不存在UserName为"bing"的记录
// 未找到(两种写法)
connect.DB.Where(connect.PersonInfo{UserName: "bing"}).Attrs(connect.PersonInfo{Age: 20}).FirstOrInit(&p1)
//SELECT * FROM `person_info` WHERE `person_info`.`username` = 'bing' ORDER BY `person_info`.`id` LIMIT 1
fmt.Printf("%#v\n", p1)
connect.DB.Where(connect.PersonInfo{UserName: "bing"}).Attrs("age",28).FirstOrInit(&p2)
//SELECT * FROM `person_info` WHERE `person_info`.`username` = 'bing' ORDER BY `person_info`.`id` LIMIT 1
fmt.Printf("%#v\n", p2)
// 找到()
connect.DB.Where(connect.PersonInfo{UserName: "xyz"}).Attrs(connect.PersonInfo{Age: 30}).FirstOrInit(&p3)
//SELECT * FROM `person_info` WHERE `person_info`.`username` = 'xyz' ORDER BY `person_info`.`id` LIMIT 1
fmt.Printf("%#v\n", p3)
运行结果如下:
从上面的例子中可以得出以下结论:
- 如果使用FirstOrInit方法没有查找到相应记录,再搭配上Attrs方法,那么返回的结果值是一个临时的结构体对象,这个临时的结构体对象的一些属性会被Attrs方法更新。并且这个临时的结构体对象并不会插入到数据库中,程序结束后就会被销毁。
- 如果使用FirstOrInit方法找到了相应的记录,则会返回相应的记录,并且Attrs方法设置的属性值不会生效。
Assign
不管记录是否找到,都将参数赋值给相应的记录。
var p1, p2 connect.PersonInfo
// 未找到
connect.DB.Where(connect.PersonInfo{UserName: "bing"}).Assign(connect.PersonInfo{Age: 29}).FirstOrInit(&p1)
//SELECT * FROM `person_info` WHERE `person_info`.`username` = 'bing' ORDER BY `person_info`.`id` LIMIT 1
fmt.Printf("%#v\n", p1)
// 找到
connect.DB.Where(connect.PersonInfo{UserName: "dyg"}).Assign(connect.PersonInfo{Age: 35}).FirstOrInit(&p2)
//SELECT * FROM `person_info` WHERE `person_info`.`username` = 'dyg' ORDER BY `person_info`.`id` LIMIT 1
fmt.Printf("%#v\n", p2)
运行结果如下:
小总结:
在使用FirstOrInit这个方法的大前提下,不管怎么写,都不会把返回的数据插入到数据库,因为FirstOrInit这个方法只是用来做查询操作。在搭配上Attrs和Assign这两个方法,理解起来就是:
- 当使用Attrs方法时:
- 没找到记录:在Attrs方法中设置的属性会更新临时结构体对象的属性。
- 找到记录:Attrs方法中设置的属性会失效
- 当使用Assign方法时;
- 不管有没有找到相应的记录,都会设置相应的属性。
FirstOrCreate
FirstOrCreate方法用于根据指定的条件查找数据库中的一条数据记录,如果找到,则返回该记录;如果没有找到,则创建一条新的数据记录,并将其保存到数据库中。
p := connect.PersonInfo{
Id: 8,
UserName: "ag",
Password: "9843",
Age: 25,
}
//如果数据库中没有UserName为"ag"的就插入p这条数据
connect.DB.Where(connect.PersonInfo{UserName: "ag"}).FirstOrCreate(&p)
fmt.Printf("%#v\n", p)
运行完成后,查询结果如下:
可以看到,其底层分为了两条sql语句,由于数据库中没有UserName为"ag"的记录,所以就把p对象的数据插入到数据库中。
Attrs
Attrs方法用于为要创建或更新的数据记录指定属性值,但不会将其保存到数据库中。
//能找到
p1 := connect.PersonInfo{Id: 4, Password: "6666"}
connect.DB.Attrs(connect.PersonInfo{UserName: "bing", Age: 23}).FirstOrCreate(&p1)
//SELECT * FROM `person_info` WHERE `person_info`.`id` = 4 ORDER BY `person_info`.`id` LIMIT 1
fmt.Printf("%#v\n", p1)
//找不到
p2 := connect.PersonInfo{Id: 9, Password: "8888"}
connect.DB.Attrs(connect.PersonInfo{UserName: "bing", Age: 23}).FirstOrCreate(&p2)
//SELECT * FROM `person_info` WHERE `person_info`.`id` = 9 ORDER BY `person_info`.`id` LIMIT 1
//INSERT INTO `person_info` (`username`,`password`,`age`,`id`) VALUES ('bing','8888',23,9)
fmt.Printf("%#v\n", p2)
运行结果如下:
如果FirstOrCreate方法能找到相应的记录,则Attrs方法设置的属性将会无效。如果FirstOrCreate方法不能找到相应的记录,则底层会使用insert的SQL语句创建一条记录,并添加Attrs方法设置的属性到新纪录中。
Assign
不管记录是否找到,都将Attrs方法设置的属性赋值给 struct 并保存至数据库。
var p1, p2 connect.PersonInfo
//未找到
connect.DB.Where(connect.PersonInfo{UserName: "hui"}).Assign(connect.PersonInfo{Age: 26, Id: 10, Password: "83492"}).FirstOrCreate(&p1)
//SELECT * FROM `person_info` WHERE `person_info`.`username` = 'hui' ORDER BY `person_info`.`id` LIMIT 1
//INSERT INTO `person_info` (`username`,`password`,`age`,`id`) VALUES ('hui','83492',26,10)
fmt.Printf("%#v\n", p1)
// 找到
connect.DB.Where(connect.PersonInfo{UserName: "bing"}).Assign(connect.PersonInfo{Age: 30}).FirstOrCreate(&p2)
//SELECT * FROM `person_info` WHERE `person_info`.`username` = 'bing' ORDER BY `person_info`.`id` LIMIT 1
//UPDATE `person_info` SET `age`=30 WHERE `person_info`.`username` = 'bing' AND `id` = 9
fmt.Printf("%#v\n", p2)
运行结果如下:
小总结:
在使用FirstOrCreate这个方法的大前提下,在搭配上Attrs和Assign这两个方法,理解起来就是:
- 当使用Attrs方法时:
- 没找到记录:往数据库里面添加一条数据,Attrs方法的作用就是给这条数据添加一些属性。
- 找到记录:返回数据库中的数据记录,并且Attrs方法中设置的属性无效。
- 当使用Assign方法时;
- 没找到记录:往数据库里面添加一条数据,Assign方法的作用就是给这条数据添加一些属性(和Attrs一样)。
- 找到记录:按照Assign方法中设置的属性,更新数据库里面的数据。