mainnika/mongox-go-driver
1
0
Fork 0
mirror of https://github.com/mainnika/mongox-go-driver.git synced 2026-06-13 01:03:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add resetter interface

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
mongox/common.go
+5
View File
@@ -19,6 +19,11 @@ type Loader interface {
Load(db *Database, filters ...interface{}) error 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 // BaseObjectID is an interface for documents that have objectId type for the _id field
type BaseObjectID interface { type BaseObjectID interface {
GetID() primitive.ObjectID GetID() primitive.ObjectID
mongox/common/loadmany.go
+5
View File
@@ -25,6 +25,11 @@ func (l *ManyLoader) Get() error {
return errors.NotFoundErrorf("%s", mongo.ErrNoDocuments) return errors.NotFoundErrorf("%s", mongo.ErrNoDocuments)
} }
resettable, canReset := l.target.(mongox.Resetter)
if canReset {
resettable.Reset()
}
err := l.Decode(l.target) err := l.Decode(l.target)
if err != nil { if err != nil {
return errors.InternalErrorf("can't decode desult: %s", err) return errors.InternalErrorf("can't decode desult: %s", err)