Modify pika exporter

Author: YuCai18

tools/pika_exporter/exporter/client.go

@@ -120,13 +120,17 @@ func (c *client) InfoCommand(command string) (string, error) {
 	return redis.String(c.conn.Do("INFO", command))
 }
 
+func (c *client) InfoBigKeys() (string, error) {
+	return redis.String(c.conn.Do("INFO", "BIGKEYS"))
+}
 func (c *client) InfoNoneCommandList() (string, error) {
 	var rst []string
 
 	sectionsMap := map[string]bool{
 		"COMMAND_EXEC_COUNT": InfoConf.Execcount,
 		"COMMANDSTATS":       InfoConf.Commandstats,
 		"ROCKSDB":            InfoConf.Rocksdb,
+		"BIGKEYS":            InfoConf.BigKeys,
 	}
 	for section, flag := range sectionsMap {
 		if flag {
@@ -155,6 +159,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

@@ -13,7 +13,7 @@ var collectBigKeysMetrics = map[string]MetricConfig{
 		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+)`,
+				`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{
@@ -37,7 +37,7 @@ var collectBigKeysMetrics = map[string]MetricConfig{
 		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+)`,
+				`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 +49,11 @@ var collectBigKeysMetrics = map[string]MetricConfig{
 				ValueName: "member_size",
 			},
 			{
-				Name:      "bigkeys_value_length",
-				Help:      "Big key value length",
+				Name:      "bigkeys_key_length",
+				Help:      "Big key length for complex type",
 				Type:      metricTypeGauge,
 				Labels:    []string{LabelNameAddr, LabelNameAlias, "type", "key"},
-				ValueName: "value_length",
+				ValueName: "key_length",
 			},
 		},
 	},