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

问题描述

当我们 在go中使用gorm进行多表join关联查询的时候

如果还有分页的需求

那么可能会是这样写

package mainimport ("gorm.io/driver/mysql""gorm.io/gorm"ormLogger "gorm.io/gorm/logger""time"
)func main() {type Detail struct {UesrId int    `json:"user_id"` // 自增 idAge    int    `json:"age"`     // 年龄Email  string `json:"email"`   // 邮箱}type User struct {Id     int    `json:"id" gorm:"primaryKey"` // 自增 idName   string `json:"name"`                 // 名字Detail `gorm:"foreignKey:UesrId"`}type MysqlConfig struct {MysqlUrl     stringLogger       ormLogger.WriterMaxIdleCount intMaxOpen      intMaxLifetime  time.DurationLogLevel     ormLogger.LogLevel}var c MysqlConfigDB, err := gorm.Open(mysql.Open(c.MysqlUrl))if err != nil {panic("GORM 连接失败," + err.Error())}tx := DB.Model(&User{}).Joins("Detail")var count int64tx.Count(&count)var data []Usertx.Limit(GetLimit()).Offset(GetOffset()).Find(&data)}

这样count会计算出值,
而 再查询数据就 会出现数据为空的情况

问题分析:

打印sql出来

SELECT count(*) FROM `users` SELECT `users`.`id`,`users`.`name`,`users`.`uesr_id`,`users`.`age`,`users`.`email` FROM `users` Detail LIMIT 1 OFFSET 3

通过查询Count方法的源码我们发现

tx.Statement.AddClause(clause.Select{Expression: clause.Expr{SQL: "count(*)"}})

这里如果调用count方法,gorm会把你的sql的select的字段转换成 count*
所以,通过join关联查询的方式不可以进行对应字段的映射了

解决思路

我们执行查询和执行记数的tx,使用两个就好了
因为go语言是引用类型传递,所以该怎么进行拷贝tx对象呢

查询gorm相关源码发现,session()的源码里包含

// Session create new db session
func (db *DB) Session(config *Session) *DB {var (txConfig = *db.Configtx       = &DB{Config:    &txConfig,Statement: db.Statement,Error:     db.Error,clone:     1,})if config.Context != nil || config.PrepareStmt || config.SkipHooks {tx.Statement = tx.Statement.clone()tx.Statement.DB = tx}

func (stmt *Statement) clone() *Statement {copy(newStmt.Joins, stmt.Joins)...// 在这里执行了copy方法
}

所以我们可以利用gorm中的session功能深拷贝一个 tx对象,即:tx2 := tx.session()

package mainimport ("gorm.io/driver/mysql""gorm.io/gorm"ormLogger "gorm.io/gorm/logger""time"
)func main() {type Detail struct {UesrId int    `json:"user_id"` // 自增 idAge    int    `json:"age"`     // 年龄Email  string `json:"email"`   // 邮箱}type User struct {Id     int    `json:"id" gorm:"primaryKey"` // 自增 idName   string `json:"name"`                 // 名字Detail `gorm:"foreignKey:UesrId"`}type MysqlConfig struct {MysqlUrl     stringLogger       ormLogger.WriterMaxIdleCount intMaxOpen      intMaxLifetime  time.DurationLogLevel     ormLogger.LogLevel}var c MysqlConfigDB, err := gorm.Open(mysql.Open(c.MysqlUrl))if err != nil {panic("GORM 连接失败," + err.Error())}tx := DB.Model(&User{}).Joins("Detail")tx2 := tx.Session(&gorm.Session{})var count int64tx2.Count(&count)var data []Usertx.Limit(GetLimit()).Offset(GetOffset()).Find(&data)分别打印出内存地址print(tx)	// 0x14000282150print(tx2)	// 0x140002821e0
}

这样通过gorm中的session可以深拷贝出一个 gorm对象
执行Count后再执行查询不会受影响