add backup status collector

reference: percona/mongodb_exporter#557

Author: Grain-Yu

exporter/backupstatus_collector.go

@@ -0,0 +1,85 @@
+// mongodb_exporter
+// Copyright (C) 2017 Percona LLC
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+package exporter
+
+import (
+	"context"
+    "strconv"
+	"strings"
+	"os"
+	"github.com/prometheus/client_golang/prometheus"
+	"github.com/sirupsen/logrus"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+)
+
+type backupStatusCollector struct {
+	ctx  context.Context
+	base *baseCollector
+
+	compatibleMode bool
+	topologyInfo   labelsGetter
+}
+
+// newBackupStatusCollector creates a collector for statistics on backup status.
+func newBackupStatusCollector(ctx context.Context, client *mongo.Client, logger *logrus.Logger, compatible bool, topology labelsGetter) *backupStatusCollector {
+	return &backupStatusCollector{
+		ctx:            ctx,
+		base:           newBaseCollector(client, logger),
+		compatibleMode: compatible,
+		topologyInfo:   topology,
+	}
+}
+
+func (d *backupStatusCollector) Describe(ch chan<- *prometheus.Desc) {
+	d.base.Describe(d.ctx, ch, d.collect)
+}
+
+func (d *backupStatusCollector) Collect(ch chan<- prometheus.Metric) {
+	d.base.Collect(ch)
+}
+
+func (d *backupStatusCollector) collect(ch chan<- prometheus.Metric) {
+	defer measureCollectTime(ch, "mongodb", "backupstatus")()
+
+	logger := d.base.logger
+
+	var m bson.M
+	var delblank string
+    filename := "/tmp/mongodbBakStatus.txt"
+    content, err := os.ReadFile(filename)
+    if err != nil {
+		m = bson.M{"10000"}
+		ch <- prometheus.NewInvalidMetric(prometheus.NewInvalidDesc(err), err)
+		return
+	}
+
+	delblank = strings.Replace(string(content), " ", "", -1)
+    delnewline, _ := strconv.ParseFloat(strings.Replace(delblank,"\n", "", -1), 64)
+	if delnewline == 1 {
+        m = bson.M{"1"}
+    } else {
+		m = bson.M{"0"}
+	}	
+
+	logger.Debug("backupStatus result:")
+	debugResult(logger, m)
+
+	for _, metric := range makeMetrics("", bson.M{"backupStatus": m}, d.topologyInfo.baseLabels(), d.compatibleMode) {
+		ch <- metric
+	}
+}

exporter/exporter.go

@@ -68,6 +68,7 @@ type Opts struct {
 	EnableTopMetrics       bool
 	EnableIndexStats       bool
 	EnableCollStats        bool
+	EnableBackupStatus     bool
 
 	EnableOverrideDescendingIndex bool
 
@@ -169,6 +170,8 @@ func (e *Exporter) makeRegistry(ctx context.Context, client *mongo.Client, topol
 		e.opts.EnableTopMetrics = true
 		e.opts.EnableReplicasetStatus = true
 		e.opts.EnableIndexStats = true
+		e.opts.EnableBackupStatus = true
+
 	}
 
 	// arbiter only have isMaster privileges
@@ -290,6 +293,8 @@ func (e *Exporter) Handler() http.Handler {
 				requestOpts.EnableIndexStats = true
 			case "collstats":
 				requestOpts.EnableCollStats = true
+			case "backupstatus":
+				requestOpts.EnableBackupStatus = true
 			}
 		}
 

main.go

@@ -51,6 +51,7 @@ type GlobalFlags struct {
 	EnableTopMetrics       bool `name:"collector.topmetrics" help:"Enable collecting metrics from top admin command"`
 	EnableIndexStats       bool `name:"collector.indexstats" help:"Enable collecting metrics from $indexStats"`
 	EnableCollStats        bool `name:"collector.collstats" help:"Enable collecting metrics from $collStats"`
+	EnableBackupStatus     bool `name:"collector.backupstatus" help:"Enable collecting metrics from backupStatus"`
 
 	EnableOverrideDescendingIndex bool `name:"metrics.overridedescendingindex" help:"Enable descending index name override to replace -1 with _DESC"`
 
@@ -129,6 +130,7 @@ func buildExporter(opts GlobalFlags) *exporter.Exporter {
 		EnableDBStats:          opts.EnableDBStats,
 		EnableIndexStats:       opts.EnableIndexStats,
 		EnableCollStats:        opts.EnableCollStats,
+		EnableBackupStatus:     opts.EnableBackupStatus,
 
 		EnableOverrideDescendingIndex: opts.EnableOverrideDescendingIndex,