Apply suggestions from code review

Co-authored-by: Nick Rolfe <nickrolfe@github.com>
This commit is contained in:
Arthur Baars
2021-07-29 09:02:57 +02:00
committed by GitHub
parent dacd3f3d19
commit fcf2d4cbd2
3 changed files with 17 additions and 5 deletions

View File

@@ -166,15 +166,15 @@ fn main() -> std::io::Result<()> {
let mut trap_file = BufWriter::new(trap_file);
match trap_compression {
TrapCompression::None => {
if erb_trap.is_some() {
write!(trap_file, "{}", erb_trap.unwrap())?;
if let Some(erb_trap) = erb_trap {
write!(trap_file, "{}", erb_trap)?;
}
write!(trap_file, "{}", trap)
}
TrapCompression::Gzip => {
let mut compressed_writer = GzEncoder::new(trap_file, flate2::Compression::fast());
if erb_trap.is_some() {
write!(compressed_writer, "{}", erb_trap.unwrap())?;
if let Some(erb_trap) = erb_trap {
write!(compressed_writer, "{}", erb_trap)?;
}
write!(compressed_writer, "{}", trap)
}