diff options
| -rw-r--r-- | atuin/src/command/client.rs | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/atuin/src/command/client.rs b/atuin/src/command/client.rs index 312bcafa..d0f58c4a 100644 --- a/atuin/src/command/client.rs +++ b/atuin/src/command/client.rs @@ -52,8 +52,20 @@ pub enum Cmd { } impl Cmd { - #[tokio::main(flavor = "current_thread")] - pub async fn run(self) -> Result<()> { + pub fn run(self) -> Result<()> { + let runtime = tokio::runtime::Builder::new_current_thread() + .enable_all() + .build() + .unwrap(); + + let res = runtime.block_on(self.run_inner()); + + runtime.shutdown_timeout(std::time::Duration::from_millis(50)); + + res + } + + async fn run_inner(self) -> Result<()> { Builder::new() .filter_level(log::LevelFilter::Off) .parse_env("ATUIN_LOG") |