feat: add only_include_metrics

README.md

@@ -4,13 +4,11 @@
 
 ```yaml
 rate_limit: 20
-
 cdn:
   metrics:
-    - 2xx
-    - 3xx
-    - 4xx
     - 5xx
+  only_include_metrics:
+    - 566
   delay_seconds: 360
   custom_query_dimensions:
     - projectId: id

main.go

@@ -6,7 +6,6 @@ import (
 	"github.com/go-kit/log/level"
 	"github.com/joho/godotenv"
 	"github.com/prometheus/client_golang/prometheus"
-	"github.com/prometheus/client_golang/prometheus/collectors"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 	"github.com/prometheus/common/promlog"
 	"github.com/prometheus/common/promlog/flag"
@@ -46,7 +45,6 @@ func main() {
 	// load config file
 	if err := tencentConfig.LoadFile(*configFile); err != nil {
 		_ = level.Error(logger).Log("msg", "Load config error", "err", err)
-		os.Exit(1)
 	} else {
 		_ = level.Info(logger).Log("msg", "Load config ok")
 	}
@@ -68,8 +66,6 @@ func main() {
 	}
 
 	prometheus.MustRegister(version.NewCollector(config.NameSpace))
-	prometheus.Unregister(collectors.NewGoCollector())
-	prometheus.Unregister(collectors.NewProcessCollector(collectors.ProcessCollectorOpts{}))
 
 	if *enableCbs {
 		cpf := profile.NewClientProfile()

pkg/collector/handler_cdn.go

@@ -27,6 +27,15 @@ type CdnExporter struct {
 	cdnInstance *prometheus.Desc
 }
 
+func contains(slice []string, item string) bool {
+	for _, s := range slice {
+		if s == item {
+			return true
+		}
+	}
+	return false
+}
+
 func NewCdnExporter(logger log.Logger, credential common.CredentialIface, tencentConfig config.TencentConfig) *CdnExporter {
 	return &CdnExporter{
 		logger:        logger,
@@ -120,6 +129,9 @@ func (e *CdnExporter) Collect(ch chan<- prometheus.Metric) {
 				}
 
 				for _, data := range cdnResponse.Response.Data[0].CdnData {
+					if !contains(e.tencentConfig.CDN.OnlyIncludeMetrics, *data.Metric) {
+						continue
+					}
 					ch <- prometheus.MustNewConstMetric(
 						e.cdnInstance,
 						prometheus.GaugeValue,

pkg/config/config.go

@@ -16,6 +16,7 @@ type CustomQueryDimension struct {
 
 type CDNConfig struct {
 	Metrics              []string               `yaml:"metrics"`
+	OnlyIncludeMetrics   []string               `yaml:"only_include_metrics"`
 	DelaySeconds         int                    `yaml:"delay_seconds"`
 	CustomQueryDimension []CustomQueryDimension `yaml:"custom_query_dimensions"`
 }