无法使用gob正确地将数据存储在文件中

时间:2016-07-05 13:16:36

标签: file dictionary go gob

当我尝试使用map[mapKey]string编码器将gob类型的地图保存到文件中时,它不会将字符串保存在文件中。

这里mapKey是struct,map值是long json string。

type mapKey struct{
    Id1 string
    Id2 string
}

每当我使用嵌套地图而不是结构如:

var m = make(map[string]map[string]string)

工作正常并正确保存字符串。我不确定我在这里缺少什么。

编码,解码并将其保存在文件中的代码:

func Save(path string, object interface{}) error {
    file, err := os.Create(path)
    if err == nil {
        encoder := gob.NewEncoder(file)
        encoder.Encode(object)
    }
    file.Close()
    return err
}

// Decode Gob file
func Load(path string, object interface{}) error {
    file, err := os.Open(path)
    if err == nil {
        decoder := gob.NewDecoder(file)
        err = decoder.Decode(object)
    }
    file.Close()
    return err
}

func Check(e error) {
    if e != nil {
        _, file, line, _ := runtime.Caller(1)
        fmt.Println(line, "\t", file, "\n", e)
        os.Exit(1)
    }
}

1 个答案:

答案 0 :(得分:4)

map[mapKey]string类型的值进行编码没有什么特别之处。

请参阅这个使用指定读者/编写器的非常简单的工作示例:

func save(w io.Writer, i interface{}) error {
    return gob.NewEncoder(w).Encode(i)

}
func load(r io.Reader, i interface{}) error {
    return gob.NewDecoder(r).Decode(i)
}

使用内存缓冲区(bytes.Buffer)测试它:

m := map[mapKey]string{
    {"1", "2"}: "12",
    {"3", "4"}: "34",
}
fmt.Println(m)

buf := &bytes.Buffer{}
if err := save(buf, m); err != nil {
    panic(err)
}

var m2 map[mapKey]string
if err := load(buf, &m2); err != nil {
    panic(err)
}
fmt.Println(m2)

按预期输出(在Go Playground上试试):

map[{1 2}:12 {3 4}:34]
map[{1 2}:12 {3 4}:34]

您有一个正常工作的代码,但知道您必须使用指针值调用Load()(否则Decoder.Decode()将无法修改其值)。

还有一些改进方法:

  • 在您的示例中,您正在吞噬Encoder.Encode()返回的error(检查它,您将看到问题所在;常见问题是使用结构mapKey而没有导出字段,在这种情况下将返回gob: type main.mapKey has no exported fields的错误。)
  • 此外,您应该将File.Close()称为延期功能。
  • 此外,如果打开文件失败,您应该提前退回,不应该关闭该文件。

这是您的代码的更正版本:

func Save(path string, object interface{}) error {
    file, err := os.Create(path)
    if err != nil {
        return err
    }
    defer file.Close()
    return gob.NewEncoder(file).Encode(object)
}

func Load(path string, object interface{}) error {
    file, err := os.Open(path)
    if err != nil {
        return err
    }
    defer file.Close()
    return gob.NewDecoder(file).Decode(object)
}

测试它:

m := map[mapKey]string{
    {"1", "2"}: "12",
    {"3", "4"}: "34",
}
fmt.Println(m)

if err := Save("testfile", m); err != nil {
    panic(err)
}

var m2 map[mapKey]string
if err := Load("testfile", &m2); err != nil {
    panic(err)
}
fmt.Println(m2)

按预期输出:

map[{1 2}:12 {3 4}:34]
map[{1 2}:12 {3 4}:34]