Skip to content

Commit

Permalink
feat(mongo): added more methods of mongo.FindResult
Browse files Browse the repository at this point in the history
  • Loading branch information
tikazyq committed Jul 10, 2022
1 parent 1ab985e commit 1a8c039
Showing 1 changed file with 21 additions and 0 deletions.
21 changes: 21 additions & 0 deletions mongo/result.go
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,10 @@ import (
type FindResultInterface interface {
One(val interface{}) (err error)
All(val interface{}) (err error)
GetCol() (col *Col)
GetSingleResult() (res *mongo.SingleResult)
GetCursor() (cur *mongo.Cursor)
GetError() (err error)
}

func NewFindResult() (fr *FindResult) {
Expand All @@ -28,6 +32,11 @@ type FindResult struct {
err error
}

func (fr *FindResult) GetError() (err error) {
//TODO implement me
panic("implement me")
}

func (fr *FindResult) One(val interface{}) (err error) {
if fr.err != nil {
return fr.err
Expand Down Expand Up @@ -59,3 +68,15 @@ func (fr *FindResult) All(val interface{}) (err error) {
}
return fr.cur.All(ctx, val)
}

func (fr *FindResult) GetCol() (col *Col) {
return fr.col
}

func (fr *FindResult) GetSingleResult() (res *mongo.SingleResult) {
return fr.res
}

func (fr *FindResult) GetCursor() (cur *mongo.Cursor) {
return fr.cur
}

0 comments on commit 1a8c039

Please sign in to comment.