refactor: refactor func Flush() of metric pkg

KyberNetwork · Aug 17, 2023 · 92d4e70 · 92d4e70
1 parent a8a6e18
commit 92d4e70
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/pkg/metric/meter.go b/pkg/metric/meter.go
@@ -2,6 +2,7 @@ package metric
 
 import (
 	"context"
+	"errors"
 
 	"go.opentelemetry.io/otel"
 	"go.opentelemetry.io/otel/metric"
@@ -16,5 +17,9 @@ func Meter() metric.Meter {
 }
 
 func Flush(ctx context.Context) error {
-	return otel.GetMeterProvider().(*metricsdk.MeterProvider).ForceFlush(ctx)
+	if m, ok := otel.GetMeterProvider().(*metricsdk.MeterProvider); ok {
+		return m.ForceFlush(ctx)
+	} else {
+		return errors.New("no meter provider was initialized")
+	}
 }