请教一下 Gorm 的 Count 问题

Count 方法这里怎么会多出一个 and 条件，导致结果不正确呢，我代码有问题么？

go 新手求指点

代码片段

util 中部分定义：

package util

type Result struct {
	Code uint        `json:"code"`
	Data interface{} `json:"data"`
}

// NewResult creates a result with Code=0, Msg="", Data=nil.
func NewResult() *Result {
	return &Result{
		Code: 200,
		Data: nil,
	}
}
type JobRequestParam struct {
	ID      []uint `form:"id" binding:"required"`
	OpsID   uint   `form:"ops_id" binding:"required"`
	RunTime string `form:"run_time"`
}

查询方法

package service

import (
	"OpsSimon/util"
	"log"
	"sync"
)

// AssetsService 资产对象
type AssetsService struct {
	mutex *sync.Mutex
}

// Assets 资产实例
var Assets = &AssetsService{
	mutex: &sync.Mutex{},
}
// GetList 列表查询
func (srv *AssetsService) GetList(result interface{}, id []uint) (total int) {
	err := db.Where(id).Find(result).Count(&total).Error
	if err != nil {
		log.Println(err.Error())
	}
	return total
}

执行

func doJob(c *gin.Context) {
	result := util.NewResult()
	defer c.JSON( http.StatusOK, result)

	var arg util.JobRequestParam
	err := c.BindJSON(&arg)
	if err != nil {
		result.Code = http.StatusBadRequest
		result.Data = err.Error()
		return
	}
	log.Println(arg.ID)
	log.Println(arg.OpsID)
	log.Println("len:", len(arg.ID))

	var host model.AssetsHosts
	total := service.Assets.GetList(&host, arg.ID)
	log.Println("total:", total)

	if total != len(arg.ID) {
		result.Code = http.StatusNotFound
		result.Data = "Some host not found!"
		return
	}

}

JimmyTinsley

2019-09-05 09:25:31 +08:00

@Ansen #5 用 goland 帮你 debug 了一下
如果按照你的写法 `db.Where(id).Find(result).Count(&total).Error` 会产生两次 sql 查询, 就像你日志里面打印的两条一样. Find(result)这个方法本身会 `select * from xxx where xxx in xxx` 查询一遍结果, 然后当你链式调用到 Count(&total)的时候, 实际上这个时候的 DB 对象本身是含有 Value 了的, Value.ID 是根据你 where 条件的查询得到结果的最后一条的 ID.
没有再深入看 gorm 里面的代码, 我猜里面是有一个循环赋值的过程, 当你用 Count(&total)进行查询的时候, DB 对象已经被 Find(result)循环赋值完, 成为 Find(result)结果的最后一条, 所以会多出来一个 and 条件. 如果你试试把 arg.ID 换成[1,2,3,4,5], 那这个 where 条件应该会变成 and id = 5.
表述的可能不太清楚, 你可以自己试试 debug, 把断点打在 Count 方法的`return s.NewScope(s.Value).count(value).db
`这一行就可以了~