Modify pika exporter

Author: YuCai18

tools/pika_exporter/exporter/client.go

@@ -155,6 +155,7 @@ func (c *client) InfoAllCommandList() (string, error) {
 		"COMMAND_EXEC_COUNT": InfoConf.Execcount,
 		"COMMANDSTATS":       InfoConf.Commandstats,
 		"ROCKSDB":            InfoConf.Rocksdb,
+		"BIGKEYS":            InfoConf.BigKeys,
 	}
 	for section, flag := range sectionsMap {
 		if flag {

tools/pika_exporter/exporter/metrics/bigkeys.go

@@ -11,14 +11,15 @@ func RegisterBigKeys() {
 var collectBigKeysMetrics = map[string]MetricConfig{
 	"bigkeys_string": {
 		Parser: &regexParser{
-			name: "bigkeys_string",
-			reg: regexp.MustCompile(
-				`Type: string, key: (?P<key>[^,]+), key_length: (?P<key_length>\d+), value_length: (?P<value_length>\d+)`,
+			name:   "bigkeys_string",
+			source: "bigkeys_output",
+			reg:    regexp.MustCompile(
+				`(?s).*?#\s*String.*?Big\s+Keys\s+number:\s*\d+.*?Type:\s*string,\s*key:\s*(?P<key>[^,]+),\s*key_length:\s*(?P<key_length>\d+),\s*value_length:\s*(?P<value_length>\d+)`,
 			),
 		},
 		MetricMeta: MetaDatas{
 			{
-				Name:      "bigkeys_key_length", 
+				Name:      "bigkeys_string_key_length", 
 				Help:      "Big key length for string type",
 				Type:      metricTypeGauge,
 				Labels:    []string{LabelNameAddr, LabelNameAlias, "type", "key"},
@@ -35,9 +36,10 @@ var collectBigKeysMetrics = map[string]MetricConfig{
 	},
 	"bigkeys_complex": {
 		Parser: &regexParser{
-			name: "bigkeys_complex",
-			reg: regexp.MustCompile(
-				`Type: (?P<type>\w+), key: (?P<key>[^,]+), value_length: (?P<value_length>\d+), member_size: (?P<member_size>\d+)`,
+			name:   "bigkeys_complex",
+			source: "bigkeys_output",
+			reg:    regexp.MustCompile(
+				`(?s).*?#\s*(Hash|List|Set|ZSet).*?Big\s+Keys\s+number:\s*\d+.*?Type:\s*(?P<type>hash|list|set|zset),\s*key:\s*(?P<key>[^,]+),\s*key_length:\s*(?P<key_length>\d+),\s*member_size:\s*(?P<member_size>\d+)`,
 			),
 		},
 		MetricMeta: MetaDatas{
@@ -49,11 +51,11 @@ var collectBigKeysMetrics = map[string]MetricConfig{
 				ValueName: "member_size",
 			},
 			{
-				Name:      "bigkeys_value_length",
-				Help:      "Big key value length",
+				Name:      "bigkeys_complex_key_length",
+				Help:      "Big key length for complex type",
 				Type:      metricTypeGauge,
 				Labels:    []string{LabelNameAddr, LabelNameAlias, "type", "key"},
-				ValueName: "value_length",
+				ValueName: "key_length",
 			},
 		},
 	},

tools/pika_exporter/exporter/parser.go

@@ -28,8 +28,36 @@ func parseInfo(info string) (*semver.Version, map[string]string, error) {
 	return version, extracts, nil
 }
 
+func parseInfoBigkey(s string) string {
+	scanner := bufio.NewScanner(strings.NewReader(s))
+	var bigKeysContent strings.Builder
+	isBigKeysSection := false
+	
+	for scanner.Scan() {
+		line := scanner.Text()
+		
+		if strings.Contains(line, "# BigKeys statistics") {
+			isBigKeysSection = true
+			bigKeysContent.WriteString(line)
+			bigKeysContent.WriteString("\n")
+			continue
+		}
+		
+		if isBigKeysSection {
+			bigKeysContent.WriteString(line)
+			bigKeysContent.WriteString("\n")
+		}
+	}
+	
+	return bigKeysContent.String()
+}
+
 func extractInfo(s string) (map[string]string, error) {
 	m := make(map[string]string)
+	bigkeysOutput := parseInfoBigkey(s)
+	if len(bigkeysOutput) > 0 {
+		m["bigkeys_output"] = bigkeysOutput
+	}
 	scanner := bufio.NewScanner(strings.NewReader(s))
 	for scanner.Scan() {
 		line := scanner.Text()