Use a multi-threaded xz encoder

2 files changed, 6 insertions, 1 deletions

diff --git a/Cargo.toml b/Cargo.toml
index 33e0091..d62cc58 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -16,6 +16,7 @@ rayon = "1.0"
 tar = "0.4.13"
 walkdir = "2"
 xz2 = "0.1.4"
+num_cpus = "1"
 
 [dependencies.clap]
 features = ["yaml"]
diff --git a/src/tarballer.rs b/src/tarballer.rs
index badac9c..9ff1bd2 100644
--- a/src/tarballer.rs
+++ b/src/tarballer.rs
@@ -48,7 +48,11 @@ impl Tarballer {
         let gz = GzEncoder::new(create_new_file(tar_gz)?, flate2::Compression::best());
 
         // Prepare the `.tar.xz` file.
-        let xz = XzEncoder::new(create_new_file(tar_xz)?, 6);
+        let stream = xz2::stream::MtStreamBuilder::new()
+            .threads(num_cpus::get() as u32)
+            .preset(6)
+            .encoder()?;
+        let xz = XzEncoder::new_stream(create_new_file(tar_xz)?, stream);
 
         // Write the tar into both encoded files. We write all directories
         // first, so files may be directly created. (See rust-lang/rustup.rs#1092.)