首页  >  文章  >  后端开发  >  在 Golang 中将一个字节切片压缩到另一个切片中

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

王林
王林转载
2024-02-12 16:57:051093浏览

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

问题内容

我想实现与此处给出的解决方案完全相反的效果,将一个字节片段压缩到另一个字节片段中 -

将压缩的 []byte 转换为解压的 []byte golang 代码

类似 -

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

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

解决方法

您可以使用 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)
    }
}

以上是在 Golang 中将一个字节切片压缩到另一个切片中的详细内容。更多信息请关注PHP中文网其他相关文章!

声明:
本文转载于:stackoverflow.com。如有侵权,请联系admin@php.cn删除