Add HandlerWithAllowlist which can control exposed metrics

Author: CatherineF-dev

prometheus/promhttp/http.go

@@ -37,6 +37,7 @@ import (
 	"fmt"
 	"io"
 	"net/http"
+	"regexp"
 	"strconv"
 	"strings"
 	"sync"
@@ -45,6 +46,8 @@ import (
 	"github.com/prometheus/common/expfmt"
 
 	"github.com/prometheus/client_golang/prometheus"
+
+	dto "github.com/prometheus/client_model/go"
 )
 
 const (
@@ -90,6 +93,13 @@ func HandlerFor(reg prometheus.Gatherer, opts HandlerOpts) http.Handler {
 	return HandlerForTransactional(prometheus.ToTransactionalGatherer(reg), opts)
 }
 
+// HandlerWithAllowlist remove metrics whose name can't match with metricNameMatcher
+func HandlerWithAllowlist(metricNameMatcher *regexp.Regexp) http.Handler {
+	return InstrumentMetricHandler(
+		prometheus.DefaultRegisterer, HandlerFor(&FilteredGatherer{gather: prometheus.DefaultGatherer, metricNameMatcher: metricNameMatcher}, HandlerOpts{}),
+	)
+}
+
 // HandlerForTransactional is like HandlerFor, but it uses transactional gather, which
 // can safely change in-place returned *dto.MetricFamily before call to `Gather` and after
 // call to `done` of that `Gather`.
@@ -406,3 +416,32 @@ func httpError(rsp http.ResponseWriter, err error) {
 		http.StatusInternalServerError,
 	)
 }
+
+type FilteredGatherer struct {
+	gather         prometheus.Gatherer
+	metricNameMatcher *regexp.Regexp
+}
+
+// NewFilteredGatherer creates FilteredGatherer.
+func NewFilteredGatherer(gather prometheus.Gatherer, metricNameMatcher *regexp.Regexp) *FilteredGatherer {
+	return &FilteredGatherer{
+		gather:         gather,
+		metricNameMatcher: metricNameMatcher,
+	}
+}
+
+func (f *FilteredGatherer) Gather() ([]*dto.MetricFamily, error) {
+	results := []*dto.MetricFamily{}
+	metrics, err := f.gather.Gather()
+	if err != nil {
+		return nil, err
+	}
+	for _, m := range metrics {
+
+		if f.metricNameMatcher.MatchString(*m.Name) {
+			results = append(results, m)
+		}
+
+	}
+	return results, nil
+}