Java基础——利用DOM4J来读取、修改、输出XML文件

本文介绍了如何使用DOM4J库在Java中读取、修改XML文件。从数据读取开始，通过SAXReader加载XML内容到Document对象。接着阐述了保存XML文件的步骤，利用OutputFormat和XMLWriter将修改后的信息输出到文件。文章还详细讲解了如何在根节点下添加Brand父节点，并在Brand节点下插入Type子节点。最后，提供了完整的代码示例。

前言：

这里有一份XML文件，我们需要为其添加内容并保存输出。

<GamerInfo>
    <Brand name="索尼">
        <Type name="PSV" price="2000"/>
        <Type name="PS4" price="3500"/>
        <Type name="PS5" price="4800"/>
    </Brand>
    <Brand name="任天堂">
        <Type name="Switch" price="2100"/>
    </Brand>
</GamerInfo>

一、XML文件的数据读取

与TransfomenterFactory类似，首先用Document类的对象储存XML文件信息。j接着利用SAXReader类的方法来实现XML文件的数据读取。

    //创建Document类的document对象来储存XML信息
    Document document = null;
    //创建SAXReader类的对象来读取XML文件信息
    SAXReader saxReader = new SAXReader();
    try {
        //用SAXReader的read方法,将XML信息储存到document对象中
        this.document = saxReader.read("src/main/java/JsonPractice/Gamers.xml");
    } catch (DocumentException e) {
        e.printStackTrace();
    }

二、XML文件的保存

首先，创建OutpitFormat对象来储存要输出的XML信息。接着创建XMLWriter类的对象，用来存储新建文件的路径，以及新建文件内的信息。最后用XMLWriter类的write方法，来输出储存XML信息的文件。

         //创建OutputFormat类的对象来储存XML信息
        OutputFormat format = OutputFormat.createCompactFormat();
        try {
            //创建XMLWriter类的对象，来储存输出的XMl文件路径和要输出的XML信息
            XMLWriter writer = new XMLWriter(new FileWriter(path), format);
            //调用XMLWriter类的write方法，来输出储存XML信息的文件
            writer.write(document);
            writer.close();
        } catch (IOException e) {
            e.printStackTrace();
        }

三、增加根节点下的父节点Brand

第一步，获取梗目录下的子节点，即Brand节点；并将其存入Element类的对象。第二步，遍历所有Brand节点。第三步，当遍历到的Brand节点和输入的一致时，说明节点存在不需要创建新的Brand节点，否则就要创建新的节点。

        //获取梗目录下的子节点，即Brand节点；并将其存入Element类的对象
        Element root = document.getRootElement();
        //存储单个的Brand节点
        Element brand = null;
        //判断品牌是否存在
        boolean tag = false;
        //遍历所有Brand节点
        Iterator BrandIterator = root.elementIterator();
        while (BrandIterator.hasNext()) {
            //获取每次遍历得到的Brand节点
            Element brands = (Element) BrandIterator.next();
            //当遍历到的Brand节点和输入的一致时，说明节点存在不需要创建新的Brand节点
            if (brands.attributeValue("name").equals(brandName)) {
                brand = brands;
                tag = true;
                System.out.println("品牌已存在，无需创建！");
                break;
            }
        }
        if (!tag) {
            //在跟目录下添加新的Brand节点
            brand = root.addElement("Brand");
            System.out.println("品牌创建成功！");
        }
        brand.addAttribute("name", brandName);
        //将修改的信息保存到指定路径的文件中
        this.saveXML("src/main/java/JsonPractice/Gamers.xml");

四、添加Brand节点的子节点Type节点

与添加Brand节点类似，每次遍历其中一个Brand节点后，对这个Brand节点的子节点进行遍历，如果不存在Tpye节点，便创建新的Type节点。

   public void addGamer(String brandName, String typeName, String phonePrice) {
        //判断是否增加Brand节点
        this.addBrand(brandName);
        //获取梗目录下的子节点，即Brand节点；并将其存入Element类的对象
        Element root = document.getRootElement();
        //存储单个的Brand节点
        Element brand = null;
        //遍历所有Brand节点
        Iterator BrandIterator = root.elementIterator();
        while (BrandIterator.hasNext()) {
            //获取每次遍历得到的Brand节点
            Element brands = (Element) BrandIterator.next();
            //找到所需节点
            if (brands.attributeValue("name").equals(brandName)) {
                brand = brands;
                break;
            }
        }
        //遍历Brand下的字节点，即Type节点
        Iterator types = brand.elementIterator();
        while (types.hasNext()) {
            //将获取的Type节点存储到Element类的对象中去
            Element typeNext = (Element) types.next();
            //判断是否重复添加
            if (typeNext.attributeValue("name").equals(typeName)) {
                System.out.println("机型已存在，无法添加！");
                return;
            }
        }
        Element type = brand.addElement("type");
        type.addAttribute("name", typeName);
        type.addAttribute("price", phonePrice);
        System.out.println("机型添加成功！");
        this.saveXML("src/main/java/JsonPractice/Gamers.xml");

    }

