refactor: use async for "obnam get-chunk"

Sponsored-by: author

1 files changed, 9 insertions, 4 deletions

diff --git a/src/cmd/get_chunk.rs b/src/cmd/get_chunk.rs
index 4ee70fe..f574c99 100644
--- a/src/cmd/get_chunk.rs
+++ b/src/cmd/get_chunk.rs
@@ -1,9 +1,10 @@
 use crate::chunkid::ChunkId;
-use crate::client::BackupClient;
+use crate::client::AsyncBackupClient;
 use crate::config::ClientConfig;
 use crate::error::ObnamError;
 use std::io::{stdout, Write};
 use structopt::StructOpt;
+use tokio::runtime::Runtime;
 
 #[derive(Debug, StructOpt)]
 pub struct GetChunk {
@@ -13,10 +14,14 @@ pub struct GetChunk {
 
 impl GetChunk {
     pub fn run(&self, config: &ClientConfig) -> Result<(), ObnamError> {
-        let client = BackupClient::new(config)?;
-        let chunk_id: ChunkId = self.chunk_id.parse().unwrap();
-        let chunk = client.fetch_chunk(&chunk_id)?;
+        let rt = Runtime::new()?;
+        rt.block_on(self.run_async(config))
+    }
 
+    async fn run_async(&self, config: &ClientConfig) -> Result<(), ObnamError> {
+        let client = AsyncBackupClient::new(config)?;
+        let chunk_id: ChunkId = self.chunk_id.parse().unwrap();
+        let chunk = client.fetch_chunk(&chunk_id).await?;
         let stdout = stdout();
         let mut handle = stdout.lock();
         handle.write_all(chunk.data())?;