feat: Added is_loaded function to mnn-sync

aftershootco · Nov 20, 2024 · f1c994e · f1c994e
1 parent ecd4120
commit f1c994e
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/flake.nix b/flake.nix
@@ -189,6 +189,7 @@
 
         devShells = {
           default = pkgs.mkShell {
+            LIBCLANG_PATH = "${pkgs.llvmPackages.libclang.lib}/lib";
             packages = with pkgs;
               [
                 mnn

diff --git a/mnn-sync/src/lib.rs b/mnn-sync/src/lib.rs
@@ -46,6 +46,7 @@ pub enum CallbackEnum {
     Callback(Callback),
     Unload(oneshot::Sender<Result<()>>),
     Load(oneshot::Sender<Result<()>>),
+    Status(oneshot::Sender<bool>),
     Close,
 }
 type CallbackSender = CallbackEnum;
@@ -189,6 +190,10 @@ impl SessionState {
     pub fn unload(&mut self) -> Result<()> {
         self.sr.unload()
     }
+
+    pub fn is_loaded(&self) -> bool {
+        self.sr.is_loaded()
+    }
 }
 
 #[non_exhaustive]
@@ -319,6 +324,13 @@ impl SessionHandle {
                             .change_context(ErrorKind::SyncError)
                             .attach_printable("Internal Error: Failed to send load message")?;
                     }
+
+                    CallbackEnum::Status(tx) => {
+                        let res = ss.is_loaded();
+                        tx.send(res)
+                            .change_context(ErrorKind::SyncError)
+                            .attach_printable("Internal Error: Failed to send status message")?;
+                    }
                     CallbackEnum::Close => {
                         break;
                     }
@@ -435,6 +447,17 @@ impl SessionHandle {
             .attach_printable("Internal Error: Failed to recv load message")?
     }
 
+    pub fn is_loaded(&self) -> Result<bool> {
+        let (tx, rx) = oneshot::channel();
+        self.sender
+            .send(CallbackEnum::Status(tx))
+            .map_err(|e| Report::new(ErrorKind::SyncError).attach_printable(e.to_string()))?;
+        Ok(rx
+            .recv()
+            .change_context(ErrorKind::SyncError)
+            .attach_printable("Internal Error: Failed to recv status message")?)
+    }
+
     pub fn close(&self) -> Result<()> {
         self.sender
             .send(CallbackEnum::Close)