从 Go 中提取 XML 属性

本文介绍了使用 Go 语言解析 XML 文档并提取特定属性的两种方法。针对 这种嵌套在 元素下的 XML 结构，我们将演示如何提取 id 属性的值，并提供相应的代码示例和注意事项。

方法一：逐个 Token 解析

第一种方法是使用 xml.Token() 函数逐个读取 XML 文档的 Token，直到找到名为 cwe 的元素。然后，定义一个结构体来映射 cwe 元素，并使用 XML 标签来指定 id 属性。

package main

import (
    "encoding/xml"
    "fmt"
    "strings"
)

type CweXml struct {
    Id string `xml:"id,attr"`
}

func main() {
    xmlData := ``
    decoder := xml.NewDecoder(strings.NewReader(xmlData))

    for {
        token, err := decoder.Token()
        if err != nil {
            break // End of document or error
        }

        switch se := token.(type) {
        case xml.StartElement:
            if se.Name.Local == "cwe" {
                var cwe CweXml
                if err := decoder.DecodeElement(&cwe, &se); err != nil {
                    fmt.Println("Error decoding cwe element:", err)
                    return
                }
                fmt.Println("CWE ID:", cwe.Id)
                return // Found and processed the CWE, exit the loop
            }
        }
    }
}

代码解释：

1. 定义 CweXml 结构体，其中 Id 字段使用 xml:"id,attr" 标签，表示映射 XML 元素的 id 属性。
2. 创建 xml.Decoder 对象，并使用 strings.NewReader 将 XML 字符串转换为 io.Reader。
3. 使用 decoder.Token() 循环读取 XML Token。
4. 判断 Token 类型是否为 xml.StartElement，并检查元素名称是否为 cwe。
5. 如果找到 cwe 元素，则使用 decoder.DecodeElement() 将元素内容解码到 CweXml 结构体中。
6. 打印 cwe.Id 的值。

注意事项：

• 这种方法需要遍历整个 XML 文档，直到找到目标元素。
• 需要处理 XML 文档结束或错误的异常情况。
• 该示例假设 XML 数据位于 标签内。

方法二：定义完整的 XML 结构体

第二种方法是定义一个完整的 XML 结构体，包括 entry 元素和 cwe 元素，并使用 XML 标签来指定它们之间的关系。

Digram

让Figma更好用的AI神器

下载

package main

import (
    "encoding/xml"
    "fmt"
    "strings"
)

type CweXml struct {
    Id string `xml:"id,attr"`
}

type EntryXml struct {
    Cwe CweXml `xml:"cwe"`
}

func main() {
    xmlData := ``
    var entry EntryXml
    err := xml.Unmarshal([]byte(xmlData), &entry)
    if err != nil {
        fmt.Println("Error unmarshaling XML:", err)
        return
    }

    fmt.Println("CWE ID:", entry.Cwe.Id)
}

代码解释：

1. 定义 CweXml 结构体，与方法一相同。
2. 定义 EntryXml 结构体，其中 Cwe 字段的类型为 CweXml，并使用 xml:"cwe" 标签表示 entry 元素包含一个 cwe 元素。
3. 使用 xml.Unmarshal() 将 XML 数据解码到 EntryXml 结构体中。
4. 打印 entry.Cwe.Id 的值。

注意事项：

• 这种方法需要预先了解 XML 文档的结构。
• 如果 XML 文档的结构比较复杂，则需要定义更复杂的结构体。
• xml.Unmarshal 在解析过程中会忽略命名空间，所以示例中的 vuln: 前缀可以忽略。如果需要处理命名空间，需要使用 xml.Decoder 和 xml.StartElement 来进行更精细的控制。

总结

以上两种方法都可以用于从 Go 中提取 XML 属性。第一种方法更加灵活，可以处理不规则的 XML 文档。第二种方法更加简洁，但需要预先了解 XML 文档的结构。选择哪种方法取决于具体的应用场景。 在实际应用中，需要根据 XML 结构的复杂度和性能要求选择合适的方法。