/
mongo_db.go
50 lines (41 loc) · 1.47 KB
/
mongo_db.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package mongokit
import (
"context"
"fmt"
"log"
"net/url"
"github.com/adipurnama/go-toolkit/db"
pmongo "github.com/pinpoint-apm/pinpoint-go-agent/plugin/mongodriver"
"github.com/pkg/errors"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)
// NewMongoDBClient returns new mongodb client using the go mongo-driver.
func NewMongoDBClient(opt *db.Option, authDB string) (*mongo.Database, error) {
connURL := &url.URL{
Scheme: "mongodb",
User: url.UserPassword(opt.Username, opt.Password),
Host: fmt.Sprintf("%s:%d", opt.Host, opt.Port),
Path: "/",
}
q := connURL.Query()
q.Add("authSource", authDB)
connURL.RawQuery = q.Encode()
clientOptions := options.Client()
clientOptions.ApplyURI(connURL.String())
clientOptions.SetConnectTimeout(opt.ConnectionOption.ConnectTimeout)
clientOptions.SetMaxConnIdleTime(opt.ConnectionOption.MaxLifetime)
clientOptions.SetMaxPoolSize(uint64(opt.ConnectionOption.MaxOpen))
clientOptions.SetMonitor(pmongo.NewMonitor())
client, err := mongo.NewClient(clientOptions)
if err != nil {
return nil, errors.Wrap(err, "mongokit: cannot create new mongo client")
}
ctx, cancel := context.WithTimeout(context.Background(), opt.ConnectionOption.ConnectTimeout)
defer cancel()
if err = client.Connect(ctx); err != nil {
return nil, errors.Wrap(err, "mongokit: cannot connect to client")
}
log.Println("successfully connected to mongo", connURL.Host)
return client.Database(opt.DatabaseName), nil
}