feat: prevent failed txn being calculated

stinkymonkeyph · Aug 6, 2024 · 5d07ef7 · 5d07ef7
1 parent 55d0620
commit 5d07ef7
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/blockchain/walletIndex.go b/blockchain/walletIndex.go
@@ -1,6 +1,10 @@
 package blockchain
 
-import "log"
+import (
+	"log"
+
+	"github.com/stinkymonkeyph/gopher-blocks/constants"
+)
 
 type WalletTransactionIndex struct {
 	BlockHeight      int          `json:"block_height"`
@@ -38,9 +42,9 @@ func (w *WalletIndex) CalculateBalance(address string) int {
 	bal := 0
 
 	for _, txn := range w.Transactions[address] {
-		if txn.Transaction.From == address {
+		if txn.Transaction.From == address && txn.Transaction.Status == constants.STATUS_SUCCESS {
 			bal -= int(txn.Transaction.Value)
-		} else if txn.Transaction.To == address {
+		} else if txn.Transaction.To == address && txn.Transaction.Status == constants.STATUS_SUCCESS {
 			bal += int(txn.Transaction.Value)
 		}
 	}