Add onclose callback

This commit is contained in:
Nikita Tokarchuk
2020-07-22 04:45:32 +02:00
parent b796d5ac3b
commit 951e5f5bef
6 changed files with 22 additions and 2 deletions
+2
View File
@@ -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{}
+2
View File
@@ -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
+3
View File
@@ -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)
}