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

add monitor options #5

Merged
luuvansinh merged 1 commits from add-monitor into master 2022-03-04 13:15:03 +00:00
Conversation 0 Commits 1 Files Changed 1 +7 -2
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mongodb.go
View File

@ -5,6 +5,7 @@ import (
"fmt" "fmt"
"github.com/logrusorgru/aurora" "github.com/logrusorgru/aurora"
"go.mongodb.org/mongo-driver/event"
"go.mongodb.org/mongo-driver/mongo" "go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options" "go.mongodb.org/mongo-driver/mongo/options"
"go.mongodb.org/mongo-driver/mongo/readpref" "go.mongodb.org/mongo-driver/mongo/readpref"
@ -14,6 +15,7 @@ import (
type Config struct { type Config struct {
Host string Host string
DBName string DBName string
Monitor *event.CommandMonitor
TLS *ConnectTLSOpts TLS *ConnectTLSOpts
Standalone *ConnectStandaloneOpts Standalone *ConnectStandaloneOpts
@ -86,6 +88,9 @@ func connectWithTLS(cfg Config) (*mongo.Database, error) {
uri := fmt.Sprintf(s, cfg.Host, caFile.Name(), certFile.Name(), pwd) uri := fmt.Sprintf(s, cfg.Host, caFile.Name(), certFile.Name(), pwd)
readPref := getReadPref(opts.ReadPreferenceMode) readPref := getReadPref(opts.ReadPreferenceMode)
clientOpts := options.Client().SetReadPreference(readPref).SetReplicaSet(opts.ReplSet).ApplyURI(uri) clientOpts := options.Client().SetReadPreference(readPref).SetReplicaSet(opts.ReplSet).ApplyURI(uri)
if cfg.Monitor != nil {
clientOpts.SetMonitor(cfg.Monitor)
}
client, err := mongo.Connect(ctx, clientOpts) client, err := mongo.Connect(ctx, clientOpts)
if err != nil { if err != nil {
return nil, err return nil, err