Optimize TaskManager (#1294)

neo-project · Nov 28, 2019 · 4a8194e · 4a8194e
1 parent 92eb258
commit 4a8194e
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/neo/Network/P2P/TaskManager.cs b/src/neo/Network/P2P/TaskManager.cs
@@ -140,27 +140,27 @@ private void OnTaskCompleted(UInt256 hash)
         [MethodImpl(MethodImplOptions.AggressiveInlining)]
         private void DecrementGlobalTask(UInt256 hash)
         {
-            if (globalTasks.ContainsKey(hash))
+            if (globalTasks.TryGetValue(hash, out var value))
             {
-                if (globalTasks[hash] == 1)
+                if (value == 1)
                     globalTasks.Remove(hash);
                 else
-                    globalTasks[hash]--;
+                    globalTasks[hash] = value - 1;
             }
         }
 
         [MethodImpl(MethodImplOptions.AggressiveInlining)]
         private bool IncrementGlobalTask(UInt256 hash)
         {
-            if (!globalTasks.ContainsKey(hash))
+            if (!globalTasks.TryGetValue(hash, out var value))
             {
                 globalTasks[hash] = 1;
                 return true;
             }
-            if (globalTasks[hash] >= MaxConncurrentTasks)
+            if (value >= MaxConncurrentTasks)
                 return false;
 
-            globalTasks[hash]++;
+            globalTasks[hash] = value + 1;
 
             return true;
         }