fix async_db tests

using ZstdEncoder from ..:tokio::bufread instead of ..:tokio::write
This commit is contained in:
Dmitry Belyaev 2023-08-08 16:54:40 +03:00
parent daf41625ff
commit 4c555df8ca
Signed by: b4tman
GPG Key ID: 41A00BF15EA7E5F3

View File

@ -1,7 +1,7 @@
use std::{path::Path, sync::Arc}; use std::{path::Path, sync::Arc};
use async_compression::tokio::bufread::ZstdDecoder; use async_compression::tokio::bufread::ZstdDecoder;
use async_compression::tokio::write::ZstdEncoder; use async_compression::tokio::bufread::ZstdEncoder;
use async_compression::Level; use async_compression::Level;
use futures::stream::StreamExt; use futures::stream::StreamExt;
use futures_core::stream::Stream; use futures_core::stream::Stream;
@ -78,9 +78,8 @@ where
let item_data = bincode::encode_to_vec(item, BINCODE_CFG).str_err()?; let item_data = bincode::encode_to_vec(item, BINCODE_CFG).str_err()?;
let mut zencoder = ZstdEncoder::with_quality(&mut self.data_buf, self.compress_lvl); let mut zencoder = ZstdEncoder::with_quality(&item_data[..], self.compress_lvl);
zencoder.write_all(&item_data).await.str_err()?; io::copy(&mut zencoder, &mut self.data_buf).await.str_err()?;
zencoder.flush().await.str_err()?;
self.table.push(pos); self.table.push(pos);
@ -116,7 +115,6 @@ where
let pos_data = (pos + tab_size).to_le_bytes(); let pos_data = (pos + tab_size).to_le_bytes();
self.out.write_all(&pos_data).await.str_err()?; self.out.write_all(&pos_data).await.str_err()?;
} }
// copy data // copy data
self.out.write_all(&self.data_buf[..]).await.str_err()?; self.out.write_all(&self.data_buf[..]).await.str_err()?;