Change to time provider (neo-project#1943)

ShawnYun · Jan 8, 2021 · 5bc6d71 · 5bc6d71
1 parent 89d1fb9
commit 5bc6d71
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 9 deletions.
diff --git a/src/neo/Ledger/MemoryPool.cs b/src/neo/Ledger/MemoryPool.cs
@@ -415,7 +415,7 @@ internal void InvalidateAllTransactions()
         private int ReverifyTransactions(SortedSet<PoolItem> verifiedSortedTxPool,
             SortedSet<PoolItem> unverifiedSortedTxPool, int count, double millisecondsTimeout, StoreView snapshot)
         {
-            DateTime reverifyCutOffTimeStamp = DateTime.UtcNow.AddMilliseconds(millisecondsTimeout);
+            DateTime reverifyCutOffTimeStamp = TimeProvider.Current.UtcNow.AddMilliseconds(millisecondsTimeout);
             List<PoolItem> reverifiedItems = new List<PoolItem>(count);
             List<PoolItem> invalidItems = new List<PoolItem>();
 
@@ -433,15 +433,15 @@ private int ReverifyTransactions(SortedSet<PoolItem> verifiedSortedTxPool,
                     else // Transaction no longer valid -- it will be removed from unverifiedTxPool.
                         invalidItems.Add(item);
 
-                    if (DateTime.UtcNow > reverifyCutOffTimeStamp) break;
+                    if (TimeProvider.Current.UtcNow > reverifyCutOffTimeStamp) break;
                 }
 
                 int blocksTillRebroadcast = BlocksTillRebroadcast;
                 // Increases, proportionally, blocksTillRebroadcast if mempool has more items than threshold bigger RebroadcastMultiplierThreshold
                 if (Count > RebroadcastMultiplierThreshold)
                     blocksTillRebroadcast = blocksTillRebroadcast * Count / RebroadcastMultiplierThreshold;
 
-                var rebroadcastCutOffTime = DateTime.UtcNow.AddMilliseconds(
+                var rebroadcastCutOffTime = TimeProvider.Current.UtcNow.AddMilliseconds(
                     -Blockchain.MillisecondsPerBlock * blocksTillRebroadcast);
                 foreach (PoolItem item in reverifiedItems)
                 {
@@ -452,7 +452,7 @@ private int ReverifyTransactions(SortedSet<PoolItem> verifiedSortedTxPool,
                         if (item.LastBroadcastTimestamp < rebroadcastCutOffTime)
                         {
                             _system.LocalNode.Tell(new LocalNode.RelayDirectly { Inventory = item.Tx }, _system.Blockchain);
-                            item.LastBroadcastTimestamp = DateTime.UtcNow;
+                            item.LastBroadcastTimestamp = TimeProvider.Current.UtcNow;
                         }
                     }
                     else

diff --git a/src/neo/Network/P2P/Payloads/PingPayload.cs b/src/neo/Network/P2P/Payloads/PingPayload.cs
@@ -27,7 +27,7 @@ public static PingPayload Create(uint height, uint nonce)
             return new PingPayload
             {
                 LastBlockIndex = height,
-                Timestamp = DateTime.UtcNow.ToTimestamp(),
+                Timestamp = TimeProvider.Current.UtcNow.ToTimestamp(),
                 Nonce = nonce
             };
         }

diff --git a/src/neo/Network/P2P/RemoteNode.ProtocolHandler.cs b/src/neo/Network/P2P/RemoteNode.ProtocolHandler.cs
@@ -319,7 +319,7 @@ private void OnInvMessageReceived(InvPayload payload)
             }
             if (hashes.Length == 0) return;
             foreach (UInt256 hash in hashes)
-                pendingKnownHashes.Add((hash, DateTime.UtcNow));
+                pendingKnownHashes.Add((hash, TimeProvider.Current.UtcNow));
             system.TaskManager.Tell(new TaskManager.NewTasks { Payload = InvPayload.Create(payload.Type, hashes) });
         }
 
@@ -377,7 +377,7 @@ private void RefreshPendingKnownHashes()
             while (pendingKnownHashes.Count > 0)
             {
                 var (_, time) = pendingKnownHashes[0];
-                if (DateTime.UtcNow - time <= PendingTimeout)
+                if (TimeProvider.Current.UtcNow - time <= PendingTimeout)
                     break;
                 pendingKnownHashes.RemoveAt(0);
             }

diff --git a/src/neo/Network/P2P/TaskManager.cs b/src/neo/Network/P2P/TaskManager.cs
@@ -110,7 +110,7 @@ private void OnNewTasks(InvPayload payload)
             foreach (UInt256 hash in hashes)
             {
                 IncrementGlobalTask(hash);
-                session.InvTasks[hash] = DateTime.UtcNow;
+                session.InvTasks[hash] = TimeProvider.Current.UtcNow;
             }
 
             foreach (InvPayload group in InvPayload.CreateGroup(payload.Type, hashes.ToArray()))
@@ -251,7 +251,7 @@ private void OnTimer()
 
                 foreach (var task in session.InvTasks.ToArray())
                 {
-                    if (DateTime.UtcNow - task.Value > TaskTimeout)
+                    if (TimeProvider.Current.UtcNow - task.Value > TaskTimeout)
                     {
                         if (session.InvTasks.Remove(task.Key))
                             DecrementGlobalTask(task.Key);