Caleb Gardner
61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
package compression
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/binary"
|
|
"io"
|
|
|
|
"github.com/therootcompany/xz"
|
|
|
|
wrtXz "github.com/ulikunitz/xz"
|
|
)
|
|
|
|
type Xz struct {
|
|
DictionarySize int32
|
|
Filters int32
|
|
}
|
|
|
|
//NewXzCompressorWithOptions creates a new Xz compressor/decompressor that reads the compressor options from the given reader.
|
|
func NewXzCompressorWithOptions(rdr io.Reader) (*Xz, error) {
|
|
var x Xz
|
|
err := binary.Read(rdr, binary.LittleEndian, &x)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &x, nil
|
|
}
|
|
|
|
//Decompress decompresses all the data from the rdr and returns the uncompressed bytes.
|
|
func (x *Xz) Decompress(rdr io.Reader) ([]byte, error) {
|
|
r, err := xz.NewReader(rdr, 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
var buf bytes.Buffer
|
|
_, err = io.Copy(&buf, r)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return buf.Bytes(), nil
|
|
}
|
|
|
|
//Compress implements compression.Compress
|
|
func (x *Xz) Compress(data []byte) ([]byte, error) {
|
|
var buf bytes.Buffer
|
|
w, err := wrtXz.NewWriter(&buf)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
w.DictCap = int(x.DictionarySize)
|
|
err = w.Verify()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
_, err = w.Write(data)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
w.Close()
|
|
return buf.Bytes(), nil
|
|
}
|