Golang实现XML文档XPath查询与解析

在Golang中处理XML文档的XPath查询，可以使用github.com/antchfx/xpath库，这是一个功能强大且高效的XPath实现。下面我将详细介绍如何使用这个库进行XML文档的XPath查询与解析。

安装xpath库

首先需要安装xpath库：

go get github.com/antchfx/xpath

基本使用方法

1. 解析XML文档

package main

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

	"github.com/antchfx/xpath"
)

func main() {
	// XML示例数据
	data := `
	<bookstore>
		<book category="cooking">
			<title lang="en">Everyday Italian</title>
			<author>Giada De Laurentiis</author>
			<year>2005</year>
			<price>30.00</price>
		</book>
		<book category="children">
			<title lang="en">Harry Potter</title>
			<author>J.K. Rowling</author>
			<year>2005</year>
			<price>29.99</price>
		</book>
	</bookstore>
	`

	// 解析XML文档
	doc := &xmlquery.Node{}
	if err := xmlquery.Unmarshal([]byte(data), doc); err != nil {
		panic(err)
	}

	// 创建XPath表达式
	expr, err := xpath.Compile("//book[price>29]/title/text()")
	if err != nil {
		panic(err)
	}

	// 执行XPath查询
	iter := expr.Evaluate(xmlquery.CreateXPathNavigator(doc)).(*xpath.NodeIterator)
	for iter.MoveNext() {
		fmt.Println(iter.Current().Value())
	}
}

2. 常用XPath查询示例

// 1. 获取所有book节点
expr := xpath.MustCompile("//book")
iter := expr.Evaluate(xmlquery.CreateXPathNavigator(doc)).(*xpath.NodeIterator)

// 2. 获取价格大于29的书籍标题
expr = xpath.MustCompile("//book[price>29]/title/text()")

// 3. 获取特定语言的标题
expr = xpath.MustCompile("//title[@lang='en']")

// 4. 获取第一个book节点
expr = xpath.MustCompile("//book[1]")

// 5. 获取所有作者
expr = xpath.MustCompile("//author/text()")

3. 处理查询结果

// 遍历查询结果
iter := expr.Evaluate(xmlquery.CreateXPathNavigator(doc)).(*xpath.NodeIterator)
for iter.MoveNext() {
    node := iter.Current().(*xmlquery.Node)
    switch node.Type {
    case xmlquery.TextNode:
        fmt.Println("Text:", node.Data)
    case xmlquery.ElementNode:
        fmt.Println("Element:", node.Data)
        for _, attr := range node.Attr {
            fmt.Printf("  Attr: %s=%s\n", attr.Name.Local, attr.Value)
        }
    }
}

高级用法

1. 自定义函数

// 注册自定义XPath函数
xpath.RegisterFunction("contains-ignore-case", func(args ...interface{}) interface{} {
    if len(args) != 2 {
        return false
    }
    s1 := args[0].(string)
    s2 := args[1].(string)
    return strings.Contains(strings.ToLower(s1), strings.ToLower(s2))
})

// 使用自定义函数
expr := xpath.MustCompile("//book[contains-ignore-case(title, 'harry')]")

2. 性能优化

对于大型XML文档，可以使用流式解析：

import "github.com/antchfx/xmlquery"

// 流式解析大型XML文件
f, err := os.Open("large.xml")
if err != nil {
    panic(err)
}
defer f.Close()

doc, err := xmlquery.Parse(f)
if err != nil {
    panic(err)
}

完整示例

下面是一个完整的示例，演示如何从XML文件中查询并处理数据：

package main

import (
	"fmt"
	"log"

	"github.com/antchfx/xmlquery"
	"github.com/antchfx/xpath"
)

func main() {
	// 示例XML数据
	data := `
	<catalog>
		<product id="1">
			<name>Laptop</name>
			<price>999.99</price>
			<stock>15</stock>
		</product>
		<product id="2">
			<name>Smartphone</name>
			<price>699.99</price>
			<stock>32</stock>
		</product>
		<product id="3">
			<name>Tablet</name>
			<price>399.99</price>
			<stock>12</stock>
		</product>
	</catalog>
	`

	// 解析XML
	doc, err := xmlquery.Parse(strings.NewReader(data))
	if err != nil {
		log.Fatal(err)
	}

	// 查询价格低于700的产品名称
	expr := xpath.MustCompile("//product[price<700]/name/text()")
	iter := expr.Evaluate(xmlquery.CreateXPathNavigator(doc)).(*xpath.NodeIterator)

	fmt.Println("Affordable products:")
	for iter.MoveNext() {
		fmt.Println("-", iter.Current().Value())
	}

	// 查询库存少于20的产品ID
	expr = xpath.MustCompile("//product[stock<20]/@id")
	iter = expr.Evaluate(xmlquery.CreateXPathNavigator(doc)).(*xpath.NodeIterator)

	fmt.Println("\nLow stock product IDs:")
	for iter.MoveNext() {
		fmt.Println("-", iter.Current().Value())
	}
}

总结

github.com/antchfx/xpath库为Golang提供了强大的XPath查询能力，可以方便地从XML文档中提取和处理数据。通过本文介绍的基本查询、结果处理、自定义函数和性能优化等方法，您可以高效地实现各种XML处理需求。

对于更复杂的XML处理场景，您还可以考虑结合使用encoding/xml标准库和xpath库，以获得更大的灵活性。