1

我有一个大的 int 数组，我想保留在文件系统上。我的理解是存储此类内容的最佳方法是使用gob包将其转换为字节数组，然后使用gzip对其进行压缩。当我再次需要它时，我会反转这个过程。我很确定我正确地存储了它，但是用 EOF 恢复它失败了。长话短说，我在下面有一些示例代码来演示这个问题。（游乐场链接在这里https://play.golang.org/p/v4rGGeVkLNh）。我不相信 gob 是必需的，但是阅读它似乎将它存储为字节数组比 int 数组更有效，但这可能不是真的。谢谢！

package main

import (
    "bufio"
    "bytes"
    "compress/gzip"
    "encoding/gob"
    "fmt"
)

func main() {
    arry := []int{1, 2, 3, 4, 5}
    //now gob this
    var indexBuffer bytes.Buffer
    writer := bufio.NewWriter(&indexBuffer)
    encoder := gob.NewEncoder(writer)
    if err := encoder.Encode(arry); err != nil {
        panic(err)
    }
    //now compress it
    var compressionBuffer bytes.Buffer
    compressor := gzip.NewWriter(&compressionBuffer)
    compressor.Write(indexBuffer.Bytes())
    defer compressor.Close()
    //<--- I think all is good until here

    //now decompress it
    buf := bytes.NewBuffer(compressionBuffer.Bytes())
    fmt.Println("byte array before unzipping: ", buf.Bytes())
    if reader, err := gzip.NewReader(buf); err != nil {
        fmt.Println("gzip failed ", err)
        panic(err)
    } else {
        //now ungob it...
        var intArray []int
        decoder := gob.NewDecoder(reader)
        defer reader.Close()
        if err := decoder.Decode(&intArray); err != nil {
            fmt.Println("gob failed ", err)
            panic(err)
        }
        fmt.Println("final int Array content: ", intArray)
    }
}
4

1 回答 1

1

您正在使用bufio.Writerwhich——顾名思义——<em>缓冲写入它的字节。这意味着如果您正在使用它，您必须刷新它以确保缓冲数据进入底层写入器：

writer := bufio.NewWriter(&indexBuffer)
encoder := gob.NewEncoder(writer)
if err := encoder.Encode(arry); err != nil {
    panic(err)
}
if err := writer.Flush(); err != nil {
    panic(err)
}

虽然使用bufio.Writer是完全没有必要的，因为您已经在写入内存缓冲区（bytes.Buffer），所以只需跳过它，直接写入bytes.Buffer（因此您甚至不必刷新）：

var indexBuffer bytes.Buffer
encoder := gob.NewEncoder(&indexBuffer)
if err := encoder.Encode(arry); err != nil {
    panic(err)
}

下一个错误是您如何关闭 gzip 流：

defer compressor.Close()

这种延迟关闭只会在封闭函数（main()函数）返回时发生，而不是早一秒。但是到那时您已经想读取压缩数据，但这可能仍然位于 的内部缓存中gzip.Writer，而不是compressionBuffer，因此您显然无法从compressionBuffer. 关闭 gzip 流而不使用defer：

if err := compressor.Close(); err != nil {
    panic(err)
}

通过这些更改，您的程序运行和输出（在Go Playground上尝试）：

byte array before unzipping:  [31 139 8 0 0 0 0 0 0 255 226 249 223 200 196 200 244 191 137 129 145 133 129 129 243 127 19 3 43 19 11 27 7 23 32 0 0 255 255 110 125 126 12 23 0 0 0]
final int Array content:  [1 2 3 4 5]

作为旁注：buf := bytes.NewBuffer(compressionBuffer.Bytes())- 这buf也是完全没有必要的，您可以开始compressionBuffer自行解码，您可以从中读取先前写入的数据。

您可能已经注意到，压缩数据比初始压缩数据大得多。有几个原因：encoding/gob和compress/gzip流都有很大的开销，并且它们（可能）只会在更大的范围内使输入更小（5 个 int 数字不符合此条件）。

请检查相关问题：Efficient Go serialization of struct to disk

对于小型数组，您还可以考虑使用可变长度编码，请参阅binary.PutVarint()。

于 2019-07-09T10:56:48.907 回答