Optimize dictionary access

neo-project · Mar 30, 2020 · def1ebf · def1ebf
1 parent 007ff3e
commit def1ebf
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/src/RpcServer/RelayActor.cs b/src/RpcServer/RelayActor.cs
@@ -21,17 +21,19 @@ protected override void OnReceive(object message)
             switch (message)
             {
                 case IInventory inventory:
-                    this.senders.Add(inventory.Hash, Sender);
-                    neoSystem.Blockchain.Tell(inventory);
-                    break;
+                    {
+                        senders.Add(inventory.Hash, Sender);
+                        neoSystem.Blockchain.Tell(inventory);
+                        break;
+                    }
                 case RelayResult reason:
-                    UInt256 hash = reason.Inventory.Hash;
-                    if (senders.ContainsKey(reason.Inventory.Hash))
                     {
-                        senders[hash].Tell(reason);
-                        senders.Remove(hash);
+                        if (senders.Remove(reason.Inventory.Hash, out var actor))
+                        {
+                            actor.Tell(reason);
+                        }
+                        break;
                     }
-                    break;
             }
         }