Fix load version for export genesis

sc/memiavl/db.go

@@ -209,6 +209,9 @@ func OpenDB(logger logger.Logger, targetVersion int64, opts Options) (*DB, error
 			return nil, errorutils.Join(err, db.Close())
 		}
 	}
+	if db.streamHandler == nil {
+		fmt.Println("[Debug] DB steam handler is nil??")
+	}
 	return db, nil
 }
 
@@ -613,10 +616,16 @@ func (db *DB) rewriteSnapshotBackground() error {
 func (db *DB) Close() error {
 	db.mtx.Lock()
 	defer db.mtx.Unlock()
+	errs := []error{}
 
-	errs := []error{
-		db.streamHandler.Close(),
+	// Close stream handler
+	if db.streamHandler != nil {
+		err := db.streamHandler.Close()
+		errs = append(errs, err)
+		db.streamHandler = nil
 	}
+
+	// Close rewrite channel
 	if db.snapshotRewriteChan != nil {
 		db.snapshotRewriteCancelFunc()
 		<-db.snapshotRewriteChan
@@ -626,8 +635,7 @@ func (db *DB) Close() error {
 
 	errs = append(errs, db.MultiTree.Close())
 
-	db.streamHandler = nil
-
+	// Close file lock
 	if db.fileLock != nil {
 		errs = append(errs, db.fileLock.Unlock())
 		errs = append(errs, db.fileLock.Destroy())

sc/store.go

@@ -1,6 +1,8 @@
 package sc
 
 import (
+	"fmt"
+
 	"github.com/sei-protocol/sei-db/common/logger"
 	"github.com/sei-protocol/sei-db/common/utils"
 	"github.com/sei-protocol/sei-db/config"
@@ -39,15 +41,8 @@ func NewCommitStore(homeDir string, logger logger.Logger, config config.StateCom
 	return commitStore
 }
 
-func (cs *CommitStore) Initialize(initialStores []string) error {
-	options := cs.opts
-	options.InitialStores = initialStores
-	db, err := memiavl.OpenDB(cs.logger, 0, options)
-	if err != nil {
-		return err
-	}
-	cs.db = db
-	return nil
+func (cs *CommitStore) Initialize(initialStores []string) {
+	cs.opts.InitialStores = initialStores
 }
 
 func (cs *CommitStore) SetInitialVersion(initialVersion int64) error {
@@ -70,22 +65,28 @@ func (cs *CommitStore) Rollback(targetVersion int64) error {
 
 // copyExisting is for creating new memiavl object given existing folder
 func (cs *CommitStore) LoadVersion(targetVersion int64, copyExisting bool) (types.Committer, error) {
-	opts := cs.opts
-	opts.ReadOnly = copyExisting
+	cs.logger.Info(fmt.Sprintf("SeiDB load target memIAVL version %d, copyExisting = %v\n", targetVersion, copyExisting))
 	if copyExisting {
+		opts := cs.opts
+		opts.ReadOnly = copyExisting
 		opts.CreateIfMissing = false
-	}
-	db, err := memiavl.OpenDB(cs.logger, targetVersion, opts)
-	if err != nil {
-		return nil, err
-	}
-	if copyExisting {
+		db, err := memiavl.OpenDB(cs.logger, targetVersion, opts)
+		if err != nil {
+			return nil, err
+		}
 		return &CommitStore{
 			logger: cs.logger,
 			db:     db,
 			opts:   opts,
 		}, nil
 	}
+	if cs.db != nil {
+		cs.db.Close()
+	}
+	db, err := memiavl.OpenDB(cs.logger, targetVersion, cs.opts)
+	if err != nil {
+		return nil, err
+	}
 	cs.db = db
 	return cs, nil
 }

sc/types/store.go

@@ -7,7 +7,7 @@ import (
 )
 
 type Committer interface {
-	Initialize(initialStores []string) error
+	Initialize(initialStores []string)
 
 	Commit() (int64, error)