[Model] Cached cold staking delegations, counting every p2cs record and not only delegation records + update cachedDelegations list using object reference.

furszy · furszy · commit fd2ca983122d · 2019-11-08T18:24:44.000-03:00
diff --git a/src/qt/transactiontablemodel.cpp b/src/qt/transactiontablemodel.cpp
@@ -160,7 +160,8 @@ class TransactionTablePriv
                     }
 
                     // Check for delegations
-                    if (record.type == TransactionRecord::P2CSDelegation || record.type == TransactionRecord::P2CSDelegationSent) {
+                    if (record.type == TransactionRecord::P2CSDelegation || record.type == TransactionRecord::P2CSDelegationSent
+                        || record.type == TransactionRecord::StakeDelegated || record.type == TransactionRecord::StakeHot) {
                         checkForDelegations(record, wallet, cachedDelegations);
                     }
                 }
@@ -177,7 +178,7 @@ class TransactionTablePriv
 
     static void checkForDelegations(const TransactionRecord& record, const CWallet* wallet, QList<CSDelegation>& cachedDelegations) {
         CSDelegation delegation(false, record.address);
-        delegation.isSpendable = record.type == TransactionRecord::P2CSDelegationSent;
+        delegation.isSpendable = record.type == TransactionRecord::P2CSDelegationSent || record.type == TransactionRecord::StakeDelegated;
 
         // Append only stakeable utxo and not every output of the record
         const QString& hashTxId = record.getTxID();
@@ -208,7 +209,7 @@ class TransactionTablePriv
         if (index == -1) {
             cachedDelegations.append(delegation);
         } else {
-            CSDelegation del = cachedDelegations[index];
+            CSDelegation& del = cachedDelegations[index];
             del.delegatedUtxo.unite(delegation.delegatedUtxo);
             del.cachedTotalAmount += delegation.cachedTotalAmount;
         }
@@ -291,7 +292,8 @@ class TransactionTablePriv
                             if (!hasZcTxes) hasZcTxes = HasZcTxesIfNeeded(rec);
 
                         // Check for delegations
-                        if (rec.type == TransactionRecord::P2CSDelegation || rec.type == TransactionRecord::P2CSDelegationSent) {
+                        if (rec.type == TransactionRecord::P2CSDelegation || rec.type == TransactionRecord::P2CSDelegationSent
+                            || rec.type == TransactionRecord::StakeDelegated || rec.type == TransactionRecord::StakeHot) {
                             checkForDelegations(rec, wallet, cachedDelegations);
                         }
 
