Merge pull request asynkron#62 from cpx86/supervisor-methods

Added supervision methods to ISupervisor

Author: rogeralsing

src/Proto.Actor/LocalContext.cs

@@ -267,6 +267,30 @@ public void EscalateFailure(PID who, Exception reason)
             Parent.SendSystemMessage(new Failure(who, reason, _restartStatistics));
         }
 
+        public void RestartChildren(params PID[] pids)
+        {
+            for (int i = 0; i < pids.Length; i++)
+            {
+                pids[i].SendSystemMessage(Restart.Instance);
+            }
+        }
+
+        public void StopChildren(params PID[] pids)
+        {
+            for (int i = 0; i < pids.Length; i++)
+            {
+                pids[i].SendSystemMessage(Stop.Instance);
+            }
+        }
+
+        public void ResumeChildren(params PID[] pids)
+        {
+            for (int i = 0; i < pids.Length; i++)
+            {
+                pids[i].SendSystemMessage(ResumeMailbox.Instance);
+            }
+        }
+
         internal static Task DefaultReceive(IContext context)
         {
             var c = (Context)context;

src/Proto.Actor/Supervision.cs

@@ -22,12 +22,15 @@ public interface ISupervisor
     {
         IReadOnlyCollection<PID> Children { get; }
         void EscalateFailure(PID who, Exception reason);
+        void RestartChildren(params PID[] pids);
+        void StopChildren(params PID[] pids);
+        void ResumeChildren(params PID[] pids);
     }
 
     public static class Supervision
     {
         public static ISupervisorStrategy DefaultStrategy { get; } =
-            new OneForOneStrategy((who, reason) => SupervisorDirective.Restart,10, TimeSpan.FromSeconds(10));
+            new OneForOneStrategy((who, reason) => SupervisorDirective.Restart, 10, TimeSpan.FromSeconds(10));
     }
 
     public interface ISupervisorStrategy
@@ -83,25 +86,25 @@ public void HandleFailure(ISupervisor supervisor, PID child, RestartStatistics c
             {
                 case SupervisorDirective.Resume:
                     //resume the failing child
-                    child.SendSystemMessage(ResumeMailbox.Instance);
+                    supervisor.ResumeChildren(child);
                     break;
                 case SupervisorDirective.Restart:
                     //restart the failing child
                     if (crs.RequestRestartPermission(_maxNrOfRetries, _withinTimeSpan))
                     {
                         Console.WriteLine($"Restarting {child.ToShortString()} Reason {reason}");
-                        child.SendSystemMessage(Restart.Instance);
+                        supervisor.RestartChildren(child);
                     }
                     else
                     {
                         Console.WriteLine($"Stopping {child.ToShortString()} Reason { reason}");
-                        child.Stop();
+                        supervisor.StopChildren(child);
                     }
                     break;
                 case SupervisorDirective.Stop:
                     //stop the failing child
                     Console.WriteLine($"Stopping {child.ToShortString()} Reason {reason}");
-                    child.Stop();
+                    supervisor.StopChildren(child);
                     break;
                 case SupervisorDirective.Escalate:
                     supervisor.EscalateFailure(child, reason);