Selly-Modules
/
mongodb
8
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

return mongo instance on connect (for multiple database connect)

This commit is contained in:
Nam Huynh 2021-10-08 15:05:13 +07:00
parent 110127e76a
commit 2761a05c01
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mongodb.go
View File

@ -12,7 +12,7 @@ import (
var db *mongo.Database var db *mongo.Database
// Connect to mongo server // Connect to mongo server
func Connect(host, user, password, dbName, mechanism, source string) error { func Connect(host, user, password, dbName, mechanism, source string) (*mongo.Database, error) {
connectOptions := options.ClientOptions{} connectOptions := options.ClientOptions{}
// Set auth if existed // Set auth if existed
if user != "" && password != "" { if user != "" && password != "" {
@ -28,14 +28,14 @@ func Connect(host, user, password, dbName, mechanism, source string) error {
client, err := mongo.Connect(context.Background(), connectOptions.ApplyURI(host)) client, err := mongo.Connect(context.Background(), connectOptions.ApplyURI(host))
if err != nil { if err != nil {
fmt.Println("Error when connect to MongoDB database", host, err) fmt.Println("Error when connect to MongoDB database", host, err)
return err return nil, err
} }
fmt.Println(aurora.Green("*** CONNECTED TO MONGODB: " + host)) fmt.Println(aurora.Green("*** CONNECTED TO MONGODB: " + host))
// Set data // Set data
db = client.Database(dbName) db = client.Database(dbName)
return nil return db, nil
} }
// GetInstance ... // GetInstance ...