1 Commits

Author SHA1 Message Date
98801edbc0 db.Session() 2022-02-09 18:55:58 +01:00

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)}
} }