有效解决Gorm中使用Count后关联查询的问题

2023-09-08 07:03:49

前言

在使用Gorm进行数据库操作时，我们经常需要进行关联查询，以便获取相关联的数据。同时，为了提高查询效率，我们也经常需要使用Count来统计记录数。然而，在某些情况下，如果我们不注意Gorm的查询机制，可能会导致关联查询和Count无法正常工作。

问题

当我们在Go中使用Gorm进行多表join关联查询的时候，如果还有分页的需求，那么可能会是这样写：

func main() {
	db, err := gorm.Open("mysql", "user:password@tcp(localhost:3306)/gorm_test?charset=utf8&parseTime=True&loc=Local")
	if err != nil {
		panic(err)
	}
	defer db.Close()

	// 定义模型
	type User struct {
		ID   uint
		Name string
	}

	type Order struct {
		ID    uint
		UserID uint
	}

	// 关联查询
	var users []User
	db.Model(&User{}).Preload("Orders").Count(&users).Limit(10).Offset(0).Find(&users)

	// 打印结果
	for _, user := range users {
		fmt.Println(user.Name)
		for _, order := range user.Orders {
			fmt.Println(order.ID)
		}
	}
}

这样，Count会计算出值，而再查询数据就会出现不存在的问题。

解决方案

要解决这个问题，我们需要理解Gorm的查询机制。Gorm在执行查询时，会先执行Count查询，然后再执行关联查询。如果我们在Count查询中使用了关联查询，那么Gorm就会先执行关联查询，然后再执行Count查询。这会导致关联查询的结果不正确，从而导致分页查询失败。

为了解决这个问题，我们可以将Count查询和关联查询分开执行。具体做法是，先执行Count查询，然后再执行关联查询。这样，Gorm就会先执行Count查询，然后再执行关联查询，从而保证关联查询的结果正确。

func main() {
	db, err := gorm.Open("mysql", "user:password@tcp(localhost:3306)/gorm_test?charset=utf8&parseTime=True&loc=Local")
	if err != nil {
		panic(err)
	}
	defer db.Close()

	// 定义模型
	type User struct {
		ID   uint
		Name string
	}

	type Order struct {
		ID    uint
		UserID uint
	}

	// Count查询
	var count int64
	db.Model(&User{}).Count(&count)

	// 关联查询
	var users []User
	db.Model(&User{}).Preload("Orders").Limit(10).Offset(0).Find(&users)

	// 打印结果
	for _, user := range users {
		fmt.Println(user.Name)
		for _, order := range user.Orders {
			fmt.Println(order.ID)
		}
	}
}