JDOM 解析 XML文件

package com.itheima.xml;

import java.io.File;
import java.io.FileWriter;
import java.util.Iterator;
import java.util.List;

import org.jdom.Attribute;
import org.jdom.Document;
import org.jdom.Element;
import org.jdom.input.SAXBuilder;
import org.jdom.output.Format;
import org.jdom.output.XMLOutputter;

public class TextJDOMparseXML {//操作使用JDOM解析XML文件的类

	/**
	 * @param args
	 */
	public static void main(String[] args) {
		// TODO Auto-generated method stub
		String fileName="F:/test1.xml";
		TextJDOMparseXML jdomXml=new TextJDOMparseXML();
		System.out.println("===============1.运用jdom创建XML文件");
		if(jdomXml.createXMLFile(fileName, "GB2312")==1)
			System.out.println("运用JDOM创建XML文件成功！");
		else
			System.out.println("运用JDOM创建XML文件失败!");
		System.out.println("===============2.运用JDOM读取XML文件");
		jdomXml.read(jdomXml.getDocument(fileName));
		System.out.println("===============3.修改XML内容，输出修改后的文档");
		jdomXml.update(fileName, "rongshu", "liushu");
		System.out.println("===============4.删除编号为002的节点,输出修改后的文件");
		jdomXml.delete(fileName, "002");

	}

	//生成一个新的Document  
	public Document createDocument(){
		Document document=new Document();
		return document;
	}
	//创建一个节点
	public Element createElement(String paramName,String paramValue){
		Element element=new Element(paramName);
		element.setText(formatTextForXML(paramValue));
		return element;
	}
	//创建 XML文件
	public int createXMLFile(String fileName,String encoding){
		int returnValue=0;
		Document document=createDocument();
		Element treesElement=createElement("trees", null);
		Element treeElement=new Element("tree");
		treeElement.setAttribute("id","001");
		treeElement.setAttribute("name","rongshu");

		Element typeElement=new Element("type");
		typeElement.setText("style");
		typeElement.setAttribute("id","xiaoshu");
		treeElement.addContent(typeElement);  //添加 子节点


		Element descElement=new Element("description");
		descElement.setText(formatTextForXML("It is a big tree"));
		treeElement.addContent(descElement);

		treesElement.addContent(treeElement);

		Element tree1Element=new Element("tree");
		tree1Element.setAttribute("id","002");
		tree1Element.setAttribute("name","yangshu");
		Element type1Element=new Element("type");
		type1Element.setText("style");
		type1Element.setAttribute("id","dashu");
		tree1Element.addContent(type1Element);
		Element desc1Element=new Element("description");
		desc1Element.setText(formatTextForXML("The tree grows taller from day to day"));
		tree1Element.addContent(desc1Element);
		treesElement.addContent(tree1Element);


		Element tree2Element=new Element("tree");
		tree2Element.setAttribute("id","003");
		tree2Element.setAttribute("name","songshu");
		Element desc2Element=new Element("description");
		desc2Element.setText(formatTextForXML("The tree is very beautiful"));
		tree2Element.addContent(desc2Element);
		treesElement.addContent(tree2Element);

		Element flowerElement=new Element("flower");
		flowerElement.setText("rose");
		treesElement.addContent(flowerElement);

		document.addContent(treesElement);
		try {
			saveFile(fileName,document,encoding);
			returnValue=1;
		} catch (Exception e) {
			// TODO: handle exception
			System.out.println("创建XML文件失败:"+e.getMessage());
		}
		return returnValue;

	}
	//读取xml
	public void read(Document document){
		try {
			Format format=Format.getPrettyFormat(); 
			format.setEncoding("GB2312");

			XMLOutputter xmlOutputter=new XMLOutputter(format);
			xmlOutputter.output(document, System.out);
		} catch (Exception e) {
			System.out.println("读取XML文件失败："+e.getMessage());
		}
	}
	//修改 xml文档节点
	public int update(String fileName,String oldName,String newName){
		int returnValue=0;
		Document document=getDocument(fileName);
		List list=document.getRootElement().getChildren();
		for (int i = 0; i < list.size(); i++) {
			Element element=(Element)list.get(i);
			Attribute attribute=element.getAttribute("name");
			if(attribute!=null){
				if(attribute.getValue().equals(oldName)){
					attribute.setValue(newName);
				}
			}
		}

		list=document.getRootElement().getChildren("flower");
		Iterator iter=list.iterator();
		if(iter.hasNext()){
			Element el=(Element)iter.next();
			el.setText("lily");
			Element date=new Element("date");
			date.setText("2009-04-10");
			date.setAttribute("type","calendar");
			el.addContent(date);
		}

		try {
			showFile(fileName,document);
			returnValue=1;
		} catch (Exception e) {
			// TODO: handle exception
			System.out.println("修改XML文件出错:"+e.getMessage());
		}

		return returnValue;


	}
	//删除XML 节点
	public boolean delete(String fileName,String id){
		boolean returnVlalue=false;
		Document doc=getDocument(fileName);
		Element root=doc.getRootElement();
		List list=root.getChildren("tree");
		for (int i = 0; i < list.size(); i++) {
			Element element=(Element)list.get(i);
			if (element.getAttribute("id").getValue().equals(id)) {
				root.removeContent(element);
			}
		}
		try {
			showFile(fileName, doc);
			returnVlalue=true;
		} catch (Exception e) {
			// TODO: handle exception
			e.printStackTrace();
		}
		return returnVlalue;

	}

