Golang中如何将JSON对象保存到平面文件

只有通过实际操作才能更好地理解这些内容。感谢您帮助我理解。

要将包含不同类型结构体的映射保存为JSON文件并正确解组，可以使用以下方法：

package main

import (
	"encoding/json"
	"fmt"
	"os"
)

type struct1 struct {
	St1B bool
	St1I int
}

type struct2 struct {
	St2S string
	St2B bool
}

var Configs map[string]interface{}

func saveToFile(filename string) error {
	// 将整个Configs映射序列化为JSON
	data, err := json.MarshalIndent(Configs, "", "  ")
	if err != nil {
		return err
	}

	// 写入文件
	return os.WriteFile(filename, data, 0644)
}

func loadFromFile(filename string) error {
	// 读取文件内容
	data, err := os.ReadFile(filename)
	if err != nil {
		return err
	}

	// 临时存储解组后的数据
	tempConfigs := make(map[string]json.RawMessage)
	
	// 第一层解组到RawMessage
	if err := json.Unmarshal(data, &tempConfigs); err != nil {
		return err
	}

	// 重新初始化Configs
	Configs = make(map[string]interface{})

	// 根据键名分别解组到对应的结构体
	for key, raw := range tempConfigs {
		switch key {
		case "st1":
			var st1 struct1
			if err := json.Unmarshal(raw, &st1); err != nil {
				return err
			}
			Configs[key] = &st1
		case "st2":
			var st2 struct2
			if err := json.Unmarshal(raw, &st2); err != nil {
				return err
			}
			Configs[key] = &st2
		default:
			// 对于未知类型，保持原始JSON数据
			var unknown interface{}
			if err := json.Unmarshal(raw, &unknown); err != nil {
				return err
			}
			Configs[key] = unknown
		}
	}

	return nil
}

func main() {
	// 初始化数据
	pp := &struct1{
		St1B: true,
		St1I: 23,
	}

	dd := &struct2{
		St2S: "Hello",
		St2B: false,
	}

	Configs = make(map[string]interface{})
	Configs["st1"] = pp
	Configs["st2"] = dd

	// 保存到文件
	if err := saveToFile("config.json"); err != nil {
		fmt.Println("保存错误:", err)
		return
	}

	// 修改前读取文件
	if err := loadFromFile("config.json"); err != nil {
		fmt.Println("读取错误:", err)
		return
	}

	// 验证读取的数据
	if st1, ok := Configs["st1"].(*struct1); ok {
		fmt.Printf("读取的st1: St1B=%v, St1I=%d\n", st1.St1B, st1.St1I)
	}

	if st2, ok := Configs["st2"].(*struct2); ok {
		fmt.Printf("读取的st2: St2S=%s, St2B=%v\n", st2.St2S, st2.St2B)
	}

	// 修改数据示例
	if st1, ok := Configs["st1"].(*struct1); ok {
		st1.St1I = 100
		st1.St1B = false
	}

	if st2, ok := Configs["st2"].(*struct2); ok {
		st2.St2S = "Modified"
		st2.St2B = true
	}

	// 保存修改后的数据
	if err := saveToFile("config.json"); err != nil {
		fmt.Println("保存修改错误:", err)
		return
	}
}

生成的JSON文件内容：

{
  "st1": {
    "St1B": true,
    "St1I": 23
  },
  "st2": {
    "St2S": "Hello",
    "St2B": false
  }
}

这种方法的关键点：

1. 使用json.RawMessage进行两阶段解组，先获取原始JSON数据，再根据键名分配到具体结构体
2. 在loadFromFile函数中通过switch语句根据键名选择对应的结构体类型
3. 保存时直接序列化整个Configs映射
4. 读取时能正确恢复原始结构体类型，支持后续修改操作

这样就能实现将包含不同类型结构体的映射保存到文件，并在需要时正确读取和修改。