Cleanup Model after Count

2020-08-19 20:30:39 +08:00 · 2020-08-19 20:30:39 +08:00 · 528e5ba5c4
commit 528e5ba5c4
parent 3313c11888
1 changed files with 3 additions and 0 deletions
--- a/finisher_api.go
+++ b/finisher_api.go
@ -289,6 +289,9 @@ func (db *DB) Count(count *int64) (tx *DB) {
 	tx = db.getInstance()
 	if tx.Statement.Model == nil {
 		tx.Statement.Model = tx.Statement.Dest
+		defer func() {
+			tx.Statement.Model = nil
+		}()
 	}

 	if len(tx.Statement.Selects) == 0 {