Send Idle message to actors when they are idle.

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

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
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
@@ -39,7 +39,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 (low.TryDequeue(out envelope))
+            {
+                if (low.Count == 0 && !(envelope.Message is Idle))
+                    low.Enqueue(new Envelope(Idle.Instance, ActorRefs.NoSender));
+                return true;
+            }
             return false;
         }
     }