五、完整代码展示

package JsonPractice;

import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.Document;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Iterator;

public class GameBoy {
    //创建Document类的document对象来储存XML信息
    Document document = null;
    //用Document类的对象来获取并存储XML文件信息
    public void getDocument(){
        //创建SAXReader类的对象来读取XML文件信息
        SAXReader saxReader = new SAXReader();
        try {
            //用SAXReader的read方法,将XML信息储存到document对象中
            this.document = saxReader.read("src/main/java/JsonPractice/Gamers.xml");
        } catch (DocumentException e) {
            e.printStackTrace();
        }
    }

    //将信息保存至XML文件的方法
    public void saveXML(String path) {
        //创建OutputFormat类的对象来储存XML信息
        OutputFormat format = OutputFormat.createCompactFormat();
        try {
            //创建XMLWriter类的对象，来储存输出的XMl文件路径和要输出的XML信息
            XMLWriter writer = new XMLWriter(new FileWriter(path), format);
            //调用XMLWriter类的write方法，来输出储存XML信息的文件
            writer.write(document);
            writer.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public boolean addBrand(String brandName){
        //获取梗目录下的子节点，即Brand节点；并将其存入Element类的对象
        Element root = document.getRootElement();
        //存储单个的Brand节点
        Element brand = null;
        //判断品牌是否存在
        boolean tag = false;
        //遍历所有Brand节点
        Iterator BrandIterator = root.elementIterator();
        while (BrandIterator.hasNext()) {
            //获取每次遍历得到的Brand节点
            Element brands = (Element) BrandIterator.next();
            //当遍历到的Brand节点和输入的一致时，说明节点存在不需要创建新的Brand节点
            if (brands.attributeValue("name").equals(brandName)) {
                brand = brands;
                tag = true;
                System.out.println("品牌已存在，无需创建！");
                break;
            }
        }
        if (!tag) {
            brand = root.addElement("Brand");
            System.out.println("品牌创建成功！");
        }
        brand.addAttribute("name", brandName);
        this.saveXML("src/main/java/JsonPractice/Gamers.xml");
        return tag;
    }


    public void addGamer(String brandName, String typeName, String phonePrice) {
        //判断是否增加Brand节点
        this.addBrand(brandName);
        //获取梗目录下的子节点，即Brand节点；并将其存入Element类的对象
        Element root = document.getRootElement();
        //存储单个的Brand节点
        Element brand = null;
        //遍历所有Brand节点
        Iterator BrandIterator = root.elementIterator();
        while (BrandIterator.hasNext()) {
            //获取每次遍历得到的Brand节点
            Element brands = (Element) BrandIterator.next();
            //找到所需节点
            if (brands.attributeValue("name").equals(brandName)) {
                brand = brands;
                break;
            }
        }
        //遍历Brand下的字节点，即Type节点
        Iterator types = brand.elementIterator();
        while (types.hasNext()) {
            //将获取的Type节点存储到Element类的对象中去
            Element typeNext = (Element) types.next();
            //判断是否重复添加
            if (typeNext.attributeValue("name").equals(typeName)) {
                System.out.println("机型已存在，无法添加！");
                return;
            }
        }
        Element type = brand.addElement("type");
        type.addAttribute("name", typeName);
        type.addAttribute("price", phonePrice);
        System.out.println("机型添加成功！");
        this.saveXML("src/main/java/JsonPractice/Gamers.xml");

    }

    public static void main(String[] args) {
        //进行测试
        GameBoy gameBoy = new GameBoy();
        gameBoy.getDocument();
        gameBoy.addBrand("世嘉");
        gameBoy.addGamer("世嘉","土星","3000");
        gameBoy.addBrand("世嘉");
        gameBoy.addGamer("世嘉","海王星","Unknown");

    }

}