gopackage/mgocrud
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

encapsulate ErrNotFound

Browse Source
This commit is contained in:
Sebastian Frank
2022-02-09 15:53:37 +01:00
parent 115c288a0b
commit 7dd238342c
1 changed files with 8 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
session.go
View File

@@ -6,6 +6,10 @@ import (
mgo "gopkg.in/mgo.v2" mgo "gopkg.in/mgo.v2"
) )
var (
ErrNotFound = errors.New("not found")
)
type Session struct { type Session struct {
session *mgo.Session session *mgo.Session
} }
@@ -111,7 +115,7 @@ func (q *Query) Select(selector interface{}) *Query {
func (q *Query) One(result interface{}) error { func (q *Query) One(result interface{}) error {
err := q.query.One(result) err := q.query.One(result)
if err == mgo.ErrNotFound { if err == mgo.ErrNotFound {
err = &ErrNotFound{msg: err.Error()} err = ErrNotFound
} }
return err return err
} }
@@ -134,7 +138,7 @@ func (q *Query) Limit(n int) *Query {
func (q *Query) All(result interface{}) error { func (q *Query) All(result interface{}) error {
err := q.query.All(result) err := q.query.All(result)
if err == mgo.ErrNotFound { if err == mgo.ErrNotFound {
err = &ErrNotFound{msg: err.Error()} err = ErrNotFound
} }
return err return err
} }
@@ -142,7 +146,7 @@ func (q *Query) All(result interface{}) error {
func (q *Query) Count() (int, error) { func (q *Query) Count() (int, error) {
c, err := q.query.Count() c, err := q.query.Count()
if err == mgo.ErrNotFound { if err == mgo.ErrNotFound {
err = &ErrNotFound{msg: err.Error()} err = ErrNotFound
} }
return c, err return c, err
} }
@@ -169,7 +173,7 @@ type Pipe struct {
func (p *Pipe) All(result interface{}) error { func (p *Pipe) All(result interface{}) error {
err := p.pipe.All(result) err := p.pipe.All(result)
if err == mgo.ErrNotFound { if err == mgo.ErrNotFound {
err = &ErrNotFound{msg: err.Error()} err = ErrNotFound
} }
return err return err
} }
@@ -177,14 +181,3 @@ func (p *Pipe) All(result interface{}) error {
func (c *Collection) Pipe(pipeline interface{}) *Pipe { func (c *Collection) Pipe(pipeline interface{}) *Pipe {
return &Pipe{pipe: c.collection.Pipe(pipeline).AllowDiskUse()} return &Pipe{pipe: c.collection.Pipe(pipeline).AllowDiskUse()}
} }
type ErrNotFound struct {
msg string
}
func (e *ErrNotFound) Error() string {
if e.msg == "" {
return "not found"
}
return e.msg
}