LZ compression in Go

compress.go

package compress

import (
	"bytes"
	"compress/lzw"
	"io/ioutil"
)

func compress(input []byte) ([]byte, error) {
	output := &bytes.Buffer{}
	compressor := lzw.NewWriter(output, lzw.LSB, 8)

	defer compressor.Close()

	_, err := compressor.Write(input)
	if err != nil {
		return []byte{}, err
	}

	compressor.Close()

	return output.Bytes(), nil
}

func decompress(input []byte) ([]byte, error) {
	inputBuf := bytes.NewReader(input)
	decompressor := lzw.NewReader(inputBuf, lzw.LSB, 8)

	defer decompressor.Close()

	output, err := ioutil.ReadAll(decompressor)
	if err != nil {
		return output, err
	}

	return output, nil
}