GORM---初级查询

初始数据表

一般查询

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)

运行结果如下:

image-20230604194900972

注意: 当通过结构体进行查询时,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)

运行结果如下:

image-20230604194434379

内联条件

作用与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)

运行结果如下:

image-20230604212637094

需要注意的是,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)

运行结果如下:

image-20230606095320631

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)

运行结果如下:

image-20230606102529977

从上面的例子中可以得出以下结论:

  1. 如果使用FirstOrInit方法没有查找到相应记录,再搭配上Attrs方法,那么返回的结果值是一个临时的结构体对象,这个临时的结构体对象的一些属性会被Attrs方法更新。并且这个临时的结构体对象并不会插入到数据库中,程序结束后就会被销毁。
  2. 如果使用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)

运行结果如下:

image-20230606105749398

小总结:

在使用FirstOrInit这个方法的大前提下,不管怎么写,都不会把返回的数据插入到数据库,因为FirstOrInit这个方法只是用来做查询操作。在搭配上Attrs和Assign这两个方法,理解起来就是:

  1. 当使用Attrs方法时:
    • 没找到记录:在Attrs方法中设置的属性会更新临时结构体对象的属性。
    • 找到记录:Attrs方法中设置的属性会失效
  2. 当使用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)

运行完成后,查询结果如下:

image-20230606111351099

可以看到,其底层分为了两条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)

运行结果如下:

image-20230606113811362

如果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)

运行结果如下:

image-20230606133605008

小总结:

在使用FirstOrCreate这个方法的大前提下,在搭配上Attrs和Assign这两个方法,理解起来就是:

  1. 当使用Attrs方法时:
    • 没找到记录:往数据库里面添加一条数据,Attrs方法的作用就是给这条数据添加一些属性。
    • 找到记录:返回数据库中的数据记录,并且Attrs方法中设置的属性无效。
  2. 当使用Assign方法时;
    • 没找到记录:往数据库里面添加一条数据,Assign方法的作用就是给这条数据添加一些属性(和Attrs一样)。
    • 找到记录:按照Assign方法中设置的属性,更新数据库里面的数据。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值