Send Idle message to actors when they are idle. (#502)

neo-project · Dec 5, 2018 · 7c8bf1f · 7c8bf1f
1 parent 56c1ee9
commit 7c8bf1f
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/neo/IO/Actors/Idle.cs b/neo/IO/Actors/Idle.cs
@@ -0,0 +1,7 @@
+namespace Neo.IO.Actors
+{
+    internal sealed class Idle
+    {
+        public static Idle Instance { get; } = new Idle();
+    }
+}
diff --git a/neo/IO/Actors/PriorityMessageQueue.cs b/neo/IO/Actors/PriorityMessageQueue.cs
@@ -38,8 +38,12 @@ public void Enqueue(IActorRef receiver, Envelope envelope)
 
         public bool TryDequeue(out Envelope envelope)
         {
-            if (high.TryDequeue(out envelope)) return true;
-            if (low.TryDequeue(out envelope)) return true;
+            if (high.TryDequeue(out envelope) || low.TryDequeue(out envelope))
+            {
+                if (Count == 0 && !(envelope.Message is Idle))
+                    low.Enqueue(new Envelope(Idle.Instance, ActorRefs.NoSender));
+                return true;
+            }
             return false;
         }
     }