fix(core-transaction-pool): copy vote target into temp wallet manager

ArkEcosystem · Jun 7, 2019 · 1209a36 · 1209a36
1 parent f9414f0
commit 1209a36
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/packages/core-transaction-pool/src/connection.ts b/packages/core-transaction-pool/src/connection.ts
@@ -541,6 +541,14 @@ export class Connection implements TransactionPool.IConnection {
             localWalletManager.reindex(sender);
         }
 
+        // HACK: need tx agonistic way for wallets which are modified by transaction
+        if (transaction.type === Enums.TransactionTypes.Vote) {
+            const vote = transaction.data.asset.votes[0].slice(1);
+            if (!localWalletManager.hasByPublicKey(vote)) {
+                localWalletManager.reindex(clonedeep(databaseWalletManager.findByPublicKey(vote)));
+            }
+        }
+
         if (recipientId) {
             if (localWalletManager.hasByAddress(recipientId)) {
                 recipient = localWalletManager.findByAddress(recipientId);