add option metric

KyberNetwork · Nov 5, 2024 · d3c9d57 · d3c9d57
1 parent 9ca69ce
commit d3c9d57
Showing 1 changed file with 19 additions and 8 deletions.
diff --git a/pkg/metrics/metrics.go b/pkg/metrics/metrics.go
@@ -13,38 +13,49 @@ func init() {
 	m = kybermetric.Meter()
 }
 
-func RecordFloat64Histogram(ctx context.Context, name string, value float64) error {
+func RecordFloat64Histogram(ctx context.Context, name string, value float64, options ...metric.RecordOption) error {
 	hist, err := m.Float64Histogram(name)
 	if err != nil {
 		return err
 	}
-	hist.Record(ctx, value)
+	hist.Record(ctx, value, options...)
 	return nil
 }
 
-func RecordFloat64Gause(ctx context.Context, name string, value float64) error {
+func RecordFloat64Gause(ctx context.Context, name string, value float64, options ...metric.ObserveOption) error {
 	_, err := m.Float64ObservableGauge(name,
 		metric.WithFloat64Callback(func(ctx context.Context, fo metric.Float64Observer) error {
-			fo.Observe(value)
+			fo.Observe(value, options...)
 			return nil
 		}))
 	return err
 }
 
-func RecordCounter(ctx context.Context, name string, value int64) error {
+func RecordCounter(ctx context.Context, name string, value int64, options ...metric.AddOption) error {
 	counter, err := m.Int64Counter(name)
 	if err != nil {
 		return err
 	}
-	counter.Add(ctx, value)
+	counter.Add(ctx, value, options...)
 	return nil
 }
 
-func RecordUpdownCounter(ctx context.Context, name string, value int64) error {
+func RecordUpdownCounter(ctx context.Context, name string, value int64, options ...metric.AddOption) error {
 	counter, err := m.Int64UpDownCounter(name)
 	if err != nil {
 		return err
 	}
-	counter.Add(ctx, value)
+	counter.Add(ctx, value, options...)
+	return nil
+}
+
+func RecordInt64HistogramWithMetricOption(ctx context.Context, name string, value int64,
+	options ...metric.Int64HistogramOption,
+) error {
+	hist, err := m.Int64Histogram(name, options...)
+	if err != nil {
+		return err
+	}
+	hist.Record(ctx, value)
 	return nil
 }