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

add monitor options

This commit is contained in:
Sinh 2022-03-04 20:14:30 +07:00
parent 3e29ed42b1
commit 29e892b420
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
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"
@ -12,8 +13,9 @@ import (
// Config ... // Config ...
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