fix: count() when use group by
and only find one record
This commit is contained in:
parent
cff7845e58
commit
b90f112239
@ -422,6 +422,11 @@ func (db *DB) Count(count *int64) (tx *DB) {
|
|||||||
if tx.RowsAffected != 1 {
|
if tx.RowsAffected != 1 {
|
||||||
*count = tx.RowsAffected
|
*count = tx.RowsAffected
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if _, ok := db.Statement.Clauses["GROUP BY"]; ok && tx.RowsAffected == 1 {
|
||||||
|
*count = tx.RowsAffected
|
||||||
|
}
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -134,4 +134,14 @@ func TestCount(t *testing.T) {
|
|||||||
t.Fatalf("Count should be 3, but got count: %v err %v", count10, err)
|
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)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user