From 3bc51771ce23db44f29057361b0ccad9705e1a65 Mon Sep 17 00:00:00 2001
From: zhuofeng <zhuofeng@yunshan.net.cn>
Date: Mon, 25 Sep 2023 00:57:18 +0800
Subject: [PATCH] [ingester] prometheus debug request

---
 .../prometheus/decoder/grpc_label_ids.go       | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/server/ingester/prometheus/decoder/grpc_label_ids.go b/server/ingester/prometheus/decoder/grpc_label_ids.go
index 9097332992b..75a84e07246 100644
--- a/server/ingester/prometheus/decoder/grpc_label_ids.go
+++ b/server/ingester/prometheus/decoder/grpc_label_ids.go
@@ -125,6 +125,8 @@ type PrometheusLabelTable struct {
 
 	counter *RequestCounter
 	utils.Closable
+	debug    bool
+	notarget bool
 }
 
 func NewPrometheusLabelTable(controllerIPs []string, port, rpcMaxMsgSize int) *PrometheusLabelTable {
@@ -208,6 +210,10 @@ func (t *PrometheusLabelTable) RequestLabelIDs(request *trident.PrometheusLabelR
 		return nil, err
 	}
 
+	if t.debug {
+		log.Infof("prometheus debug request(%d): %s\nresponse(%d): %s", len(request.GetRequestLabels()), request, len(response.GetResponseLabelIds()), response)
+	}
+
 	t.counter.ResponseLabelsCount += int64(len(response.GetResponseLabelIds()))
 	t.counter.RequestTotalDelayNs += int64(time.Since(requestStart))
 	t.updatePrometheusLabels(response)
@@ -535,6 +541,18 @@ func (t *PrometheusLabelTable) HandleSimpleCommand(op uint16, arg string) string
 
 // request string as: metric=xxx,,job==xxx,,instance==xxx,,label1==xxx,,label2==xxx
 func (t *PrometheusLabelTable) testString(request string) string {
+	if request == "debug" {
+		t.debug = true
+		return fmt.Sprintf("open request debug")
+	} else {
+		t.debug = false
+	}
+	if request == "notarget" {
+		t.notarget = true
+		return fmt.Sprintf("set no target request")
+	} else {
+		t.notarget = false
+	}
 	req := &trident.PrometheusLabelRequest{}
 	metricReq := &trident.MetricLabelRequest{}
 	targetReq := &trident.TargetRequest{}