It's probably a good idea to reset "restarts" after we get the message, not when we send it; we don't want to jump back into restarting things as normal just because we have scheduled a restart.

1 files changed, 4 insertions, 3 deletions

diff --git a/src/supervisor2.erl b/src/supervisor2.erl
index 90fb0423..4536a72e 100644
--- a/src/supervisor2.erl
+++ b/src/supervisor2.erl
@@ -680,11 +680,12 @@ handle_info({'EXIT', Pid, Reason}, State) ->
 
 handle_info({delayed_restart, {RestartType, Reason, Child}}, State)
   when ?is_simple(State) ->
-    try_restart(RestartType, Reason, Child, State);
+    try_restart(RestartType, Reason, Child, State#state{restarts = []});
 handle_info({delayed_restart, {RestartType, Reason, Child}}, State) ->
     case get_child(Child#child.name, State) of
         {value, Child1} ->
-            try_restart(RestartType, Reason, Child1, State);
+            try_restart(RestartType, Reason, Child1,
+                        State#state{restarts = []});
         _What ->
             {noreply, State}
     end;
@@ -917,7 +918,7 @@ do_restart_delay({RestartType, Delay}, Reason, Child, State) ->
             _TRef = erlang:send_after(trunc(Delay*1000), self(),
                                       {delayed_restart,
                                        {{RestartType, Delay}, Reason, Child}}),
-            {ok, state_del_child(Child, State#state{restarts = []})}
+            {ok, state_del_child(Child, State)}
     end.
 
 restart(Child, State) ->