optimize

Qiao-Jin · Feb 2, 2021 · 2354dfd · 2354dfd
1 parent 9692809
commit 2354dfd
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/neo/Network/P2P/TaskManager.cs b/src/neo/Network/P2P/TaskManager.cs
@@ -109,8 +109,13 @@ private void OnNewTasks(InvPayload payload)
         private void OnPersistCompleted(Block block)
         {
             foreach (var (actor, session) in sessions)
-                if (session.ReceivedBlock.Remove(block.Index, out Block receivedBlock) && block.Hash == receivedBlock.Hash)
-                    RequestTasks(actor, session);
+                if (session.ReceivedBlock.Remove(block.Index, out Block receivedBlock))
+                {
+                    if (block.Hash == receivedBlock.Hash)
+                        RequestTasks(actor, session);
+                    else
+                        actor.Tell(Tcp.Abort.Instance);
+                }
         }
 
         protected override void OnReceive(object message)