fix node db (#40)

* fix node db * delete da too
initia-labs · Nov 6, 2024 · e03e229 · e03e229
1 parent 0b2fa08
commit e03e229
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/challenger/db.go b/challenger/db.go
@@ -133,11 +133,11 @@ func ResetHeight(db types.DB, nodeName string) error {
 	}
 	nodeDB := db.WithPrefix([]byte(nodeName))
 
-	if err := DeletePendingEvents(db); err != nil {
+	if err := DeletePendingEvents(nodeDB); err != nil {
 		return err
 	}
 
-	if err := DeletePendingChallenges(db); err != nil {
+	if err := DeletePendingChallenges(nodeDB); err != nil {
 		return err
 	}
 

diff --git a/executor/db.go b/executor/db.go
@@ -13,6 +13,8 @@ func ResetHeights(db types.DB) error {
 		types.HostName,
 		types.ChildName,
 		types.BatchName,
+		types.DAHostName,
+		types.DACelestiaName,
 	}
 	for _, dbName := range dbNames {
 		if err := ResetHeight(db, dbName); err != nil {
@@ -25,18 +27,20 @@ func ResetHeights(db types.DB) error {
 func ResetHeight(db types.DB, nodeName string) error {
 	if nodeName != types.HostName &&
 		nodeName != types.ChildName &&
-		nodeName != types.BatchName {
+		nodeName != types.BatchName &&
+		nodeName != types.DAHostName &&
+		nodeName != types.DACelestiaName {
 		return errors.New("unknown node name")
 	}
 	nodeDB := db.WithPrefix([]byte(nodeName))
 	err := node.DeleteSyncInfo(nodeDB)
 	if err != nil {
 		return err
 	}
-	if err := node.DeletePendingTxs(db); err != nil {
+	if err := node.DeletePendingTxs(nodeDB); err != nil {
 		return err
 	}
-	if err := node.DeleteProcessedMsgs(db); err != nil {
+	if err := node.DeleteProcessedMsgs(nodeDB); err != nil {
 		return err
 	}
 	fmt.Printf("reset height to 0 for node %s\n", string(nodeDB.GetPrefix()))