Fixing issues found during publishing

These were found in v0.4.0, but apparently I forgot to apply them to main.
khonsulabs · Apr 5, 2022 · d677d07 · d677d07
1 parent 500b0a0
commit d677d07
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/crates/bonsaidb-client/Cargo.toml b/crates/bonsaidb-client/Cargo.toml
@@ -58,7 +58,7 @@ wasm-bindgen = "0.2"
 
 [target.'cfg(not(target_arch = "wasm32"))'.dependencies]
 fabruic = { version = "0.0.1-dev.5" }
-tokio = { version = "1.16.1", features = ["sync"] }
+tokio = { version = "1.16.1", features = ["macros", "sync"] }
 tokio-tungstenite = { version = "0.16", optional = true, features = [
     "rustls-tls-native-roots",
 ] }

diff --git a/crates/bonsaidb-keystorage-s3/src/lib.rs b/crates/bonsaidb-keystorage-s3/src/lib.rs
@@ -59,7 +59,7 @@ use bonsaidb_local::{
     StorageId,
 };
 pub use http;
-use tokio::runtime::{Handle, Runtime};
+use tokio::runtime::{self, Handle, Runtime};
 
 /// S3-compatible [`VaultKeyStorage`] implementor.
 #[derive(Default, Debug)]
@@ -86,7 +86,17 @@ enum Tokio {
 
 impl Default for Tokio {
     fn default() -> Self {
-        Handle::try_current().map_or_else(|_| Self::Runtime(Runtime::new().unwrap()), Self::Handle)
+        Handle::try_current().map_or_else(
+            |_| {
+                Self::Runtime(
+                    runtime::Builder::new_current_thread()
+                        .enable_all()
+                        .build()
+                        .unwrap(),
+                )
+            },
+            Self::Handle,
+        )
     }
 }