PenaSide/common
3
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

added monitor only for command 'find'

Browse Source
This commit is contained in:
Pavel 2024-08-04 10:09:05 +03:00
parent 502bc1d1c4
commit 78c38993d3
1 changed files with 19 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
mongo/connection.go
View File

@ -3,6 +3,7 @@ package mongo
import (
"context"
"fmt"
"go.mongodb.org/mongo-driver/event"
"log"
"net"
"net/url"
@ -27,17 +28,23 @@ func Connect(ctx context.Context, deps *ConnectDeps) (*mongo.Database, error) {
Host: net.JoinHostPort(deps.Configuration.Host, deps.Configuration.Port),
}
//cmdMonitor := &event.CommandMonitor{
// Started: func(_ context.Context, evt *event.CommandStartedEvent) {
// log.Println(evt.Command)
// },
// Succeeded: func(_ context.Context, evt *event.CommandSucceededEvent) {
// log.Println(evt.Reply)
// },
// Failed: func(_ context.Context, evt *event.CommandFailedEvent) {
// log.Println(evt.Failure)
// },
//}
cmdMonitor := &event.CommandMonitor{
Started: func(_ context.Context, evt *event.CommandStartedEvent) {
if evt.CommandName == "find" {
log.Println(evt.Command)
}
},
Succeeded: func(_ context.Context, evt *event.CommandSucceededEvent) {
if evt.CommandName == "find" {
log.Println(evt.Reply)
}
},
Failed: func(_ context.Context, evt *event.CommandFailedEvent) {
if evt.CommandName == "find" {
log.Println(evt.Failure)
}
},
}
connectionOptions := options.Client().
ApplyURI(mongoURI.String()).
@ -46,7 +53,7 @@ func Connect(ctx context.Context, deps *ConnectDeps) (*mongo.Database, error) {
AuthSource: deps.Configuration.Auth,
Username: deps.Configuration.User,
Password: deps.Configuration.Password,
})
}).SetMonitor(cmdMonitor)
ticker := time.NewTicker(1 * time.Second)
timeoutExceeded := time.After(deps.Timeout)