logmower-shipper/cmd/mongo.go

package logmower

import (
	"context"
	"fmt"

	prom "github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promauto"
	mongoEvent "go.mongodb.org/mongo-driver/event"
	mongoOpt "go.mongodb.org/mongo-driver/mongo/options"
	"go.uber.org/zap"
)

func mongoMonitoredClientOptions(l *zap.Logger) *mongoOpt.ClientOptions {
	promMongoHeartbeat := promauto.NewHistogramVec(prom.HistogramOpts{
		Subsystem: "mongo",
		Name:      "heartbeat_time",
		Help:      "Time in ns for succeeded heartbeat, or 0 on failure",
		Buckets:   []float64{1},
	}, []string{"connection_id"})

	promMongoCmd := promauto.NewHistogramVec(prom.HistogramOpts{
		Subsystem: "mongo",
		Name:      "command_time",
		Help:      "Time in ns of commands",
		Buckets:   prom.DefBuckets,
	}, []string{"connection_id", "command_name"})

	promMongoCmdErr := promauto.NewCounterVec(prom.CounterOpts{
		Subsystem: "mongo",
		Name:      "errors",
		Help:      "Count of failed commands",
	}, []string{"connection_id", "command_name"})

	return mongoOpt.Client().
		SetServerMonitor(&mongoEvent.ServerMonitor{
			ServerHeartbeatSucceeded: func(ev *mongoEvent.ServerHeartbeatSucceededEvent) {
				promMongoHeartbeat.WithLabelValues(ev.ConnectionID).Observe(float64(ev.DurationNanos))
			},
			ServerHeartbeatFailed: func(ev *mongoEvent.ServerHeartbeatFailedEvent) {
				promMongoHeartbeat.WithLabelValues(ev.ConnectionID).Observe(0)
				l.Error("mongo heartbeat", zap.Error(ev.Failure), zap.String("connection_id", ev.ConnectionID))
			},
		}).
		SetMonitor(&mongoEvent.CommandMonitor{
			Succeeded: func(_ context.Context, ev *mongoEvent.CommandSucceededEvent) {
				promMongoCmd.WithLabelValues(ev.ConnectionID, ev.CommandName).Observe(float64(ev.DurationNanos))
			},
			Failed: func(_ context.Context, ev *mongoEvent.CommandFailedEvent) {
				promMongoCmd.WithLabelValues(ev.ConnectionID, ev.CommandName).Observe(float64(ev.DurationNanos))

				promMongoCmdErr.WithLabelValues(ev.ConnectionID, ev.CommandName).Add(1)
				l.Error("mongo command", zap.Error(fmt.Errorf("%s", ev.Failure)), zap.String("connection_id", ev.ConnectionID), zap.String("command_name", ev.CommandName)) // TODO: https://github.com/mongodb/mongo-go-driver/pull/1105
			},
		})
}