dom4j生成XML文档及解析

package com.dom4j.sample;
import java.io.File;
import java.io.FileOutputStream;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.dom4j.Attribute;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;


public class HandleXmlSample {
/**
* 建立一个XML文档,文档名由输入属性决定,文档内容在内部由initialDocument（Document document）进行初始化
*
* @param filename
*            需建立的文件名,如"test.xml",可以加路径
* @return返回操作结果, 0表失败, 1表成功
*/
public int createXMLFile(String filename) {
   // 返回操作结果, 0表失败, 1表成功
   int returnValue = 0;
   // 建立document对象
   Document document = DocumentHelper.createDocument();
   // 对document初始化，加入xml所需要的内容节点
   document = initialDocument(document);
   try {
    // 对document格式化输出到指定名称文件, 格式化后，使xml符合节点缩进样式
    formatXMLOutput(document, filename);
    // 输出正确，修改返回值为1
    returnValue = 1;
   } catch (Exception e) {
    System.out.println("格式化输出文件出错！");
    e.printStackTrace();
   }
   return returnValue;
}

/**
* <b>方法描述</b>： 创建xml字符串,文档内容在内部由initialDocument（Document document）进行初始化
* <p>
* <b>方法流程</b>：
* <p>
*
* @return
*/
public String createXMLStr() {
   // 建立document对象
   Document document = DocumentHelper.createDocument();
   // 对document初始化，加入xml所需要的内容节点
   document = initialDocument(document);
   return document.asXML();
}

/**
* <b>方法描述</b>： 初始化document对象
* <p>
* <b>方法流程</b>：
* <p>
*
* @return 返回初始化完成对象
*/

private Document initialDocument(Document document) {

   /** 建立XML文档的根books */
   Element booksElement = document.addElement("books");
   /** 加入一行注释 */
   booksElement.addComment("This is a test for dom4j");
   /** 加入第一个book节点 */
   Element bookElement = booksElement.addElement("book");
   /** 加入show属性内容 */
   bookElement.addAttribute("show", "yes");
   /** 加入title节点 */
   Element titleElement = bookElement.addElement("title");
   /** 为title设置内容 */
   titleElement.setText("java程序设计");

   /** 类似的完成后两个book */
   bookElement = booksElement.addElement("book");
   bookElement.addAttribute("show", "yes");
   titleElement = bookElement.addElement("title");
   titleElement.setText("java编程思想");
   bookElement = booksElement.addElement("book");
   bookElement.addAttribute("show", "no");
   titleElement = bookElement.addElement("title");
   titleElement.setText("Head First 设计模式");

   /** 加入owner节点 */
   Element ownerElement = booksElement.addElement("owner");
   ownerElement.setText("vcom");
   return document;
}

/**
* <b>方法描述</b>： 格式化xml输出格式，指定输出文件名、采用编码格式
* <p>
* <b>方法流程</b>：
* <p>
*
* @param document
* @param filename
*            文件名
* @param encoding
*            编码格式
* @throws Exception
*/
private void formatXMLOutput(Document document, String filename) throws Exception {
   OutputFormat format = OutputFormat.createPrettyPrint();
   // 默认为Utf-8 编码，可以根据需要改变编码格式
   // format.setEncoding("GBK");
   /** 将document中的内容写入文件中 */
   XMLWriter writer = new XMLWriter(new FileOutputStream(new File(filename)), format);
   writer.write(document);
   writer.close();
}

/**
* 修改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 java.io.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项内容改为"zzvcom"
    * 并在owner节点中加入date节点,date节点的内容为2009-10-22,还为date节点添加一个属性type
    */
    list = document.selectNodes("/books/owner");
    iter = list.iterator();
    if (iter.hasNext()) {
     Element ownerElement = (Element) iter.next();
     ownerElement.setText("测试修改");
     Element dateElement = ownerElement.addElement("date");
     dateElement.setText("2009-10-22");
     dateElement.addAttribute("type", "日期");
    }

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

    formatXMLOutput(document, newFilename);
    /** 执行成功,需返回1 */
    returnValue = 1;
   } catch (Exception ex) {
    ex.printStackTrace();
   }
   return returnValue;
}

/**
* <b>方法描述</b>： 得到xml字符串中owner节点的值
* <p>
* <b>方法流程</b>：
* <p>
*
* @param xmlStr
* @return
*/
public String getOwnerValue(String xmlStr) {
   Document document;
   try {
    document = DocumentHelper.parseText(xmlStr);
    Element rootElement = document.getRootElement();
    return rootElement.element("owner").getText();
   } catch (DocumentException e) {
    e.printStackTrace();
   }
   return null;
}

/**
* <b>方法描述</b>： 得到xml字符串中book节点的值的集合
* <p>
* <b>方法流程</b>：
* <p>
*
* @param xmlStr
* @return
*/
public List getBookListValue(String xmlStr) {
   Document document;
   List list = new ArrayList();
   try {
    document = DocumentHelper.parseText(xmlStr);
    Element rootElement = document.getRootElement();
    List<Element> elementList = rootElement.elements("book");
    for (Element element : elementList) {
     list.add(element.elementText("title"));
    }
    return list;
   } catch (DocumentException e) {
    e.printStackTrace();
   }
   return null;
}

public static void main(String[] args) {
   HandleXmlSample handleXml = new HandleXmlSample();

   // 创建xml字符串
   String xmlStr = handleXml.createXMLStr();
   // 得到xml字符串中唯一节点"owner"属性的值
   String ownerValue = handleXml.getOwnerValue(xmlStr);
   System.out.println("book拥有者：" + ownerValue);
   // 得到xml字符串中所有"book"下的"title"的值
   List list = handleXml.getBookListValue(xmlStr);
   for (Iterator it = list.iterator(); it.hasNext();) {
    System.out.println("book名称: " + it.next());
   }
 
   // 生成xml文件的测试代码
   handleXml.createXMLFile("测试.xml");
   // 修改生成新xml（注意:此方法内部调用的selectNodes方法需要jaxen-1.1-beta-6.jar的依赖包）
   handleXml.modiXMLFile("测试.xml", "测试修改后.xml");

}

}