如何在 Go 中安全地将 YAML 转换为 JSON，而不丢失数据完整性？-Golang-PHP中文网

如何在 Go 中安全地将 YAML 转换为 JSON，而不丢失数据完整性？

Barbara Streisand

发布： 2024-12-01 18:02:12

原创

244 人浏览过

How to Safely Convert YAML to JSON in Go Without Losing Data Integrity?

在没有结构体绑定的情况下将 YAML 转换为 JSON

使用 YAML 格式的动态数据时，通常需要将其转换为 JSON 以便进一步处理。然而，直接将 YAML 解组为空的 interface{} 可能会在遇到 map[interface{}]interface{} 类型时导致问题，如给定场景所示。

解决方案：递归类型转换

为了解决这个问题，使用了一个名为convert()的递归函数。该函数迭代interface{}值并执行以下转换：

将map[interface{}]interface{}值转换为map[string]interface{}。
将 []interface{} 值转换为 []interface{}。

通过这样做，输出的值可以安全地编组为 JSON 字符串。

示例

以下是如何使用 Convert() 函数的示例：

import (
    "fmt"
    "log"

    "github.com/go-yaml/yaml"
    "encoding/json"
)

func convert(i interface{}) interface{} {
    switch x := i.(type) {
    case map[interface{}]interface{}:
        m2 := map[string]interface{}{}
        for k, v := range x {
            m2[k.(string)] = convert(v)
        }
        return m2
    case []interface{}:
        for i, v := range x {
            x[i] = convert(v)
        }
    }
    return i
}

func main() {
    // Define the YAML string
    const s = `Services:
    -   Orders: 
        -   ID: $save ID1
            SupplierOrderCode: $SupplierOrderCode
        -   ID: $save ID2
            SupplierOrderCode: 111111
    `

    // Unmarshal the YAML string into an empty interface
    var body interface{}
    if err := yaml.Unmarshal([]byte(s), &body); err != nil {
        log.Fatal(err)
    }

    // Recursively convert the interface to a map[string]interface{}
    body = convert(body)

    // Marshal the converted interface into a JSON string
    if b, err := json.Marshal(body); err != nil {
        log.Fatal(err)
    } else {
        fmt.Println("Converted JSON:", string(b))
    }
}

登录后复制

输出

程序的输出是转换后的JSON:

Converted JSON: {"Services":[{"Orders":[
    {"ID":"$save ID1","SupplierOrderCode":"$SupplierOrderCode"},
    {"ID":"$save ID2","SupplierOrderCode":111111}]}]}

登录后复制