diff --git a/app_async/src/main.rs b/app_async/src/main.rs index b3e273e..00f4980 100644 --- a/app_async/src/main.rs +++ b/app_async/src/main.rs @@ -160,7 +160,7 @@ async fn read_from_db(id: u32) -> Option { match reader.get(index).await { Ok(question) => Some(question), - Err(_) => None + Err(_) => None, } } async fn write_db() { @@ -168,7 +168,8 @@ async fn write_db() { tokio::try_join!( tokio::spawn(zip_reader_task(tx)), tokio::spawn(db_writer_task(rx)) - ).expect("tokio join"); + ) + .expect("tokio join"); println!("all done"); } async fn db_writer_task(rx: UnboundedReceiver) { diff --git a/lib/src/async_db.rs b/lib/src/async_db.rs index 181244e..e363814 100644 --- a/lib/src/async_db.rs +++ b/lib/src/async_db.rs @@ -59,7 +59,7 @@ where let out = fs::File::create(path).await.str_err()?; let out = io::BufWriter::with_capacity(opts.out_buf_size, out); let data_buf: Vec = Vec::with_capacity(opts.data_buf_size); - + let compress_lvl = opts.compress_lvl; let table: Vec = vec![]; @@ -79,8 +79,10 @@ where let item_data = bincode::encode_to_vec(item, BINCODE_CFG).str_err()?; let mut zencoder = ZstdEncoder::with_quality(&item_data[..], self.compress_lvl); - io::copy(&mut zencoder, &mut self.data_buf).await.str_err()?; - + io::copy(&mut zencoder, &mut self.data_buf) + .await + .str_err()?; + self.table.push(pos); Ok(())