zstd-rs

zstd

This library is a rust binding for the zstd compression library.

Documentation

1 - Add to cargo.toml

$ cargo add zstd

# Cargo.toml

[dependencies]
zstd = "0.14"

2 - Usage

This library provides Read and Write wrappers to handle (de)compression, along with convenience functions to made common tasks easier.

For instance, stream::copy_encode and stream::copy_decode are easy-to-use wrappers around std::io::copy. Check the stream example:

use std::io;

// This function use the convenient `copy_encode` method
fn compress(level: i32) {
    zstd::stream::copy_encode(io::stdin(), io::stdout(), level).unwrap();
}

// This function does the same thing, directly using an `Encoder`:
fn compress_manually(level: i32) {
    let mut encoder = zstd::stream::Encoder::new(io::stdout(), level).unwrap();
    io::copy(&mut io::stdin(), &mut encoder).unwrap();
    encoder.finish().unwrap();
}

fn decompress() {
    zstd::stream::copy_decode(io::stdin(), io::stdout()).unwrap();
}

Asynchronous support

The async-compression crate provides an async-ready integration of various compression algorithms, including zstd-rs.

Performance

Which function to use

The frame header can record how big the data will be once decompressed, and whoever reads it can then size their buffer in one go instead of growing it - worth roughly a factor of two to them. Only a compressor that knows the length up front can record it:

Everything but the last row records the size. Reach for the Write column when the data is large enough that you would rather not hold a Vec of it - these stream straight through, so only the compressor's own buffers are in memory. For a source that is neither a file nor a slice, its length is the one thing you have to supply yourself.

On the way back, decode_all reads that header and sizes its output for you, and decompressed_size gives you the same number for a buffer of your own.

Matching the C library

This wrapper adds little over the C library. On the same data at level 3, a re-used bulk::Compressor and bulk::Decompressor land within a few percent of what zstd -b3 reports for the same file. If you are seeing much less than that, it is usually one of these:

• A debug build. Always measure with --release; a debug build times this crate's wrappers rather than zstd.
• Growing a Vec rather than sizing it. The convenience functions that return a Vec start from an empty one. On decompression, sizing the output up front - bulk::decompress, or copy_decode into a Vec::with_capacity - was worth roughly a factor of two in our measurements.
• Streaming data that is already in memory. For a slice you already hold, the one-shot bulk API avoids a copy through the streaming buffers. It is worth about ten percent on decompression, so it is a much smaller effect than sizing the output - the Read/Write wrappers are not the problem people usually assume they are.
• Leaving compression single-threaded. With the zstdmt feature and several workers, compression of a 33 MB input went about three times faster here. It is a loss on small inputs, where there is not enough data to keep the workers busy.

To measure all of these on your own data:

cargo run --release --features zstdmt --example throughput -- 

It prints the matching zstd -b command so you can compare against the C library directly.

Cargo features

Enabled by default: legacy, arrays, zdict_builder.

The following change how the C library is obtained or built:

The zstd-sys readme also covers the build-time environment variables, and what to do if the linker complains about a hidden zstd symbol being "referenced by DSO".

Compile it yourself

zstd is included as a submodule. To get everything during your clone, use:

git clone https://github.com/gyscos/zstd-rs --recursive

Or, if you cloned it without the --recursive flag, call this from inside the repository:

git submodule update --init

Then, running cargo build should take care of building the C library and linking to it.

Build-time bindgen

This library includes a pre-generated bindings.rs file. You can also generate new bindings at build-time, using the bindgen feature:

cargo build --features bindgen

TODO

• Benchmarks, optimizations, ...

Disclaimer

This implementation is largely inspired by bozaro's lz4-rs.

License

• The zstd C library is under a dual BSD/GPLv2 license.
• This zstd-rs binding library is under a BSD-3-Clause license.