From 6ef4ce3682ca5b78fc1a0702ff14fdb4a90d4fef Mon Sep 17 00:00:00 2001
From: yisaer <disxiaofei@163.com>
Date: Mon, 17 May 2021 18:57:46 +0800
Subject: [PATCH] add metrics

Signed-off-by: yisaer <disxiaofei@163.com>
---
 store/tikv/kv.go              |  6 +++++-
 store/tikv/metrics/metrics.go | 20 ++++++++++++++++++++
 2 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/store/tikv/kv.go b/store/tikv/kv.go
index bbf8517a42a8c..eee41265b9751 100644
--- a/store/tikv/kv.go
+++ b/store/tikv/kv.go
@@ -18,6 +18,7 @@ import (
 	"crypto/tls"
 	"math"
 	"math/rand"
+	"strconv"
 	"sync"
 	"sync/atomic"
 	"time"
@@ -414,17 +415,20 @@ func (s *KVStore) updateSafeTS(ctx context.Context) {
 		storeAddr := store.addr
 		go func(ctx context.Context, wg *sync.WaitGroup, storeID uint64, storeAddr string) {
 			defer wg.Done()
-			// TODO: add metrics for updateSafeTS
 			resp, err := tikvClient.SendRequest(ctx, storeAddr, tikvrpc.NewRequest(tikvrpc.CmdStoreSafeTS, &kvrpcpb.StoreSafeTSRequest{KeyRange: &kvrpcpb.KeyRange{
 				StartKey: []byte(""),
 				EndKey:   []byte(""),
 			}}), ReadTimeoutShort)
+			storeIDStr := strconv.Itoa(int(storeID))
 			if err != nil {
+				metrics.TiKVSafeTSUpdateCounter.WithLabelValues("fail", storeIDStr).Inc()
 				logutil.BgLogger().Debug("update safeTS failed", zap.Error(err), zap.Uint64("store-id", storeID))
 				return
 			}
 			safeTSResp := resp.Resp.(*kvrpcpb.StoreSafeTSResponse)
 			s.setSafeTS(storeID, safeTSResp.GetSafeTs())
+			metrics.TiKVSafeTSUpdateCounter.WithLabelValues("success", storeIDStr).Inc()
+			metrics.TiKVSafeTSUpdateStats.WithLabelValues(storeIDStr).Set(float64(safeTSResp.GetSafeTs()))
 		}(ctx, wg, storeID, storeAddr)
 	}
 	wg.Wait()
diff --git a/store/tikv/metrics/metrics.go b/store/tikv/metrics/metrics.go
index 8d71582fa2522..6b8ea32d456f7 100644
--- a/store/tikv/metrics/metrics.go
+++ b/store/tikv/metrics/metrics.go
@@ -59,6 +59,8 @@ var (
 	TiKVPanicCounter                       *prometheus.CounterVec
 	TiKVForwardRequestCounter              *prometheus.CounterVec
 	TiKVTSFutureWaitDuration               prometheus.Histogram
+	TiKVSafeTSUpdateCounter                *prometheus.CounterVec
+	TiKVSafeTSUpdateStats                  *prometheus.GaugeVec
 )
 
 // Label constants.
@@ -414,6 +416,22 @@ func initMetrics(namespace, subsystem string) {
 			Buckets:   prometheus.ExponentialBuckets(0.000005, 2, 30), // 5us ~ 2560s
 		})
 
+	TiKVSafeTSUpdateCounter = prometheus.NewCounterVec(
+		prometheus.CounterOpts{
+			Namespace: namespace,
+			Subsystem: subsystem,
+			Name:      "safets_update_counter",
+			Help:      "Counter of tikv safe_ts being updated.",
+		}, []string{LblResult, LblStore})
+
+	TiKVSafeTSUpdateStats = prometheus.NewGaugeVec(
+		prometheus.GaugeOpts{
+			Namespace: namespace,
+			Subsystem: subsystem,
+			Name:      "safets_update_stats",
+			Help:      "stat of tikv updating safe_ts stats",
+		}, []string{LblStore})
+
 	initShortcuts()
 }
 
@@ -468,6 +486,8 @@ func RegisterMetrics() {
 	prometheus.MustRegister(TiKVPanicCounter)
 	prometheus.MustRegister(TiKVForwardRequestCounter)
 	prometheus.MustRegister(TiKVTSFutureWaitDuration)
+	prometheus.MustRegister(TiKVSafeTSUpdateCounter)
+	prometheus.MustRegister(TiKVSafeTSUpdateStats)
 }
 
 // readCounter reads the value of a prometheus.Counter.