Ensuring Object Reference check of SortedSets for speed-up (#557)

neo-project · Jan 17, 2019 · 0e5434c · 0e5434c
1 parent ab8bb4d
commit 0e5434c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/neo/Ledger/MemoryPool.cs b/neo/Ledger/MemoryPool.cs
@@ -279,7 +279,7 @@ private PoolItem GetLowestFeeTransaction(out Dictionary<UInt256, PoolItem> unsor
 
             if (minItem != null)
             {
-                unsortedTxPool = sortedPool == _unverifiedSortedLowPriorityTransactions
+                unsortedTxPool = Object.ReferenceEquals(sortedPool, _unverifiedSortedLowPriorityTransactions)
                     ? _unverifiedTransactions : _unsortedTransactions;
                 return minItem;
             }
@@ -291,7 +291,7 @@ private PoolItem GetLowestFeeTransaction(out Dictionary<UInt256, PoolItem> unsor
             }
             finally
             {
-                unsortedTxPool = sortedPool == _unverifiedSortedHighPriorityTransactions
+                unsortedTxPool = Object.ReferenceEquals(sortedPool, _unverifiedSortedHighPriorityTransactions)
                     ? _unverifiedTransactions : _unsortedTransactions;
             }
         }
@@ -446,7 +446,7 @@ private int ReverifyTransactions(SortedSet<PoolItem> verifiedSortedTxPool,
             _txRwLock.EnterWriteLock();
             try
             {
-                int blocksTillRebroadcast = unverifiedSortedTxPool == _sortedHighPrioTransactions
+                int blocksTillRebroadcast = Object.ReferenceEquals(unverifiedSortedTxPool, _sortedHighPrioTransactions)
                     ? BlocksTillRebroadcastHighPriorityPoolTx : BlocksTillRebroadcastLowPriorityPoolTx;
 
                 if (Count > RebroadcastMultiplierThreshold)