address comment

sunchao · May 1, 2018 · 26136e9 · 26136e9
1 parent df990ca
commit 26136e9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/compression.rs b/src/compression.rs
@@ -230,6 +230,9 @@ impl ZSTDCodec {
   }
 }
 
+/// Compression level (1-21) for ZSTD. Choose 1 here for better compression speed.
+const ZSTD_COMPRESSION_LEVEL: i32 = 1;
+
 impl Codec for ZSTDCodec {
   fn decompress(&mut self, input_buf: &[u8], output_buf: &mut Vec<u8>) -> Result<usize> {
     let mut decoder = zstd::Decoder::new(input_buf)?;
@@ -241,7 +244,7 @@ impl Codec for ZSTDCodec {
 
   fn compress(&mut self, input_buf: &[u8]) -> Result<Vec<u8>> {
     let output = Vec::new();
-    let mut encoder = zstd::Encoder::new(output, 1)?;
+    let mut encoder = zstd::Encoder::new(output, ZSTD_COMPRESSION_LEVEL)?;
     encoder.write_all(&input_buf[..])?;
     encoder.finish()
       .map_err(|e| general_err!("Error when compressing using ZSTD: {}", e))