Update wallet transactions when confirmed IS locks are removed

dashpay · Apr 5, 2019 · d34ec78 · d34ec78
1 parent b897505
commit d34ec78
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/llmq/quorums_instantsend.cpp b/src/llmq/quorums_instantsend.cpp
@@ -828,11 +828,12 @@ void CInstantSendManager::UpdatedBlockTip(const CBlockIndex* pindexNew)
 
 void CInstantSendManager::HandleFullyConfirmedBlock(const CBlockIndex* pindex)
 {
+    std::unordered_map<uint256, CInstantSendLockPtr> removeISLocks;
     {
         LOCK(cs);
 
-        auto islocks = db.RemoveConfirmedInstantSendLocks(pindex->nHeight);
-        for (auto& p : islocks) {
+        removeISLocks = db.RemoveConfirmedInstantSendLocks(pindex->nHeight);
+        for (auto& p : removeISLocks) {
             auto& islockHash = p.first;
             auto& islock = p.second;
             LogPrint("instantsend", "CInstantSendManager::%s -- txid=%s, islock=%s: removed islock as it got fully confirmed\n", __func__,
@@ -845,6 +846,10 @@ void CInstantSendManager::HandleFullyConfirmedBlock(const CBlockIndex* pindex)
         }
     }
 
+    for (auto& p : removeISLocks) {
+        UpdateWalletTransaction(p.second->txid, nullptr);
+    }
+
     // Retry all not yet locked mempool TXs and TX which where mined after the fully confirmed block
     RetryLockTxs(uint256());
 }