diff --git a/src/cli.rs b/src/cli.rs index 44317dd56..24dbd7429 100644 --- a/src/cli.rs +++ b/src/cli.rs @@ -260,7 +260,7 @@ pub struct Options { #[arg( long = "compression-algo", env = "P_PARQUET_COMPRESSION_ALGO", - default_value = "lz4", + default_value = "lz4_raw", value_parser = validation::compression, help = "Parquet compression algorithm" )] diff --git a/src/correlation.rs b/src/correlation.rs index 7ffde6a8a..22974ee95 100644 --- a/src/correlation.rs +++ b/src/correlation.rs @@ -129,10 +129,9 @@ impl Correlations { .await?; // Update in memory - self.write().await.insert( - correlation.id.to_owned(), - correlation.clone(), - ); + self.write() + .await + .insert(correlation.id.to_owned(), correlation.clone()); Ok(correlation) } diff --git a/src/option.rs b/src/option.rs index 86e198713..c6f185397 100644 --- a/src/option.rs +++ b/src/option.rs @@ -168,8 +168,9 @@ pub enum Compression { Gzip, Lzo, Brotli, - #[default] Lz4, + #[default] + Lz4Raw, Zstd, } @@ -182,6 +183,7 @@ impl From for parquet::basic::Compression { Compression::Lzo => parquet::basic::Compression::LZO, Compression::Brotli => parquet::basic::Compression::BROTLI(BrotliLevel::default()), Compression::Lz4 => parquet::basic::Compression::LZ4, + Compression::Lz4Raw => parquet::basic::Compression::LZ4_RAW, Compression::Zstd => parquet::basic::Compression::ZSTD(ZstdLevel::default()), } } @@ -277,6 +279,7 @@ pub mod validation { "lzo" => Ok(Compression::Lzo), "brotli" => Ok(Compression::Brotli), "lz4" => Ok(Compression::Lz4), + "lz4_raw" => Ok(Compression::Lz4Raw), "zstd" => Ok(Compression::Zstd), _ => Err("Invalid COMPRESSION provided".to_string()), }