使用Go语言将通用CSV文件转换为XML格式的教程

Name,Age,City
Alice,30,New York
Bob,24,London
Charlie,35,Paris

<?xml version="1.0" encoding="UTF-8"?>
<records>
  <record Name="Alice" Age="30" City="New York"/>
  <record Name="Bob" Age="24" City="London"/>
  <record Name="Charlie" Age="35" City="Paris"/>
</records>

package main

import (
    "encoding/csv"
    "fmt"
    "io/ioutil"
    "log"
    "os"
    "strings"
)

// xmlHeader 定义了标准的XML声明
const xmlHeader = `<?xml version="1.0" encoding="UTF-8"?>` + "\n"

func main() {
    csvFilePath := "data.csv" // 输入CSV文件路径
    xmlFilePath := "output.xml" // 输出XML文件路径

    // 1. 打开并读取CSV文件
    file, err := os.Open(csvFilePath)
    if err != nil {
        log.Fatalf("Error opening CSV file: %v", err)
    }
    defer file.Close()

    reader := csv.NewReader(file)
    // 如果CSV使用其他分隔符，可以在这里设置，例如 reader.Comma = ';'
    records, err := reader.ReadAll()
    if err != nil {
        log.Fatalf("Error reading CSV file: %v", err)
    }

    if len(records) == 0 {
        log.Println("CSV file is empty, no XML will be generated.")
        return
    }

    // 2. 获取CSV的标题行
    header := records[0]
    // 确保标题没有特殊字符，这里简单处理，实际应用可能需要更复杂的净化
    for i, h := range header {
        header[i] = strings.TrimSpace(h)
        // 进一步处理，例如替换空格为下划线，或移除非法XML属性字符
        // header[i] = strings.ReplaceAll(header[i], " ", "_")
    }

    // 3. 构建XML字符串
    var xmlBuilder strings.Builder
    xmlBuilder.WriteString(xmlHeader)
    xmlBuilder.WriteString("<records>\n")

    // 遍历CSV数据行，从第二行开始 (跳过标题行)
    for i := 1; i < len(records); i++ {
        row := records[i]
        xmlBuilder.WriteString("  <record ") // 开始一个新记录元素

        // 遍历当前行的数据，将其作为属性添加到 <record> 元素中
        for j := 0; j < len(row) && j < len(header); j++ {
            attrName := header[j]
            attrValue := row[j]

            // XML属性值需要进行转义，这里使用fmt.Sprintf的%q来处理字符串，
            // 但更严谨的XML转义应该使用 encoding/xml.EscapeText 或自定义函数
            // 对于简单的字符串，直接拼接并确保引号正确即可
            xmlBuilder.WriteString(fmt.Sprintf(`%s="%s" `, attrName, escapeXMLAttribute(attrValue)))
        }
        xmlBuilder.WriteString("/>\n") // 结束当前记录元素
    }

    xmlBuilder.WriteString("</records>")

    // 4. 将构建好的XML字符串写入文件
    err = ioutil.WriteFile(xmlFilePath, []byte(xmlBuilder.String()), 0644)
    if err != nil {
        log.Fatalf("Error writing XML file: %v", err)
    }

    fmt.Printf("Successfully converted '%s' to '%s'\n", csvFilePath, xmlFilePath)
}

// escapeXMLAttribute 简单地转义XML属性值中的特殊字符
// 实际生产环境中可能需要更全面的转义逻辑
func escapeXMLAttribute(s string) string {
    s = strings.ReplaceAll(s, "&", "&")
    s = strings.ReplaceAll(s, "<", "<")
    s = strings.ReplaceAll(s, ">", ">")
    s = strings.ReplaceAll(s, `"`, """)
    s = strings.ReplaceAll(s, "'", "&apos;")
    return s
}