clean up verify vms

nervosnetwork · Jan 16, 2024 · 6161623 · 6161623
1 parent 1924783
commit 6161623
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/script/src/verify.rs b/script/src/verify.rs
@@ -1382,7 +1382,7 @@ fn run_vms(
     }
 }
 
-// Run a series of VMs that are just freshly resumed
+// Run a series of VMs with control signal, will only return when verification finished
 async fn run_vms_with_signal(
     script_group: &ScriptGroup,
     max_cycles: Cycle,
@@ -1409,11 +1409,13 @@ async fn run_vms_with_signal(
     loop {
         tokio::select! {
             _ = signal.changed() => {
-                let state = signal.borrow().to_owned();
-                if state == ChunkCommand::Suspend {
-                    pause.interrupt();
-                } else if state == ChunkCommand::Resume {
-                    child_sender.send(ChunkCommand::Resume).unwrap();
+                match signal.borrow().to_owned() {
+                    ChunkCommand::Suspend => {
+                        pause.interrupt();
+                    }
+                    ChunkCommand::Resume => {
+                        child_sender.send(ChunkCommand::Resume).unwrap();
+                    }
                 }
             }
             Some(res) = finished_recv.recv() => {