Add resetter interface

This commit is contained in:
Nikita Tokarchuk
2019-05-16 13:56:55 +02:00
parent ea72b26e1d
commit 2207dc5afd
2 changed files with 10 additions and 0 deletions
+5
View File
@@ -19,6 +19,11 @@ type Loader interface {
Load(db *Database, filters ...interface{}) error
}
// Resetter is an interface for documenta that can be resetted
type Resetter interface {
Reset()
}
// BaseObjectID is an interface for documents that have objectId type for the _id field
type BaseObjectID interface {
GetID() primitive.ObjectID
+5
View File
@@ -25,6 +25,11 @@ func (l *ManyLoader) Get() error {
return errors.NotFoundErrorf("%s", mongo.ErrNoDocuments)
}
resettable, canReset := l.target.(mongox.Resetter)
if canReset {
resettable.Reset()
}
err := l.Decode(l.target)
if err != nil {
return errors.InternalErrorf("can't decode desult: %s", err)