refactor: switch to using Batching datastore

celestiaorg · Jan 27, 2023 · 179fd50 · 179fd50
1 parent 50c1db3
commit 179fd50
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/nodebuilder/node/uptime.go b/nodebuilder/node/uptime.go
@@ -29,7 +29,7 @@ type UptimeMetrics struct {
 	totalNodeUptime asyncfloat64.Counter
 
 	// store is the datastore used to store the node uptime metrics.
-	store datastore.Datastore
+	store datastore.Batching
 }
 
 var (
@@ -40,7 +40,7 @@ var (
 
 // NewUptimeMetrics creates a new UptimeMetrics
 // and registers a callback to re-meter the totalNodeUptime metric.
-func NewUptimeMetrics(ds datastore.Datastore) (*UptimeMetrics, error) {
+func NewUptimeMetrics(ds datastore.Batching) (*UptimeMetrics, error) {
 	nodeStartTS, err := meter.
 		AsyncFloat64().
 		Gauge(

diff --git a/nodebuilder/settings.go b/nodebuilder/settings.go
@@ -44,7 +44,7 @@ func WithMetrics(metricOpts []otlpmetrichttp.Option, nodeType node.Type) fx.Opti
 		fx.Invoke(header.WithMetrics),
 		fx.Invoke(state.WithMetrics),
 		fx.Invoke(fraud.WithMetrics),
-		fx.Invoke(func(ctx context.Context, ds datastore.Datastore) error {
+		fx.Invoke(func(ctx context.Context, ds datastore.Batching) error {
 			m, err := node.NewUptimeMetrics(ds)
 			if err != nil {
 				return err