Force reindex when dex live does not present

Signed-off-by: Anthony Fieroni <[email protected]>
DeFiCh · Apr 27, 2022 · e8049b5 · e8049b5
1 parent 28d23a6
commit e8049b5
Showing 1 changed file with 18 additions and 3 deletions.
diff --git a/src/init.cpp b/src/init.cpp
@@ -27,6 +27,7 @@
 #include <key_io.h>
 #include <masternodes/accountshistory.h>
 #include <masternodes/anchors.h>
+#include <masternodes/govvariables/attributes.h>
 #include <masternodes/masternodes.h>
 #include <masternodes/vaulthistory.h>
 #include <miner.h>
@@ -1618,9 +1619,23 @@ bool AppInitMain(InitInterfaces& interfaces)
                 pcustomcsview.reset();
                 pcustomcsview = MakeUnique<CCustomCSView>(*pcustomcsDB.get());
                 if (!fReset && !fReindexChainState) {
-                    if (!pcustomcsDB->IsEmpty() && pcustomcsview->GetDbVersion() != CCustomCSView::DbVersion) {
-                        strLoadError = _("Account database is unsuitable").translated;
-                        break;
+                    if (!pcustomcsDB->IsEmpty()) {
+                        if (pcustomcsview->GetDbVersion() != CCustomCSView::DbVersion) {
+                            strLoadError = _("Account database is unsuitable").translated;
+                            break;
+                        }
+                        // force reindex iif there is at least one pool swap
+                        PoolHeightKey anyPoolSwap{DCT_ID{}, ~0u};
+                        auto it = pcustomcsview->LowerBound<CPoolPairView::ByPoolSwap>(anyPoolSwap);
+                        auto shouldReindex = it.Valid();
+                        if (auto attributes = pcustomcsview->GetAttributes()) {
+                            CDataStructureV0 dexKey{AttributeTypes::Live, ParamIDs::Economy, EconomyKeys::DexTokens};
+                            shouldReindex &= !attributes->CheckKey(dexKey);
+                        }
+                        if (shouldReindex) {
+                            strLoadError = _("Live dex needs reindex").translated;
+                            break;
+                        }
                     }
                 }