diff options
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/src/main.rs b/src/main.rs index 6bec85b..7ff8725 100644 --- a/src/main.rs +++ b/src/main.rs @@ -74,13 +74,6 @@ fn print_completions<G: Generator>(generator: G, cmd: &mut Command) { } fn main() -> Result<()> { - let running = Arc::new(AtomicBool::new(true)); - let r = running.clone(); - - ctrlc::set_handler(move || { - r.store(false, Ordering::SeqCst); - })?; - let args = build_cli().get_matches(); if let Some(generator) = args.get_one::<Shell>("shell").copied() { @@ -90,6 +83,13 @@ fn main() -> Result<()> { return Ok(()); } + let running = Arc::new(AtomicBool::new(true)); + let r = running.clone(); + + ctrlc::set_handler(move || { + r.store(false, Ordering::SeqCst); + })?; + let conn = if let Some(path) = args.get_one::<PathBuf>("db") { smol::block_on(async { db::Database::new(path).await })? } else { |
