Java实现CSV数据到XML属性格式转换教程

Col1,Col2,Col3,Col4,Col5
All,0,,0,0
All,935,231,0,30
None,1011,257,0,30

<?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>

<dataset>
    <row>
        <Col1>All</Col1>
        <Col2>0</Col2>
        <Col3></Col3>
        <Col4>0</Col4>
        <Col5>0</Col5>
    </row>
    <!-- ... 更多行 ... -->
</dataset>

import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlRootElement;
import java.util.List;
import java.util.ArrayList;

// 根元素包装类
@XmlRootElement(name = "root")
public class RootElement {
    private List<Row> rows = new ArrayList<>();

    @XmlElement(name = "row")
    public List<Row> getRows() {
        return rows;
    }

    public void setRows(List<Row> rows) {
        this.rows = rows;
    }

    public void addRow(Row row) {
        this.rows.add(row);
    }
}

import javax.xml.bind.annotation.XmlAttribute;
import javax.xml.bind.annotation.XmlRootElement;
import javax.xml.bind.annotation.XmlAccessorType;
import javax.xml.bind.annotation.XmlAccessType;

// 表示CSV每一行的POJO
@XmlRootElement(name = "row") // 可选，但为了清晰性可以加上
@XmlAccessorType(XmlAccessType.FIELD) // 指定JAXB访问字段
public class Row {
    @XmlAttribute(name = "col1")
    private String col1;

    @XmlAttribute(name = "Col2") // 注意：XML属性名是大小写敏感的
    private String col2;

    @XmlAttribute(name = "Col3")
    private String col3;

    @XmlAttribute(name = "Col4")
    private String col4;

    @XmlAttribute(name = "Col5")
    private String col5;

    // 无参构造函数是JAXB的要求
    public Row() {}

    public Row(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和Setter方法 (JAXB在XmlAccessType.FIELD时不需要，但良好实践建议保留)
    public String getCol1() { return col1; }
    public void setCol1(String col1) { this.col1 = col1; }
    public String getCol2() { return col2; }
    public void setCol2(String col2) { this.col2 = col2; }
    public String getCol3() { return col3; }
    public void setCol3(String col3) { this.col3 = col3; }
    public String getCol4() { return col4; }
    public void setCol4(String col4) { this.col4 = col4; }
    public String getCol5() { return col5; }
    public void setCol5(String col5) { this.col5 = col5; }
}

import javax.xml.bind.JAXBContext;
import javax.xml.bind.Marshaller;
import java.io.BufferedReader;
import java.io.FileReader;
import java.io.File;
import java.io.IOException;
import java.util.Arrays;
import java.util.List;
import java.util.ArrayList;

public class CsvToXmlConverter {

    public static void convertCsvToXmlWithAttributes(String csvFileName, String xmlFileName, String delimiter) {
        RootElement root = new RootElement();
        List<String> header = new ArrayList<>();

        try (BufferedReader csvReader = new BufferedReader(new FileReader(csvFileName))) {
            String line;
            boolean isHeader = true;

            while ((line = csvReader.readLine()) != null) {
                if (line.trim().isEmpty()) {
                    continue; // 跳过空行
                }

                String[] values = line.split(delimiter, -1); // -1参数确保保留末尾的空字符串

                if (isHeader) {
                    // 假设第一行是CSV头，我们在这里不直接使用它来创建Row对象，
                    // 但在更复杂的场景中，可以用来动态映射列名
                    header.addAll(Arrays.asList(values));
                    isHeader = false;
                } else {
                    // 创建并填充Row对象
                    // 这里假设CSV列的顺序与Row类的字段顺序一致
                    // 实际应用中可能需要更健壮的映射逻辑
                    if (values.length >= 5) { // 确保有足够的列
                        Row row = new Row(
                            values[0].trim(),
                            values[1].trim(),
                            values[2].trim(),
                            values[3].trim(),
                            values[4].trim()
                        );
                        root.addRow(row);
                    } else {
                        System.err.println("警告: CSV行数据列数不足，跳过此行: " + line);
                    }
                }
            }

            // 使用JAXB进行编组
            JAXBContext jaxbContext = JAXBContext.newInstance(RootElement.class, Row.class);
            Marshaller jaxbMarshaller = jaxbContext.createMarshaller();

            // 设置输出格式
            jaxbMarshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
            jaxbMarshaller.setProperty(Marshaller.JAXB_ENCODING, "UTF-8"); // 设置编码

            // 将对象编组到文件
            jaxbMarshaller.marshal(root, new File(xmlFileName));
            System.out.println("CSV数据已成功转换为XML文件: " + xmlFileName);

        } catch (IOException e) {
            System.err.println("文件操作错误: " + e.getMessage());
        } catch (Exception e) {
            System.err.println("JAXB或数据处理错误: " + 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";
        String csvFileName = "sample.csv";
        try {
            java.nio.file.Files.write(java.nio.file.Paths.get(csvFileName), csvContent.getBytes());
        } catch (IOException e) {
            e.printStackTrace();
        }

        String xmlFileName = "output_with_attributes.xml";
        String delimiter = ","; // CSV分隔符

        convertCsvToXmlWithAttributes(csvFileName, xmlFileName, delimiter);
    }
}