1. 问题背景与传统方法分析
在数据交换和存储中，将csv（comma separated values）格式的数据转换为xml（extensible markup language）格式是一种常见需求。然而，有时我们需要的xml结构并非简单的元素嵌套，而是将csv的列名作为xml元素的属性（attribute），将列值作为属性值。例如，将如下csv数据：
Col1,Col2,Col3,Col4,Col5 All,0,,0,0 All,935,231,0,30 None,1011,257,0,30
转换为如下带有属性的XML格式：
<?xml version="1.0"?>
<root>
<row Col1="All" Col2="0" Col3="" Col4="0" Col5="0"></row>
<row Col1="All" Col2="935" Col3="231" Col4="0" Col5="30"></row>
<row Col1="None" Col2="1011" Col3="257" Col4="0" Col5="30"></row>
</root>传统的DOM（Document Object Model）操作方法，如直接使用Document、Element和appendChild等，在实现这种属性映射时容易出错。原始代码示例中，尝试通过newDoc.createElement(csvFields[i++])创建元素，然后将值作为文本节点添加到这些元素中，最终导致生成的是嵌套元素而非属性，例如：
<dataset>
<row>
<Col1>All</Col1>
<Col2>0</Col2>
<!-- ... -->
</row>
</dataset>这与我们期望的属性格式不符。为了更优雅、更声明式地实现CSV到XML属性格式的转换，JAXB框架是更优的选择。
2. JAXB解决方案：POJO与注解驱动转换
JAXB（Java Architecture for XML Binding）提供了一种将Java对象（POJO）与XML模式之间进行映射的机制。通过在POJO上使用特定的注解，我们可以声明式地定义Java对象如何序列化为XML（Marshalling）以及如何从XML反序列化为Java对象（Unmarshalling）。
立即学习“Java免费学习笔记（深入）”；
2.1 核心概念
- POJO (Plain Old Java Object)：普通的Java类，用于表示XML结构中的数据。
- JAXB注解：用于在POJO中定义Java字段或方法与XML元素、属性之间的映射关系。
- Marshaller：将Java对象转换为XML文档的过程。
- Unmarshaller：将XML文档转换为Java对象的过程。
2.2 定义数据模型（POJO）
为了将CSV的每一行数据转换为一个带有属性的<row>元素，我们需要定义两个POJO：一个用于表示CSV的单行数据（即XML中的<row>元素），另一个用于包装所有行数据并作为整个XML的根元素（即<root>元素）。
2.2.1 RowData.java：表示单行数据
这个POJO将映射到XML中的一个<row>元素，其字段将作为<row>元素的属性。
import javax.xml.bind.annotation.XmlAttribute;
import javax.xml.bind.annotation.XmlRootElement;
import javax.xml.bind.annotation.XmlType;
/**
* 表示CSV中的一行数据，将被映射为XML中的一个<row>元素，
* 其字段作为该元素的属性。
*/
@XmlRootElement(name = "row") // 将此POJO映射为XML中的<row>元素
@XmlType(propOrder = {"col1", "col2", "col3", "col4", "col5"}) // 可选：定义属性在XML中的顺序
public class RowData {
private String col1;
private String col2;
private String col3;
private String col4;
private String col5;
// JAXB需要一个无参构造函数
public RowData() {}
public RowData(String col1, String col2, String col3, String col4, String col5) {
this.col1 = col1;
this.col2 = col2;
this.col3 = col3;
this.col4 = col4;
this.col5 = col5;
}
// 每个getter方法都用@XmlAttribute注解，将其映射为XML属性
@XmlAttribute(name = "Col1")
public String getCol1() { return col1; }
public void setCol1(String col1) { this.col1 = col1; }
@XmlAttribute(name = "Col2")
public String getCol2() { return col2; }
public void setCol2(String col2) { this.col2 = col2; }
@XmlAttribute(name = "Col3")
public String getCol3() { return col3; }
public void setCol3(String col3) { this.col3 = col3; }
@XmlAttribute(name = "Col4")
public String getCol4() { return col4; }
public void setCol4(String col4) { this.col4 = col4; }
@XmlAttribute(name = "Col5")
public String getCol5() { return col5; }
public void setCol5(String col5) { this.col5 = col5; }
}2.2.2 RootWrapper.java：表示XML根元素
这个POJO将映射到XML中的<root>元素，并包含一个RowData对象的列表。
import java.util.List;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlRootElement;
/**
* 表示XML的根元素<root>，包含一个RowData对象的列表。
*/
@XmlRootElement(name = "root") // 将此POJO映射为XML中的<root>元素
public class RootWrapper {
private List<RowData> rows;
// JAXB需要一个无参构造函数
public RootWrapper() {}
public RootWrapper(List<RowData> rows) {
this.rows = rows;
}
// 使用@XmlElement注解将列表中的每个RowData对象映射为<row>元素
@XmlElement(name = "row")
public List<RowData> getRows() { return rows; }
public void setRows(List<RowData> rows) { this.rows = rows; }
}2.3 实现转换逻辑
有了POJO定义后，转换过程就变得相对简单：读取CSV文件，将每行数据解析成RowData对象，收集所有RowData对象到一个列表中，然后将这个列表封装到RootWrapper对象中，最后使用JAXB的Marshaller将RootWrapper对象序列化为XML。
import java.io.BufferedReader;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.Marshaller;
public class CsvToXmlConverter {
/**
* 将CSV文件转换为指定格式的XML文件。
*
* @param csvFileName CSV文件的路径
* @param xmlFileName 输出XML文件的路径
* @param delimiter CSV文件中字段的分隔符
*/
public static void convertFile(String csvFileName, String xmlFileName, String delimiter) {
List<RowData> rowDataList = new ArrayList<>();
List<String> header = new ArrayList<>(); // 用于存储CSV头信息
try (BufferedReader csvReader = new BufferedReader(new FileReader(csvFileName))) {
String line;
boolean isFirstLine = true;
while ((line = csvReader.readLine()) != null) {
if (line.trim().isEmpty()) {
continue; // 跳过空行
}
// 使用split方法解析CSV行，-1参数确保保留末尾的空字符串
String[] values = line.split(delimiter, -1);
if (isFirstLine) {
// 读取CSV头，可以用于动态映射或验证
header.addAll(Arrays.asList(values));
isFirstLine = false;
continue;
}
// 假设CSV每行有5列，并按顺序映射到RowData的属性
// 在实际应用中，可以根据header动态创建RowData对象
if (values.length >= 5) {
RowData row = new RowData(
values[0],
values[1],
values[2],
values[3],
values[4]
);
rowDataList.add(row);
} else {
System.err.println("警告: 跳过格式不正确的CSV行 (列数少于5): " + line);
}
}
// 创建RootWrapper对象，包含所有RowData
RootWrapper rootWrapper = new RootWrapper(rowDataList);
// 初始化JAXB上下文和Marshaller
// JAXBContext.newInstance需要注册所有涉及到的POJO类
JAXBContext jaxbContext = JAXBContext.newInstance(RootWrapper.class, RowData.class);
Marshaller jaxbMarshaller = jaxbContext.createMarshaller();
// 配置Marshaller，使输出XML格式化，提高可读性
jaxbMarshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
// 将Java对象序列化为XML并写入文件
try (FileWriter writer = new FileWriter(xmlFileName)) {
jaxbMarshaller.marshal(rootWrapper, writer);
System.out.println("CSV文件成功转换为XML文件: " + xmlFileName);
}
} catch (IOException e) {
System.err.println("读取CSV或写入XML文件时发生IO错误: " + e.getMessage());
e.printStackTrace();
} catch (Exception e) {
System.err.println("XML转换过程中发生意外错误: " + e.getMessage());
e.printStackTrace();
}
}
public static void main(String[] args) {
// 示例用法：创建虚拟CSV文件并执行转换
String csvContent = "Col1,Col2,Col3,Col4,Col5\n" +
"All,0,,0,0\n" +
"All,935,231,0,30\n" +
"None,1011,257,0,30\n";
String csvFileName = "input.csv";
String xmlFileName = "output.xml";
String delimiter = ",";
// 创建一个用于测试的CSV文件
try (FileWriter fw = new FileWriter(csvFileName)) {
fw.write(csvContent);
System.out.println("虚拟CSV文件 '" + csvFileName + "' 已创建。");
} catch (IOException e) {
System.err.println("创建虚拟CSV文件失败: " + e.getMessage());
return;
}
// 执行转换
convertFile(csvFileName, xmlFileName, delimiter);
}
}2.4 JAXB依赖说明
从Java 9及更高版本开始，JAXB API不再是JDK的默认模块。如果您的项目使用Java 9+，需要手动添加JAXB依赖。对于Maven项目，可以在pom.xml中添加如下依赖：
<dependencies>
<!-- JAXB API -->
<dependency>
<groupId>jakarta.xml.bind</groupId>
<artifactId>jakarta.xml.bind-api</artifactId>
<version>2.3.3</version> <!-- 或更高版本 -->
</dependency>
<!-- JAXB Runtime 
