refactor(server/v2/cometbft): typecast SC & SS to correct interfaces (c…

…osmos#21050)
surgebuild · Jul 24, 2024 · e001457 · e001457
1 parent 18dd65d
commit e001457
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/server/v2/cometbft/server.go b/server/v2/cometbft/server.go
@@ -72,6 +72,7 @@ func (s *CometBFTServer[T]) Init(appI serverv2.AppI[T], v *viper.Viper, logger l
 	}
 
 	s.logger = logger.With(log.ModuleKey, s.Name())
+	store := appI.GetStore().(types.Store)
 	consensus := NewConsensus(
 		s.logger,
 		appI.Name(),
@@ -80,7 +81,7 @@ func (s *CometBFTServer[T]) Init(appI serverv2.AppI[T], v *viper.Viper, logger l
 		s.serverOptions.Mempool,
 		indexEvents,
 		appI.GetGRPCQueryDecoders(),
-		appI.GetStore().(types.Store),
+		store,
 		s.config,
 		s.initTxCodec,
 	)
@@ -91,9 +92,8 @@ func (s *CometBFTServer[T]) Init(appI serverv2.AppI[T], v *viper.Viper, logger l
 	consensus.addrPeerFilter = s.serverOptions.AddrPeerFilter
 	consensus.idPeerFilter = s.serverOptions.IdPeerFilter
 
-	// TODO: set these; what is the appropriate presence of the Store interface here?
-	var ss snapshots.StorageSnapshotter
-	var sc snapshots.CommitSnapshotter
+	ss := store.GetStateStorage().(snapshots.StorageSnapshotter)
+	sc := store.GetStateCommitment().(snapshots.CommitSnapshotter)
 
 	snapshotStore, err := GetSnapshotStore(s.config.ConfigTomlConfig.RootDir)
 	if err != nil {