You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
1.4 KiB
83 lines
1.4 KiB
package database
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/mainnika/mongox-go-driver/v2/mongox"
|
|
"github.com/mainnika/mongox-go-driver/v2/mongox/base"
|
|
"github.com/mainnika/mongox-go-driver/v2/mongox/query"
|
|
)
|
|
|
|
// StreamLoader is a controller for a database cursor
|
|
type StreamLoader struct {
|
|
cur *mongox.Cursor
|
|
query *query.Query
|
|
ctx context.Context
|
|
target interface{}
|
|
}
|
|
|
|
// DecodeNext loads next documents to a target or returns an error
|
|
func (l *StreamLoader) DecodeNext() (err error) {
|
|
|
|
err = l.Next()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = l.Decode()
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Decode function decodes the current cursor document into the target
|
|
func (l *StreamLoader) Decode() (err error) {
|
|
|
|
base.Reset(l.target)
|
|
|
|
err = l.cur.Decode(l.target)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = l.query.OnDecode().Invoke(l.ctx, l.target)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
// Next loads next documents but doesn't perform decoding
|
|
func (l *StreamLoader) Next() (err error) {
|
|
|
|
hasNext := l.cur.Next(l.ctx)
|
|
err = l.cur.Err()
|
|
|
|
if err != nil {
|
|
return
|
|
}
|
|
if !hasNext {
|
|
err = mongox.ErrNoDocuments
|
|
}
|
|
|
|
return
|
|
}
|
|
|
|
func (l *StreamLoader) Cursor() (cursor *mongox.Cursor) {
|
|
return l.cur
|
|
}
|
|
|
|
// Close cursor
|
|
func (l *StreamLoader) Close() (err error) {
|
|
|
|
_ = l.query.OnClose().Invoke(l.ctx, l.target)
|
|
|
|
return l.cur.Close(l.ctx)
|
|
}
|
|
|
|
func (l *StreamLoader) Err() (err error) {
|
|
return l.cur.Err()
|
|
}
|
|
|