HiPoweredDev
/
xorm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix test

This commit is contained in:
Lunny Xiao 2019-11-12 13:59:01 +08:00
parent 6ff3fb9289
commit a5ac0dfdf3
No known key found for this signature in database
GPG Key ID: C3B7C91B632F738A
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
session_iterate.go
View File

@ -4,7 +4,9 @@
package xorm package xorm
import "reflect" import (
"reflect"
)
// IterFunc only use by Iterate // IterFunc only use by Iterate
type IterFunc func(idx int, bean interface{}) error type IterFunc func(idx int, bean interface{}) error
@ -74,9 +76,9 @@ func (session *Session) bufferIterate(bean interface{}, fun IterFunc) error {
session.autoResetStatement = true session.autoResetStatement = true
}() }()
for { for bufferSize > 0 {
slice := reflect.New(sliceType) slice := reflect.New(sliceType)
if err := session.Limit(bufferSize, start).find(slice.Interface(), bean); err != nil { if err := session.NoCache().Limit(bufferSize, start).find(slice.Interface(), bean); err != nil {
return err return err
} }
@ -87,13 +89,13 @@ func (session *Session) bufferIterate(bean interface{}, fun IterFunc) error {
idx++ idx++
} }
if slice.Elem().Len() == 0 || idx == limit { if bufferSize > slice.Elem().Len() {
break break
} }
start = start + slice.Elem().Len() start = start + slice.Elem().Len()
if limit > 0 && idx+bufferSize > limit { if limit > 0 && start+bufferSize > limit {
bufferSize = limit - idx bufferSize = limit - start
} }
} }