fix: count() when use group by and only find one record

This commit is contained in:
李龙 2021-11-25 12:57:16 +08:00
parent cff7845e58
commit b90f112239
2 changed files with 15 additions and 0 deletions

View File

@ -422,6 +422,11 @@ func (db *DB) Count(count *int64) (tx *DB) {
if tx.RowsAffected != 1 {
*count = tx.RowsAffected
}
if _, ok := db.Statement.Clauses["GROUP BY"]; ok && tx.RowsAffected == 1 {
*count = tx.RowsAffected
}
return
}

View File

@ -134,4 +134,14 @@ func TestCount(t *testing.T) {
t.Fatalf("Count should be 3, but got count: %v err %v", count10, err)
}
var count11 int64
sameUsers := make([]*User, 0)
for i := 0; i < 3; i++ {
sameUsers = append(sameUsers, GetUser("count-4", Config{}))
}
DB.Create(sameUsers)
if err := DB.Model(&User{}).Where("name = ?", "count-4").Group("name").Count(&count11).Error; err != nil || count11 != 1 {
t.Fatalf("Count should be 3, but got count: %v err %v", count11, err)
}
}