db connect

register.go

@@ -7,7 +7,7 @@ import (
 
 // API is wrapper for one RouterGroup and mgo DB
 type API struct {
-	DBSession            Session
+	DBConnection         Connection
 	DBName               string
 	routerGroup          *gin.RouterGroup
 	jwtSecret            []byte
@@ -22,11 +22,11 @@ type Context struct {
 }
 
 // New returns new instance of the API
-func New(session Session, dbname string, routerGroup *gin.RouterGroup) *API {
+func New(connection Connection, dbname string, routerGroup *gin.RouterGroup) *API {
 	return &API{
-		DBSession:   session,
-		DBName:      dbname,
-		routerGroup: routerGroup,
+		DBConnection: connection,
+		DBName:       dbname,
+		routerGroup:  routerGroup,
 	}
 }
 
@@ -37,7 +37,7 @@ func (api *API) RegisterModel(m mgocrud.ModelInterface) {
 
 	modelRegistry = append(modelRegistry, m)
 
-	session := api.DBSession.Copy()
+	session := api.DBConnection.NewSession()
 	defer session.Close()
 	db := session.DB(api.DBName)