feat: add logic of prometheus counter, histogram, and update gauge

Author: coderi421

gmicro/core/metric/counter.go

@@ -0,0 +1,43 @@
+package metric
+
+import "github.com/prometheus/client_golang/prometheus"
+
+type (
+	CounterVecOpts VectorOpts // CounterVecOpts 计数选项
+
+	// CounterVec 计数向量重写了 prometheus.CounterVec
+	CounterVec interface {
+		Inc(labels ...string)            // Inc 增加1标签
+		Add(v float64, labels ...string) // Add 添加任意值标签
+	}
+
+	// promCounterVec 封装了 prometheus.CounterVec
+	promCounterVec struct {
+		counter *prometheus.CounterVec
+	}
+)
+
+// NewCounterVec 封装 prometheus.NewCounterVec 为了增加 labels 参数
+func NewCounterVec(cfg *CounterVecOpts) CounterVec {
+	if cfg == nil {
+		return nil
+	}
+
+	vec := prometheus.NewCounterVec(prometheus.CounterOpts{
+		Namespace: cfg.Namespace,
+		Subsystem: cfg.Subsystem,
+		Name:      cfg.Name,
+		Help:      cfg.Help,
+	}, cfg.Labels)
+	prometheus.MustRegister(vec)
+
+	return &promCounterVec{counter: vec}
+}
+
+func (p *promCounterVec) Inc(labels ...string) {
+	p.counter.WithLabelValues(labels...).Inc()
+}
+
+func (p *promCounterVec) Add(v float64, labels ...string) {
+	p.counter.WithLabelValues(labels...).Add(v)
+}

gmicro/core/metric/gauge.go

@@ -7,8 +7,10 @@ type (
 
 	GaugeVec interface {
 		Set(v float64, labels ...string) // Set 设置标签
-		Inc(labels ...string)            // Inc 增加标签
-		Add(v float64, labels ...string) // Add 添加标签
+		Inc(labels ...string)            // Inc 增加1标签
+		Dec(labels ...string)            // Dec 减少1标签
+		Add(v float64, labels ...string) // Add 添加任意值标签
+		Sub(v float64, labels ...string) // Sub 减少任意标签
 	}
 
 	promGaugeVec struct {
@@ -34,14 +36,22 @@ func NewGaugeVec(cfg *GaugeVecOpts) GaugeVec {
 	return &promGaugeVec{gauge: vec}
 }
 
-func (p promGaugeVec) Set(v float64, labels ...string) {
+func (p *promGaugeVec) Set(v float64, labels ...string) {
 	p.gauge.WithLabelValues(labels...).Set(v)
 }
 
-func (p promGaugeVec) Inc(labels ...string) {
+func (p *promGaugeVec) Inc(labels ...string) {
 	p.gauge.WithLabelValues(labels...).Inc()
 }
 
-func (p promGaugeVec) Add(v float64, labels ...string) {
+func (p *promGaugeVec) Dec(labels ...string) {
+	p.gauge.WithLabelValues(labels...).Dec()
+}
+
+func (p *promGaugeVec) Add(v float64, labels ...string) {
 	p.gauge.WithLabelValues(labels...).Add(v)
 }
+
+func (p *promGaugeVec) Sub(v float64, labels ...string) {
+	p.gauge.WithLabelValues(labels...).Sub(v)
+}

gmicro/core/metric/histogram.go

@@ -0,0 +1,43 @@
+package metric
+
+import "github.com/prometheus/client_golang/prometheus"
+
+type (
+	// HistogramVecOpts 柱状图选项
+	HistogramVecOpts struct {
+		VectorOpts
+		Buckets []float64 //柱状图指标新增桶的概念
+	}
+
+	// A HistogramVec interface represents a histogram vector.
+	HistogramVec interface {
+		// Observe adds observation v to labels.
+		Observe(v int64, labels ...string)
+	}
+
+	promHistogramVec struct {
+		histogram *prometheus.HistogramVec
+	}
+)
+
+// NewHistogramVec returns a HistogramVec.
+func NewHistogramVec(cfg *HistogramVecOpts) HistogramVec {
+	if cfg == nil {
+		return nil
+	}
+
+	vec := prometheus.NewHistogramVec(prometheus.HistogramOpts{
+		Namespace: cfg.Namespace,
+		Subsystem: cfg.Subsystem,
+		Name:      cfg.Name,
+		Help:      cfg.Help,
+		Buckets:   cfg.Buckets,
+	}, cfg.Labels)
+	prometheus.MustRegister(vec)
+
+	return &promHistogramVec{histogram: vec}
+}
+
+func (p promHistogramVec) Observe(v int64, labels ...string) {
+	p.histogram.WithLabelValues(labels...).Observe(float64(v))
+}