package org.lee.dom4j.demo;
import java.io.File;
import java.io.FileWriter;
import java.util.Iterator;
import java.util.List;
import org.dom4j.Attribute;
import org.dom4j.Document;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;
/**
* 创建日期:2008-5-4 创建时间:上午11:26:21 创建作者:李赵华
*/
public class Dom4jDemo {
public Dom4jDemo() {
}
/**
*
* 建立一个XML文档,文档名由输入参数决定
*
* @param filename
* 需建立的文件名
*
* @return 返回操作结果, 0表失败, 1表成功
*
*/
public int createXMLFile(String fileName) {
/** 返回操作结果, 0表失败, 1表成功 */
int returnValue = 0;
/** 建立document对象 */
Document document = DocumentHelper.createDocument();
/** 建立XML文档的根books */
Element booksElement = document.addElement("books");
/** 加入一行注释 */
booksElement.addComment("This is a test for dom4j, 李赵华, 2008.5.4");
/** 加入第一个book节点 */
Element bookElement = booksElement.addElement("book");
/** 加入show参数内容 */
bookElement.addAttribute("show", "yes");
/** 加入title节点 */
Element titleElement = bookElement.addElement("title");
/** 为title设置内容 */
titleElement.setText("九阳神功");
/** 类似的完成后两个book */
bookElement = booksElement.addElement("book");
bookElement.addAttribute("show", "yes");
titleElement = bookElement.addElement("title");
titleElement.setText("玉女心经");
bookElement = booksElement.addElement("book");
bookElement.addAttribute("show", "no");
titleElement = bookElement.addElement("title");
titleElement.setText("降龙十八掌");
/** 加入owner节点 */
Element ownerElement = booksElement.addElement("owner");
ownerElement.setText("张无忌");
try {
OutputFormat format = OutputFormat.createPrettyPrint();
format.setEncoding("UTF-8");
/** 将document中的内容写入文件中 */
XMLWriter writer = new XMLWriter(
new java.io.OutputStreamWriter(new java.io.FileOutputStream(fileName),"UTF-8"));
writer.write(document);
writer.close();
/** 执行成功,需返回1 */
returnValue = 1;
} catch (Exception ex) {
ex.printStackTrace();
}
return returnValue;
}
/**
*
* 修改XML文件中内容,并另存为一个新文件 重点掌握dom4j中如何添加节点,修改节点,删除节点
*
* @param filename
* 修改对象文件
* @param newfilename
* 修改后另存为该文件
* @return 返回操作结果, 0表失败, 1表成功
*/
public int ModiXMLFile(String fileName, String newFileName) {
int returnValue = 0;
try {
SAXReader saxReader = new SAXReader();
//saxReader.setEncoding("UTF-8");
Document document = saxReader.read(new File(fileName));
/** 修改内容之一: 如果book节点中show参数的内容为yes,则修改成no */
/** 先用xpath查找对象 */
List list = document.selectNodes("/books/book/@show");
Iterator iter = list.iterator();
while (iter.hasNext()) {
Attribute attribute = (Attribute) iter.next();
if (attribute.getValue().equals("yes")) {
attribute.setValue("no");
}
}
/**
* 修改内容之二: 把owner项内容改为 西本樽月
* 并在owner节点中加入date节点,date节点的内容为2008-05-04,还为date节点添加一个参数type
*/
list = document.selectNodes("/books/owner");
iter = list.iterator();
if (iter.hasNext()) {
Element ownerElement = (Element) iter.next();
ownerElement.setText(" ");
ownerElement.addAttribute("name", "李赵华");
Element dateElement = ownerElement.addElement("date");
dateElement.setText("2008-05-04");
dateElement.addAttribute("type", "Gregorian calendar");
}
/** 修改内容之三: 若title内容为九阳神功,则删除该节点 */
list = document.selectNodes("/books/book");
iter = list.iterator();
while (iter.hasNext()) {
Element bookElement = (Element) iter.next();
Iterator iterator = bookElement.elementIterator("title");
while (iterator.hasNext()) {
Element titleElement = (Element) iterator.next();
if (titleElement.getText().equals("九阳神功")) {
bookElement.remove(titleElement);
}
}
}
try {
OutputFormat format = OutputFormat.createPrettyPrint();
format.setEncoding("UTF-8");
/** 将document中的内容写入文件中 */
XMLWriter writer = new XMLWriter(new FileWriter(new File(newFileName)),format);
writer.write(document);
writer.close();
/** 执行成功,需返回1 */
returnValue = 1;
} catch (Exception ex) {
ex.printStackTrace();
}
} catch (Exception ex) {
ex.printStackTrace();
}
return returnValue;
}
/**
* 格式化XML文档,并解决中文问题
* @param filename
* @return
*/
public int formatXMLFile(String filename){
int returnValue = 0;
try{
SAXReader saxReader = new SAXReader();
Document document = saxReader.read(new File(filename));
XMLWriter output = null;
/** 格式化输出,类型IE浏览一样 */
OutputFormat format = OutputFormat.createPrettyPrint();
/** 指定XML字符集编码 */
format.setEncoding("GBK");
output = new XMLWriter(new FileWriter(new File(filename)),format);
output.write(document);
output.close();
/** 执行成功,需返回1 */
returnValue = 1;
}catch(Exception ex){
ex.printStackTrace();
}
return returnValue;
}
public static void main(String[] args) {
Dom4jDemo temp = new Dom4jDemo();
System.out.println(temp.createXMLFile("d://book1.xml"));
System.out.println("成功");
System.out.println(temp.ModiXMLFile("d://book1.xml", "d://book2.xml"));
System.out.println(temp.formatXMLFile("d://book1.xml"));
}
}