Call unhook for all evicted elements in SpendDAG.

iotaledger · Apr 18, 2024 · ab053d6 · ab053d6
1 parent e2eb24c
commit ab053d6
Showing 1 changed file with 6 additions and 7 deletions.
diff --git a/pkg/protocol/engine/mempool/spenddag/spenddagv1/spenddag.go b/pkg/protocol/engine/mempool/spenddag/spenddagv1/spenddag.go
@@ -412,13 +412,12 @@ func (c *SpendDAG[SpenderID, ResourceID, VoteRank]) evictSpender(spenderID Spend
 	// remove the spenders from the spenddag dictionary
 	for _, evictedSpenderID := range evictedSpenderIDs {
 		c.spendersByID.Delete(evictedSpenderID)
-	}
-
-	// unhook the spend events and remove the unhook method from the storage
-	unhookFunc, unhookExists := c.spendUnhooks.Get(spenderID)
-	if unhookExists {
-		unhookFunc()
-		c.spendUnhooks.Delete(spenderID)
+		// unhook the spend events and remove the unhook method from the storage
+		unhookFunc, unhookExists := c.spendUnhooks.Get(spenderID)
+		if unhookExists {
+			unhookFunc()
+			c.spendUnhooks.Delete(spenderID)
+		}
 	}
 
 	return evictedSpenderIDs