clear onroad db before loading onroad

vitelabs · Jul 20, 2022 · 3a8bd37 · 3a8bd37
1 parent fb6e2b6
commit 3a8bd37
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/ledger/chain/chain.go b/ledger/chain/chain.go
@@ -230,6 +230,18 @@ func (c *chain) NewDb(dirName string) (*leveldb.DB, error) {
 	return db, nil
 }
 
+func (c *chain) PrepareOnroadDb() (*leveldb.DB, error) {
+	dirName := "onroad"
+	absoluteDirName := path.Join(c.chainDir, dirName)
+	c.log.Info("clear onroad db", "dir", absoluteDirName)
+
+	err := os.RemoveAll(absoluteDirName)
+	if err != nil {
+		return nil, err
+	}
+	return c.NewDb(dirName)
+}
+
 func (c *chain) SetConsensus(verifier interfaces.ConsensusVerifier, periodTimeIndex interfaces.TimeIndex) {
 	c.log.Info("Start set consensus", "method", "SetConsensus")
 	c.verifier = verifier

diff --git a/ledger/chain/interface.go b/ledger/chain/interface.go
@@ -282,6 +282,8 @@ type Chain interface {
 
 	NewDb(dirName string) (*leveldb.DB, error)
 
+	PrepareOnroadDb() (*leveldb.DB, error)
+
 	Plugins() *chain_plugins.Plugins
 
 	SetConsensus(verifier interfaces.ConsensusVerifier, periodTimeIndex interfaces.TimeIndex)

diff --git a/ledger/onroad/manager.go b/ledger/onroad/manager.go
@@ -102,13 +102,13 @@ func (manager *Manager) Close() error {
 }
 
 func (manager *Manager) prepareOnRoadPool(gid types.Gid) {
-	db, err := manager.chain.NewDb("onroad")
+	db, err := manager.chain.PrepareOnroadDb()
 	if err != nil {
 		panic(fmt.Sprintf("new onroad storage fail, err is %v", err))
 	}
 	manager.db = db
 	orPool, exist := manager.onRoadPools.Load(gid)
-	manager.log.Info(fmt.Sprintf("prepareOnRoadPool"), "gid", gid, "exist", exist, "orPool", orPool)
+	manager.log.Info("prepare OnRoadPool", "gid", gid, "exist", exist, "orPool", orPool)
 	if !exist || orPool == nil {
 		manager.onRoadPools.Store(gid, onroad_pool.NewContractOnRoadPool(gid, manager.chain, db))
 		return