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

db.Session()

Browse Source
This commit is contained in:
Sebastian Frank 2022-02-09 18:55:58 +01:00
parent 7dd238342c
commit 98801edbc0
Signed by: apairon
GPG Key ID: A0E05A8199CE3F57
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
session.go
View File

@ -33,16 +33,21 @@ func NewSession(dial string) (*Session, error) {
type Database struct { type Database struct {
database *mgo.Database database *mgo.Database
session *Session
} }
func (s *Session) DB(name string) *Database { func (s *Session) DB(name string) *Database {
return &Database{database: s.session.DB(name)} return &Database{database: s.session.DB(name), session: s}
} }
type Collection struct { type Collection struct {
collection *mgo.Collection collection *mgo.Collection
} }
func (db *Database) Session() *Session {
return db.session
}
func (db *Database) C(name string) *Collection { func (db *Database) C(name string) *Collection {
return &Collection{collection: db.database.C(name)} return &Collection{collection: db.database.C(name)}
} }