JAVA-Dom4j: 读取与写入xml文件

前言

使用dom4j读取与写入XML文件操作。
祖传JDK 1.8
dom4j-2.1.1
jaxen-1.1.6

XML文件结构

命名为book.xml
存放路径:

xml文件内容为:

<?xml version="1.0" encoding="UTF-8"?>
<!-- 数据文件 -->
<root>
	<book id="1">
		<category>1</category>
		<cname>文学传记类</cname>
	</book>
</root>

实体类

Book.java

package com.sishi.bookCategory.entity;

public class Book {
	private Integer id;
	private String cname;
	private Integer category;
	public Book() {
		
	}
	
	public Integer getCategory() {
		return category;
	}

	public void setCategory(Integer category) {
		this.category = category;
	}

	public Integer getId() {
		return id;
	}
	public void setId(Integer id) {
		this.id = id;
	}
	public String getCname() {
		return cname;
	}
	public void setCname(String cname) {
		this.cname = cname;
	}
	
}

工具类

xml数据操作
XmlSource.java

package com.sishi.bookCategory.utils;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.List;

import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.Node;
import org.dom4j.io.SAXReader;

import com.sishi.bookCategory.entity.Book;

public class XmlSource {
	private static List<Book> data = new ArrayList<Book>();
	private static String dataFile;
	
	static {
		dataFile=XmlSource.class.getResource("/book.xml").getPath();
		reload();
	}
	
	//重新加载数据
	private static void reload() {
		URLDecoder decoder = new URLDecoder();
		
		try {
			dataFile = decoder.decode(dataFile,"UTF-8");
			SAXReader reader = new SAXReader();
			Document document = reader.read(dataFile);
			
			List<Node>nodes = document.selectNodes("/root/book");
			data.clear();
			for(Node node:nodes) {
				Element element =(Element)node;
				String id = element.attributeValue("id");
				String cname = element.elementText("cname");
				String category = element.elementText("category");
				
				Book book = new Book();
				book.setId(Integer.parseInt(id));
				book.setCname(cname);

				book.setCategory(Integer.parseInt(category));
				
				data.add(book);
				
			}
			
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
	
	public static List<Book> getRawData(){
		return data;
	}
	
	//追加数据
	public static void append(Book book) {

		SAXReader reader = new SAXReader();
		Writer writer=null;
		try {
			//读取XML文档,得到Document对象
			Document document = reader.read(dataFile);
			//2.创建新的book节点
			//获取根节点
			Element root = document.getRootElement();
			//添加根节点的子节点->新的book节点
			Element b = root.addElement("book");
			//3.创建book节点的各个子节点
			//3.1添加子节点的属性
			b.addAttribute("id", String.valueOf(data.size()+1));//原有长度+1
			//3.2添加子节点的的子节点category
			b.addElement("category").setText(book.getCategory().toString());
			//3.3添加子节点的子节点cname
			b.addElement("cname").setText(book.getCname());
			
			//4.写入XML,完成追加操作
			writer = new OutputStreamWriter(new FileOutputStream(dataFile),"UTF-8");
			document.write(writer);
			System.out.println(dataFile);		
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}finally {
			//关闭写入操作
			if(writer!=null) {
				try {
					writer.close();
				} catch (IOException e) {
					// TODO Auto-generated catch block
					e.printStackTrace();
				}
			}
			//重新加载
			reload();
		}
			
	}
	
	/**
	 * 通过id获取并删除对应节点
	 * @param id
	 */
	public static void delete(Integer id) {
		SAXReader reader = new SAXReader();
		Writer writer=null;
		try {
			Document document = reader.read(dataFile);
			List<Node> nodes = document.selectNodes("/root/book[@id="+id+"]");
			
			if(nodes.size()==0) {
				throw new RuntimeException("分类ID:"+id+"不存在");
			}else {
				Element p = (Element)nodes.get(0);
				p.getParent().remove(p);
				
				writer = new OutputStreamWriter(new FileOutputStream(dataFile));
				document.write(writer);
			}
		} catch (Exception e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}finally {
			if(writer != null) {
				try {
					writer.close();
				} catch (IOException e) {
					// TODO Auto-generated catch block
					e.printStackTrace();
				}
			}
			reload();
		}
			
	}
	
	/**
	 * 主方法,测试用
	 * @param args
	 */
	public static void main(String args[]) {

		Book book = new Book();
		book.setCategory(2);
		book.setCname("异世大陆");
		XmlSource.append(book);
		
		List<Book>tests = XmlSource.getRawData();
		for(Book test:tests) {
			System.out.println(test.getId());
			System.out.println(test.getCname());
			System.out.println(test.getCategory());
		}
	}
	
}