diff options
Diffstat (limited to 'tool/src/main.rs')
| -rw-r--r-- | tool/src/main.rs | 12 |
1 files changed, 7 insertions, 5 deletions
diff --git a/tool/src/main.rs b/tool/src/main.rs index adf0a35..6aeb617 100644 --- a/tool/src/main.rs +++ b/tool/src/main.rs @@ -19,11 +19,13 @@ fn main() -> anyhow::Result<()> { let args = Args::parse(); match args { - a @ Action::Add { .. } => tokio::runtime::Builder::new_multi_thread() - .enable_all() - .build() - .unwrap() - .block_on(add(a)), + a @ Action::Add { .. } => { + let rt = tokio::runtime::Builder::new_multi_thread() + .enable_all() + .build() + .unwrap(); + add(a, rt.handle()) + } a @ Action::Migrate { .. } => migrate(a), _ => Ok(()), // Action::Reimport { |