Make callback type

This commit is contained in:
Nikita Tokarchuk
2020-07-22 04:45:32 +02:00
parent 08c3c5b377
commit b796d5ac3b
7 changed files with 23 additions and 26 deletions
-19
View File
@@ -1,19 +0,0 @@
package database
import (
"context"
"github.com/mainnika/mongox-go-driver/v2/mongox/query"
)
func onDecode(ctx context.Context, iter interface{}, callbacks ...query.OnDecode) (err error) {
for _, cb := range callbacks {
err = cb(ctx, iter)
if err != nil {
return
}
}
return
}
+1 -1
View File
@@ -69,7 +69,7 @@ func (d *Database) LoadArray(target interface{}, filters ...interface{}) (err er
return
}
err = onDecode(d.ctx, elem, composed.OnDecode()...)
err = composed.OnDecode().Invoke(d.Context(), elem)
if err != nil {
_ = result.Close(d.Context())
return
+1 -1
View File
@@ -40,7 +40,7 @@ func (d *Database) LoadOne(target interface{}, filters ...interface{}) (err erro
return
}
err = onDecode(d.ctx, target, composed.OnDecode()...)
err = composed.OnDecode().Invoke(d.Context(), target)
if err != nil {
return
}
+1 -1
View File
@@ -42,7 +42,7 @@ func (l *StreamLoader) Decode() (err error) {
return
}
err = onDecode(l.ctx, l.target, l.query.OnDecode()...)
err = l.query.OnDecode().Invoke(l.ctx, l.target)
if err != nil {
return
}