	private void showFile(String fileName, Document document) {
		// TODO Auto-generated method stub
		Format format=Format.getPrettyFormat();
		format.setEncoding("GB2312");
		format.setIndent("　");
		try {
			XMLOutputter xmlOutputter=new XMLOutputter(format);
			FileWriter fileWriter=new FileWriter(fileName);
			xmlOutputter.output(document, fileWriter);
			xmlOutputter.output(document, System.out);
		} catch (Exception e) {
			System.out.println("XML文件输出到控制台出错:"+e.getMessage());
		}
	}

	private Document getDocument(String fileName) {
		// TODO Auto-generated method stub
		Document document=null;
		try {
			SAXBuilder saxBuilder=new SAXBuilder();
			document=saxBuilder.build(new File(fileName));
		} catch (Exception e) {
			// TODO: handle exception
			System.out.println("载入失败:"+e.getMessage());
			return null;
		}
		return document;
	}

	private boolean saveFile(String fileName, Document document, String encoding) {
		// TODO Auto-generated method stub
		Format format=Format.getPrettyFormat();
		if(encoding!=null){
			format.setEncoding(encoding);
		}
		format.setIndent(" "); //设置xml文件的缩进
		try {
			XMLOutputter xmlOutputter=new XMLOutputter(format);
			FileWriter fileWriter=new FileWriter(fileName);
			xmlOutputter.output(document, fileWriter);
		} catch (Exception e) {
			System.out.println("保存文件出错:"+e.getMessage());
			return false;
		}
		return true;
	}

	private String formatTextForXML(String sourceString) {
		// TODO Auto-generated method stub
		if (sourceString==null) {
			return null;
		}
		int stLen=0;
		StringBuffer reString= new StringBuffer();
		String deString="";
		stLen=sourceString.length();
		for (int i = 0; i < stLen; i++) {
			char ch=sourceString.charAt(i);
			switch (ch) {
			case '<':
				deString="<";
				break;
			case '>':
				deString=">";
				break;
			case '\"':
				deString="\"";
				break;
			case '&':
				deString="&";
				break;
			case 13:
				deString="\n";
				break;
			default:
				deString=""+ch;
				break;
			}
			reString.append(deString);
		}
		return reString.toString();
	}	


}