refactor(runtime): return status instead (#15)

noctisynth · Sep 26, 2024 · d8f688a · d8f688a
1 parent 80fe3be
commit d8f688a
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 6 deletions.
diff --git a/crates/aionbot-adapter-onebot/src/lib.rs b/crates/aionbot-adapter-onebot/src/lib.rs
@@ -1,6 +1,6 @@
 use std::sync::Arc;
 
-use aionbot_core::runtime::{Runtime, StateManager};
+use aionbot_core::runtime::{Runtime, RuntimeStatus, StateManager};
 use anyhow::Result;
 use onebot_v11::connect::ws_reverse::ReverseWsConnect;
 
@@ -55,7 +55,7 @@ impl Runtime for OnebotRuntime {
         Ok(())
     }
 
-    async fn run(&self) -> Result<()> {
-        Ok(())
+    async fn run(&self) -> Result<RuntimeStatus> {
+        Ok(RuntimeStatus::Exit)
     }
 }
diff --git a/crates/aionbot-core/src/runtime.rs b/crates/aionbot-core/src/runtime.rs
@@ -51,13 +51,29 @@ where
         self
     }
 
-    pub async fn run(&mut self) -> Result<()> {
+    async fn prepare(&mut self) -> Result<()> {
         self.runtime.prepare().await?;
         if let Some(setup) = self.setup.take() {
             self.runtime.setup(setup);
         }
         self.runtime.finalize().await?;
-        self.runtime.run().await
+        Ok(())
+    }
+
+    pub async fn run(&mut self) -> Result<()> {
+        self.prepare().await?;
+
+        loop {
+            match self.runtime.run().await? {
+                RuntimeStatus::Pending => {}
+                RuntimeStatus::Exit => break,
+                RuntimeStatus::Next => {}
+                RuntimeStatus::Restart => {
+                    self.runtime.prepare().await?;
+                }
+            }
+        }
+        Ok(())
     }
 }
 
@@ -95,5 +111,12 @@ pub trait Runtime {
         async move { Ok(()) }
     }
 
-    fn run(&self) -> impl std::future::Future<Output = Result<()>> + Send;
+    fn run(&self) -> impl std::future::Future<Output = Result<RuntimeStatus>> + Send;
+}
+
+pub enum RuntimeStatus {
+    Pending,
+    Exit,
+    Next,
+    Restart,
 }