mirror of
https://github.com/mainnika/mongox-go-driver.git
synced 2026-05-22 15:53:36 +00:00
Add onclose callback
This commit is contained in:
@@ -48,6 +48,8 @@ func (d *Database) LoadArray(target interface{}, filters ...interface{}) (err er
|
||||
return
|
||||
}
|
||||
|
||||
defer composed.OnClose().Invoke(d.Context(), target)
|
||||
|
||||
for i = 0; result.Next(d.Context()); {
|
||||
|
||||
var elem interface{}
|
||||
|
||||
@@ -25,6 +25,8 @@ func (d *Database) LoadOne(target interface{}, filters ...interface{}) (err erro
|
||||
return fmt.Errorf("can't create find result: %w", err)
|
||||
}
|
||||
|
||||
defer composed.OnClose().Invoke(d.Context(), target)
|
||||
|
||||
hasNext := result.Next(d.Context())
|
||||
if result.Err() != nil {
|
||||
return err
|
||||
|
||||
@@ -72,6 +72,9 @@ func (l *StreamLoader) Cursor() (cursor *mongox.Cursor) {
|
||||
|
||||
// Close cursor
|
||||
func (l *StreamLoader) Close() (err error) {
|
||||
|
||||
_ = l.query.OnClose().Invoke(l.ctx, l.target)
|
||||
|
||||
return l.cur.Close(l.ctx)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user