package com.hengtian.util;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
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 XmlManager {
public static void main(String[] args) {
XmlManager xmlManager = new XmlManager();
// 初始化xml文档
Document doc = null;
// 通过dom4j方法创建xml
String[] arr = new String[]{"1","2","3","4","5","6","7","8","9"};
doc = xmlManager.createXml(arr);
// XML字符串
// String strXMl = "<?xml version=\"1.0\" encoding=\"UTF-8\"?>"
// + "<?xml-stylesheet type=\"text/xsl\" href=\"students.xsl\"?>"
// + "<students><!--Students Table--> <student stu=\"001\">"
// + "<name>张三</name><age>18</age></student><student stu=\"002\">"
// + "<name>李四</name><age>19</age></student></students>";
// 通过字符串创建xml
// doc = xmlManager.createDocumentByString(strXMl);
// XMl输出路径
String outputPath = "xml/Students.xml";
// 输出xml
// xmlManager.saveDocument(doc, outputPath);
// xml输入路径
// String inputPath = "xml/Students.xml";
// 根据xml路径更改XML
//xmlManager.ModifyXml(inputPath);
// 根据xml路径获取doc
//doc = xmlManager.getDocument(inputPath);
// 遍历XML
// xmlManager.traversalDocumentByElementIterator(doc);
// xmlManager.traversalDocumentByVisitor(doc);
//xmlManager.traversalDocumentByElements(doc);
//xmlManager.traversalDocumentByselectNodes(doc, "/Students/student/name");
}
/**
* 获取XML文件
* @param inputPath
* @return
*/
public Document getDocument(String inputPath) {
// 输入文件
File inputXml = new File(inputPath);
if(!inputXml.exists()){//判断文件是否存在
try {
inputXml.createNewFile(); //创建文件
} catch (IOException e) {
e.printStackTrace();
}
}
SAXReader saxReader = new SAXReader();
Document document = null;
try {
document = saxReader.read(inputXml);
} catch (DocumentException e) {
e.printStackTrace();
}
return document;
}
/**
* 通过Dom4j方法创建xml文档
*
* @return
*/
public Document createXml(String[] arr) {
Document doc = DocumentHelper.createDocument();
// 创建ProcessingInstruction
Map<String, String> inMap = new HashMap<String, String>();
// inMap.put("type", "text/xsl");
// inMap.put("href", "students.xsl");
//doc.addProcessingInstruction("xml-stylesheet", inMap);
// 增加根节点
Element rssSourcesElement = doc.addElement("RSSsources");
// 增加注释
// studentsElement.addComment("Students Table");
for(String str : arr){
Element rssElement = rssSourcesElement.addElement("rssSource");
rssElement.addAttribute("sourceId", str);
}
// // 增加子节点
// Element stuElement = studentsElement.addElement("student");
// // 增加属性
// stuElement.addAttribute("stu", "001");
// // 增加名称节点
// Element nameElement = stuElement.addElement("name");
// // 设置名称节点的值
// nameElement.setText("张三");
// // 增加年龄节点
// Element ageElement = stuElement.addElement("age");
// // 设置年龄节点的值
// ageElement.setText("18");
//
// // 同上
// Element anotherStuElement = studentsElement.addElement("student");
// anotherStuElement.addAttribute("stu", "002");
// Element anotherNameElement = anotherStuElement.addElement("name");
// anotherNameElement.setText("李四");
// Element anotherAgeElement = anotherStuElement.addElement("age");
// anotherAgeElement.setText("19");
return doc;
}
public Document createXml(List<Long> arr) {
Document doc = DocumentHelper.createDocument();
Map<String, String> inMap = new HashMap<String, String>();
// 增加根节点
Element rssSourcesElement = doc.addElement("Urls");
for(Long str : arr){
Element rssElement = rssSourcesElement.addElement("url");
rssElement.addAttribute("CRC32", str.toString());
}
return doc;
}
/**
* 通过字符串创建xml文档
* @param xmlStr
* @return
*/
public Document createDocumentByString(String xmlStr) {
Document doc = null;
try {
// 通过字符串转换直接构建xml文档
doc = DocumentHelper.parseText(xmlStr);
} catch (DocumentException e) {
e.printStackTrace();
}
return doc;
}
/**
* 修改xml
*
* @param inputXmlPath
*/
public void ModifyXml(String inputXmlPath) {
// 获取文件
File inputXml = new File(inputXmlPath);
try {
SAXReader saxReader = new SAXReader();
// 创建document
Document doc = saxReader.read(inputXml);
// 读取Students/student下所有具有属性stu的元素
List list = doc.selectNodes("/Students/student/@stu");
Iterator iter = list.iterator();
while (iter.hasNext()) {
Attribute attribute = (Attribute) iter.next();
if (attribute.getValue().equals("001"))
attribute.setValue("0001");
}
list = doc.selectNodes("/Students/student");
iter = list.iterator();
while (iter.hasNext()) {
Element element = (Element) iter.next();
Iterator iterator = element.elementIterator("name");
while (iterator.hasNext()) {
Element nameElement = (Element) iterator.next();
if (nameElement.getText().equals("张三"))
nameElement.setText("王五");
}
}
String outputPath = "xml/Students-Modified.xml";
saveDocument(doc, outputPath);
}
catch (DocumentException e) {
System.out.println(e.getMessage());
}
}
/**
* 将文档输出到文件保存，可指定格式化输出,可指定字符编码。
*
* @param document
* @param outputFile
*/
public void saveDocument(Document doc, String outputPath) {
// 输出文件
File outputFile = new File(outputPath);
try {
// 美化格式
OutputFormat format = OutputFormat.createPrettyPrint();
// 指定XML编码,不指定的话，默认为UTF-8
format.setEncoding("UTF-8");
XMLWriter output = new XMLWriter(new FileWriter(outputFile), format);
output.write(doc);
output.flush();
output.close();
} catch (IOException e) {
System.out.println(e.getMessage());
}
}
/**
* 普通方法遍历xml
*
* @param doc
*/
public void traversalDocumentByElementIterator(Document doc) {
// 获取根节点
Element root = doc.getRootElement();
// 枚举根节点下所有子节点
for (Iterator ie = root.elementIterator(); ie.hasNext();) {
System.out.println("======");
Element element = (Element) ie.next();
System.out.println(element.getName());
// 枚举属性
for (Iterator ia = element.attributeIterator(); ia.hasNext();) {
Attribute attribute = (Attribute) ia.next();
System.out.println(attribute.getName() + ":"
+ attribute.getData());
}
// 枚举当前节点下所有子节点
for (Iterator ieson = element.elementIterator(); ieson.hasNext();) {
Element elementSon = (Element) ieson.next();
System.out.println(elementSon.getName() + ":"
+ elementSon.getText());
}
}
}
/**
* 使用elements方法进行xml的读取，相当于条件查询，可以根据不同的节点，利用for循环查询该节点下所有的数据。
*
* @throws DocumentException
*/
public static void traversalDocumentByElements(Document doc) {
// 获取根节点
Element root = doc.getRootElement();
// 根据根节点，将根节点下 student中的所有数据放到list容器中。
List list = root.elements("student");
// 这种遍历方式，是jdk1.5以上的版本支持的遍历方式,嘿嘿试试
for (Object obj : list) {
Element el = (Element) obj;
System.out.println("----------"+el.getName()+"-----------");
// 获取name节点下所有的内容，存入listName容器中
List listName = el.elements("name");
// 获取age节点下所有的内容，存入age容器中
List listAge = el.elements("age");
for (int i=0;i<listName.size();i++) {
Element elname = (Element) listName.get(i);
// 获取name节点下的数据。
System.out.println(elname.getName() + ": " + elname.getText());
Element elage = (Element) listAge.get(i);
// 获取age节点下的数据。
System.out.println(elage.getName() + ": " + elage.getText());
}
}
}
/**
* 使用selectNodes读取xml文件
*
* @param args
* @throws DocumentException
*/
public static void traversalDocumentByselectNodes(Document doc,
String elementpath) {
// 使用selectNodes获取所要查询xml的节点。
List list = doc.selectNodes(elementpath);
// 遍历节点，获取节点内数据。
for (Iterator ie = list.iterator(); ie.hasNext();) {
Element el = (Element) ie.next();
System.out.println(el.getName() + ": " + el.getText());
}
}
}
