Merge pull request #275 from j-gourdon/exchanges-filters

feat:Add exchanges filtering capabilities

README.md

@@ -60,6 +60,8 @@ SKIP_VHOST | ^$ |regex, matching vhost names are not exported. First performs IN
 INCLUDE_VHOST | .* | regex vhost filter. Only queues in matching vhosts are exported
 INCLUDE_QUEUES | .* | regex queue filter. Just matching names are exported
 SKIP_QUEUES | ^$ |regex, matching queue names are not exported (useful for short-lived rpc queues). First performed INCLUDE, after SKIP
+INCLUDE_EXCHANGES | .* | regex exchange filter. (Only exchanges in matching vhosts are exported)
+SKIP_EXCHANGES  | ^$ | regex, matching exchanges names are not exported. First performed INCLUDE, after SKIP
 RABBIT_CAPABILITIES | bert,no_sort | comma-separated list of extended scraping capabilities supported by the target RabbitMQ server
 RABBIT_EXPORTERS | exchange,node,queue | List of enabled modules. Possible modules: connections,shovel,federation,exchange,node,queue,memory
 RABBIT_TIMEOUT | 30 | timeout in seconds for retrieving data from management plugin.

config.example.json

@@ -10,6 +10,8 @@
     "key_file": "client-key.pem",
     "insecure_skip_verify": false,
     "exlude_metrics": [],
+    "include_exchanges": ".*",
+    "skip_exchanges": "^$",
     "include_queues": ".*",
     "skip_queues": "^$",
     "skip_vhost": "^$",

config.go

@@ -25,6 +25,8 @@ var (
 		KeyFile:            "client-key.pem",
 		InsecureSkipVerify: false,
 		ExcludeMetrics:     []string{},
+		SkipExchanges:      regexp.MustCompile("^$"),
+		IncludeExchanges:   regexp.MustCompile(".*"),
 		SkipQueues:         regexp.MustCompile("^$"),
 		IncludeQueues:      regexp.MustCompile(".*"),
 		SkipVHost:          regexp.MustCompile("^$"),
@@ -49,10 +51,14 @@ type rabbitExporterConfig struct {
 	KeyFile                  string              `json:"key_file"`
 	InsecureSkipVerify       bool                `json:"insecure_skip_verify"`
 	ExcludeMetrics           []string            `json:"exlude_metrics"`
+	SkipExchanges            *regexp.Regexp      `json:"-"`
+	IncludeExchanges         *regexp.Regexp      `json:"-"`
 	SkipQueues               *regexp.Regexp      `json:"-"`
 	IncludeQueues            *regexp.Regexp      `json:"-"`
 	SkipVHost                *regexp.Regexp      `json:"-"`
 	IncludeVHost             *regexp.Regexp      `json:"-"`
+	IncludeExchangesString   string              `json:"include_exchanges"`
+	SkipExchangesString      string              `json:"skip_exchanges"`
 	IncludeQueuesString      string              `json:"include_queues"`
 	SkipQueuesString         string              `json:"skip_queues"`
 	SkipVHostString          string              `json:"skip_vhost"`
@@ -91,6 +97,8 @@ func initConfigFromFile(configFile string) error {
 		}
 	}
 
+	config.SkipExchanges = regexp.MustCompile(config.SkipExchangesString)
+	config.IncludeExchanges = regexp.MustCompile(config.IncludeExchangesString)
 	config.SkipQueues = regexp.MustCompile(config.SkipQueuesString)
 	config.IncludeQueues = regexp.MustCompile(config.IncludeQueuesString)
 	config.SkipVHost = regexp.MustCompile(config.SkipVHostString)
@@ -173,6 +181,14 @@ func initConfig() {
 		config.ExcludeMetrics = strings.Split(ExcludeMetrics, ",")
 	}
 
+	if SkipExchanges := os.Getenv("SKIP_EXCHANGES"); SkipExchanges != "" {
+		config.SkipExchanges = regexp.MustCompile(SkipExchanges)
+	}
+
+	if IncludeExchanges := os.Getenv("INCLUDE_EXCHANGES"); IncludeExchanges != "" {
+		config.IncludeExchanges = regexp.MustCompile(IncludeExchanges)
+	}
+
 	if SkipQueues := os.Getenv("SKIP_QUEUES"); SkipQueues != "" {
 		config.SkipQueues = regexp.MustCompile(SkipQueues)
 	}

exporter_exchange.go

@@ -62,6 +62,20 @@ func (e exporterExchange) Collect(ctx context.Context, ch chan<- prometheus.Metr
 
 	for key, countvec := range e.exchangeMetrics {
 		for _, exchange := range exchangeData {
+			ename := exchange.labels["name"]
+			vname := exchange.labels["vhost"]
+			if vhostIncluded := config.IncludeVHost.MatchString(vname); !vhostIncluded {
+				continue
+			}
+			if skipVhost := config.SkipVHost.MatchString(vname); skipVhost {
+				continue
+			}
+			if exchangeIncluded := config.IncludeExchanges.MatchString(ename); !exchangeIncluded {
+				continue
+			}
+			if exchangeSkipped := config.SkipExchanges.MatchString(ename); exchangeSkipped {
+				continue
+			}
 			if value, ok := exchange.metrics[key]; ok {
 				// log.WithFields(log.Fields{"vhost": exchange.vhost, "exchange": exchange.name, "key": key, "value": value}).Debug("Set exchange metric for key")
 				ch <- prometheus.MustNewConstMetric(countvec, prometheus.CounterValue, value, cluster, exchange.labels["vhost"], exchange.labels["name"])