aboutsummaryrefslogtreecommitdiff
path: root/vendor/github.com/go-xorm/xorm/session_iterate.go
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/go-xorm/xorm/session_iterate.go')
-rw-r--r--vendor/github.com/go-xorm/xorm/session_iterate.go96
1 files changed, 0 insertions, 96 deletions
diff --git a/vendor/github.com/go-xorm/xorm/session_iterate.go b/vendor/github.com/go-xorm/xorm/session_iterate.go
deleted file mode 100644
index 071fce49..00000000
--- a/vendor/github.com/go-xorm/xorm/session_iterate.go
+++ /dev/null
@@ -1,96 +0,0 @@
-// Copyright 2016 The Xorm Authors. All rights reserved.
-// Use of this source code is governed by a BSD-style
-// license that can be found in the LICENSE file.
-
-package xorm
-
-import "reflect"
-
-// IterFunc only use by Iterate
-type IterFunc func(idx int, bean interface{}) error
-
-// Rows return sql.Rows compatible Rows obj, as a forward Iterator object for iterating record by record, bean's non-empty fields
-// are conditions.
-func (session *Session) Rows(bean interface{}) (*Rows, error) {
- return newRows(session, bean)
-}
-
-// Iterate record by record handle records from table, condiBeans's non-empty fields
-// are conditions. beans could be []Struct, []*Struct, map[int64]Struct
-// map[int64]*Struct
-func (session *Session) Iterate(bean interface{}, fun IterFunc) error {
- if session.isAutoClose {
- defer session.Close()
- }
-
- if session.statement.bufferSize > 0 {
- return session.bufferIterate(bean, fun)
- }
-
- rows, err := session.Rows(bean)
- if err != nil {
- return err
- }
- defer rows.Close()
-
- i := 0
- for rows.Next() {
- b := reflect.New(rows.beanType).Interface()
- err = rows.Scan(b)
- if err != nil {
- return err
- }
- err = fun(i, b)
- if err != nil {
- return err
- }
- i++
- }
- return err
-}
-
-// BufferSize sets the buffersize for iterate
-func (session *Session) BufferSize(size int) *Session {
- session.statement.bufferSize = size
- return session
-}
-
-func (session *Session) bufferIterate(bean interface{}, fun IterFunc) error {
- if session.isAutoClose {
- defer session.Close()
- }
-
- var bufferSize = session.statement.bufferSize
- var limit = session.statement.LimitN
- if limit > 0 && bufferSize > limit {
- bufferSize = limit
- }
- var start = session.statement.Start
- v := rValue(bean)
- sliceType := reflect.SliceOf(v.Type())
- var idx = 0
- for {
- slice := reflect.New(sliceType)
- if err := session.Limit(bufferSize, start).find(slice.Interface(), bean); err != nil {
- return err
- }
-
- for i := 0; i < slice.Elem().Len(); i++ {
- if err := fun(idx, slice.Elem().Index(i).Addr().Interface()); err != nil {
- return err
- }
- idx++
- }
-
- start = start + slice.Elem().Len()
- if limit > 0 && idx+bufferSize > limit {
- bufferSize = limit - idx
- }
-
- if bufferSize <= 0 || slice.Elem().Len() < bufferSize || idx == limit {
- break
- }
- }
-
- return nil
-}