fix: metrics上报支持按业务基本黑名单设置 (#171)

Author: Ambition9186

cmd/feed-server/service/metric.go

@@ -20,7 +20,7 @@ import (
 )
 
 // nolint:funlen
-func initMetric(name string) *metric {
+func initMetric(name string, blacklistBizIds []uint32) *metric {
 	m := new(metric)
 	labels := prm.Labels{"name": name}
 	m.watchTotal = prometheus.NewGaugeVec(prometheus.GaugeOpts{
@@ -141,6 +141,12 @@ func initMetric(name string) *metric {
 		}, versionChange)
 	metrics.Register().MustRegister(m.changeTotalSeconds)
 
+	black := make(map[uint32]struct{}, len(blacklistBizIds))
+	for _, id := range blacklistBizIds {
+		black[id] = struct{}{}
+	}
+	m.blacklist = black
+
 	return m
 }
 
@@ -170,6 +176,7 @@ type metric struct {
 	changeTotalFileSize *prometheus.HistogramVec
 	// 变更总耗时
 	changeTotalSeconds *prometheus.HistogramVec
+	blacklist          map[uint32]struct{}
 }
 
 // collectDownload collects metrics for download
@@ -178,3 +185,8 @@ func (m *metric) collectDownload(biz string, totalSize, delayRequests, delayMill
 	m.downloadDelayRequests.With(prm.Labels{"bizID": biz}).Set(float64(delayRequests))
 	m.downloadDelayMilliseconds.With(prm.Labels{"bizID": biz}).Set(float64(delayMilliseconds))
 }
+
+func (m *metric) shouldReport(bizID uint32) bool {
+	_, exists := m.blacklist[bizID]
+	return !exists
+}

cmd/feed-server/service/rpc_sidecar.go

@@ -1069,6 +1069,9 @@ func (s *Service) GetSingleFileContent(req *pbfs.GetSingleFileContentReq,
 }
 
 func (s *Service) handleResourceUsageMetrics(bizID uint32, appName string, resource sfs.ResourceUsage) {
+	if !s.mc.shouldReport(bizID) {
+		return
+	}
 	s.mc.clientMaxCPUUsage.WithLabelValues(strconv.Itoa(int(bizID)), appName).Set(resource.CpuMaxUsage)
 	s.mc.clientCurrentCPUUsage.WithLabelValues(strconv.Itoa(int(bizID)), appName).Set(resource.CpuUsage)
 	s.mc.clientMaxMemUsage.WithLabelValues(strconv.Itoa(int(bizID)), appName).Set(float64(resource.MemoryMaxUsage))
@@ -1077,6 +1080,9 @@ func (s *Service) handleResourceUsageMetrics(bizID uint32, appName string, resou
 
 // 暴露客户端版本变更事件到metrics
 func (s *Service) clientEventChangeRecord(basicData *sfs.BasicData, appMeta *sfs.SideAppMeta) {
+	if !s.mc.shouldReport(basicData.BizID) {
+		return
+	}
 	if basicData == nil && appMeta == nil {
 		return
 	}

cmd/feed-server/service/service.go

@@ -102,7 +102,7 @@ func NewService(sd serviced.Discover, name string) (*Service, error) {
 		state:      state,
 		name:       name,
 		provider:   provider,
-		mc:         initMetric(name),
+		mc:         initMetric(name, cc.FeedServer().Metric.BlacklistBizIDs),
 		gwMux:      gwMux,
 		rl:         rl,
 	}, nil

pkg/cc/service.go

@@ -391,6 +391,7 @@ type FeedServerSetting struct {
 	Downstream   Downstream          `yaml:"downstream"`
 	MRLimiter    MatchReleaseLimiter `yaml:"matchReleaseLimiter"`
 	RateLimiter  RateLimiter         `yaml:"rateLimiter"`
+	Metric       Metric              `yaml:"metrics"`
 }
 
 // trySetFlagBindIP try set flag bind ip.

pkg/cc/types.go

@@ -1156,6 +1156,11 @@ type RateLimiter struct {
 	IP              BasicRL `yaml:"ip"`
 }
 
+// metrics 上报时过滤的业务名单
+type Metric struct {
+	BlacklistBizIDs []uint32 `yaml:"blacklistBizIds"`
+}
+
 // BizRLs defines the rate limiters for biz
 type BizRLs struct {
 	Default BasicRL          `yaml:"default"`