Merge pull request #28 from payload/fix/supervisor-panic

Don't call rx.next after receiving None, fixes #27
sunli829 · Sep 28, 2020 · f36533d · f36533d
2 parents 8a7e089 + d83a870
commit f36533d
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/src/supervisor.rs b/src/supervisor.rs
@@ -90,12 +90,13 @@ impl Supervisor {
 
         spawn({
             async move {
-                loop {
-                    while let Some(event) = rx.next().await {
-                        match event {
-                            ActorEvent::Exec(f) => f(&mut actor, &mut ctx).await,
-                            ActorEvent::Stop(_err) => break,
-                            ActorEvent::RemoveStream(id) => {
+                'restart_loop: loop {
+                    'event_loop: loop {
+                        match rx.next().await {
+                            None => break 'restart_loop,
+                            Some(ActorEvent::Stop(_err)) => break 'event_loop,
+                            Some(ActorEvent::Exec(f)) => f(&mut actor, &mut ctx).await,
+                            Some(ActorEvent::RemoveStream(id)) => {
                                 if ctx.streams.contains(id) {
                                     ctx.streams.remove(id);
                                 }
@@ -104,10 +105,11 @@ impl Supervisor {
                     }
 
                     actor.stopped(&mut ctx).await;
-
                     actor = f();
                     actor.started(&mut ctx).await.ok();
                 }
+
+                actor.stopped(&mut ctx).await;
             }
         });