Apply zstd compression

crumblingstatue · Oct 2, 2021 · 3be0dde · 3be0dde
1 parent d7e61aa
commit 3be0dde
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -20,6 +20,7 @@ fnv = "1.0.7"
 directories = "4.0.1"
 rfd = "0.5.0"
 open = "2.0.1"
+zstd = "0.9.0"
 
 [dependencies.sfml]
 git = "https://github.com/jeremyletang/rust-sfml.git"

diff --git a/src/serialization.rs b/src/serialization.rs
@@ -15,7 +15,7 @@ use serde::{Deserialize, Serialize};
 pub fn read<T: for<'de> Deserialize<'de>>(mut source: impl Read) -> anyhow::Result<T> {
     let mut ver = [0];
     source.read_exact(&mut ver)?;
-    Ok(from_read(source)?)
+    Ok(from_read(zstd::Decoder::new(source)?)?)
 }
 
 pub fn read_from_file<T: for<'de> Deserialize<'de>>(path: impl AsRef<Path>) -> anyhow::Result<T> {
@@ -28,5 +28,8 @@ pub fn write_to_file<T: Serialize>(obj: &T, path: impl AsRef<Path>) -> anyhow::R
 
 pub fn write<T: Serialize>(obj: &T, mut sink: impl Write) -> anyhow::Result<()> {
     sink.write_all(&[DUMMY_VERSION])?;
-    Ok(write_named(&mut sink, obj)?)
+    let mut zstd = zstd::Encoder::new(sink, 0)?;
+    write_named(&mut zstd, obj)?;
+    zstd.finish()?;
+    Ok(())
 }