将字节切片压缩到Golang中的另一个切片中

izkcnapc  于 2023-02-06  发布在  Go
关注(0)|答案(1)|浏览(165)

我想得到和这里给出的完全相反的解,把一个字节片压缩成另一个字节片-
将压缩的[]字节转换为解压缩的[]字节golang代码
像是-

func ZipBytes(unippedBytes []byte) ([]byte, error) {
// ...
}

[我将把压缩文件作为POST请求的多部分表单数据上传]

quhf5bfb

quhf5bfb1#

您可以使用bytes.Buffer直接压缩到内存中。
以下示例使用compress/zlib,因为它与问题中给出的示例相反。根据您的用例,您也可以轻松地将其更改为compress/gzip(非常相似的API)。

package data_test

import (
    "bytes"
    "compress/zlib"
    "io"
    "testing"
)

func compress(buf []byte) ([]byte, error) {
    var out bytes.Buffer
    w := zlib.NewWriter(&out)
    if _, err := w.Write(buf); err != nil {
        return nil, err
    }
    if err := w.Close(); err != nil {
        return nil, err
    }
    return out.Bytes(), nil
}

func decompress(buf []byte) (_ []byte, e error) {
    r, err := zlib.NewReader(bytes.NewReader(buf))
    if err != nil {
        return nil, err
    }
    defer func() {
        if err := r.Close(); e == nil {
            e = err
        }
    }()
    return io.ReadAll(r)
}

func TestRoundtrip(t *testing.T) {
    want := []byte("test data")

    zdata, err := compress(want)
    if err != nil {
        t.Fatalf("compress: %v", err)
    }
    got, err := decompress(zdata)
    if err != nil {
        t.Fatalf("decompress: %v", err)
    }
    if !bytes.Equal(want, got) {
        t.Errorf("roundtrip: got = %q; want = %q", got, want)
    }
}

相关问题