chore: handle close when runtime is dropped (#127)

.github/workflows/ci.yml

@@ -18,7 +18,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        target: [ 'x86_64-unknown-linux-gnu', 'aarch64-unknown-linux-gnu', 'x86_64-pc-windows-msvc', 'x86_64-apple-darwin', 'aarch64-apple-darwin' ]
+        target: [ 'x86_64-unknown-linux-gnu', 'aarch64-unknown-linux-gnu', 'x86_64-pc-windows-msvc', 'x86_64-apple-darwin', 'aarch64-apple-darwin', 'wasm32-unknown-unknown' ]
     steps:
       - uses: actions/checkout@v2
       - name: Cache

Cargo.toml

@@ -28,6 +28,8 @@ tokio = { version = "1.36", features = ["rt", "sync"] }
 quick_cache = "0.6.0"
 vart = "0.7.0"
 revision = "0.10.0"
+[target.'cfg(target_arch = "wasm32")'.dependencies]
+getrandom = { version = "0.2.15", features = ["js"] }
 
 [dev-dependencies]
 tokio = { version = "1", features = ["full"] }

src/storage/kv/store.rs

@@ -181,13 +181,18 @@ impl Store {
 impl Drop for Store {
     fn drop(&mut self) {
         if let Some(inner) = self.inner.take() {
-            // Close the store asynchronously
-            tokio::spawn(async move {
-                if let Err(err) = inner.close().await {
-                    // TODO: use log/tracing instead of eprintln
-                    eprintln!("Error occurred while closing the kv store: {}", err);
-                }
-            });
+            // Try to get existing runtime handle first
+            if let Ok(handle) = tokio::runtime::Handle::try_current() {
+                // We're in a runtime, spawn normally
+                handle.spawn(async move {
+                    if let Err(err) = inner.close().await {
+                        // TODO: use log/tracing instead of eprintln
+                        eprintln!("Error closing store: {}", err);
+                    }
+                });
+            } else {
+                eprintln!("No runtime available for closing the store correctly");
+            }
         }
     }
 }