DOM-4J

package com.firstflow;

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;

/**
 * @author Holen Chen
 */

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, holen, 2004.9.11");
        /** 加入第一个book节点 */
        Element bookElement = booksElement.addElement("book");
        /** 加入show参数内容 */
        bookElement.addAttribute("show", "yes");
        /** 加入title节点 */
        Element titleElement = bookElement.addElement("title");
        /** 为title设置内容 */
        titleElement.setText("Dom4j Tutorials");
        /** 类似的完成后两个book */
        bookElement = booksElement.addElement("book");
        bookElement.addAttribute("show", "yes");
        titleElement = bookElement.addElement("title");
        titleElement.setText("Lucene Studing");
        bookElement = booksElement.addElement("book");
        bookElement.addAttribute("show", "no");
        titleElement = bookElement.addElement("title");
        titleElement.setText("Lucene in Action");
        /** 加入owner节点 */
        Element ownerElement = booksElement.addElement("owner");
        ownerElement.setText("O'Reilly");
        try {
            /** 将document中的内容写入文件中 */
            XMLWriter writer = new XMLWriter(new FileWriter(new File(filename)));
            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();
            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项内容改为Tshinghua
             * 并在owner节点中加入date节点,date节点的内容为2004-09-11,还为date节点添加一个参数type
             */
            list = document.selectNodes("/books/owner");
            iter = list.iterator();
            if (iter.hasNext()) {
                Element ownerElement = (Element) iter.next();
                ownerElement.setText("Tshinghua");
                Element dateElement = ownerElement.addElement("date");
                dateElement.setText("2004-09-11");
                dateElement.addAttribute("type", "Gregorian calendar");
            }

            /** 修改内容之三: 若title内容为Dom4j Tutorials,则删除该节点 */
            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("Dom4j Tutorials")) {
                        bookElement.remove(titleElement);
                    }
                }
            }

            try {
                /** 将document中的内容写入文件中 */
                XMLWriter writer = new XMLWriter(new FileWriter(new File(
                        newfilename)));
                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();
        temp.createXMLFile("test.xml");       
    }
}