chaindb: add trigger to force rollback the tree for reorgs

handshake-org · Sep 10, 2019 · cccd8e5 · cccd8e5
1 parent cc35dc2
commit cccd8e5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/blockchain/chaindb.js b/lib/blockchain/chaindb.js
@@ -1825,9 +1825,10 @@ class ChainDB {
    * @private
    * @param {CoinView} view
    * @param {Number} height
+   * @param {Boolean} force - force a commitment threw (used for reorgs)
    */
 
-  async saveNames(view, height) {
+  async saveNames(view, height, force=false) {
     for (const ns of view.names.values()) {
       const {nameHash} = ns;
 
@@ -1839,7 +1840,7 @@ class ChainDB {
       await this.txn.insert(nameHash, ns.encode());
     }
 
-    if ((height % this.network.names.treeInterval) === 0)
+    if ((height % this.network.names.treeInterval) === 0 || force)
       this.put(layout.s.encode(), await this.txn.commit());
   }
 
@@ -1883,7 +1884,7 @@ class ChainDB {
       this.del(layout.w.encode(height));
     }
 
-    return this.saveNames(view, height - 1);
+    return this.saveNames(view, height - 1, true);
   }
 
   /**