XMLUtils

package com.mypack;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.net.URL;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.Element;
import org.dom4j.dom.DOMElement;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;

/**
 * 
 * <ul>
 * xml工具类
 * </ul>
 * 
 * @author helena
 * 
 */
public final class XMLUtils {
	/**
	 * 
	 * <li>获取文档</li>
	 * 
	 * @param xmlFilePath
	 * @return
	 */
	public static Document getDocument(String xmlFilePath) {
		if (null == xmlFilePath || xmlFilePath.equals("")) {
			throw new RuntimeException("parameter is not null!");
		}
		Document document = null;
		try {
			SAXReader reader = new SAXReader();
			document = reader.read(new File(xmlFilePath));
		} catch (DocumentException e) {

			throw new RuntimeException(e.getMessage(), e);
		}

		return document;
	}

	/**
	 * 
	 * <li></li>
	 * 
	 * @param url
	 * @return
	 */
	public static Document getDocument(URL url) {
		if (null == url) {
			throw new RuntimeException("parameter is not null");
		}
		Document document = null;
		try {
			SAXReader reader = new SAXReader();
			document = reader.read(url);
		} catch (DocumentException e) {

			throw new RuntimeException(e.getMessage(), e);
		}

		return document;
	}

	/**
	 * 
	 * <li></li>
	 * 
	 * @param document
	 * @return
	 */
	public static Element getRootElement(Document document) {
		if (null != document) {
			return document.getRootElement();
		}
		return null;
	}

	/**
	 * 
	 * <li></li>
	 * 
	 * @param elementName
	 * @return
	 */
	public static Element addElement(String elementName) {
		if (elementName == null || elementName.equals("")) {
			throw new RuntimeException("parameter is not null!");
		}
		return new DOMElement(elementName);
	}

	/**
	 * 
	 * <li></li>
	 * 
	 * @param element
	 * @param elementName
	 * @param attributeName
	 * @param attributeValue
	 */
	public static void addElement(Element element, String elementName,
			String attributeName, String attributeValue) {
		if (null == element) {
			throw new RuntimeException("parameter element is not null!");
		}

		if ((attributeName == null || attributeName.equals(""))
				|| (attributeName == null || attributeName.equals(""))) {
			return;
		}

		String[] arrAttributeName = attributeName.split("[,;]");
		String[] arrAttributeValue = attributeValue.split("[,;]");

		if (arrAttributeName.length != arrAttributeValue.length) {
			throw new RuntimeException("parameter is not correct!");
		}
		Element tempElement = new DOMElement(elementName);

		for (int i = 0; i < arrAttributeName.length; i++) {
			tempElement.addAttribute(arrAttributeName[i], arrAttributeValue[i]);

		}

		element.add(tempElement);
	}

	/**
	 * 
	 * <li></li>
	 * 
	 * @param document
	 * @param xmlFilePath
	 */
	public static void saveXML(Document document, String xmlFilePath) {
		XMLWriter out = null;
		try {
			OutputFormat outputFormat = OutputFormat.createPrettyPrint();
			outputFormat.setEncoding("GBK");
			out = new XMLWriter(new FileWriter(xmlFilePath), outputFormat);
			out.write(document);
		} catch (UnsupportedEncodingException e) {
			throw new RuntimeException(e.getMessage(), e);
		} catch (IOException e) {
			throw new RuntimeException(e.getMessage(), e);
		} finally {
			if (null != out) {
				try {
					out.close();
				} catch (IOException e) {
					throw new RuntimeException(e.getMessage(), e);
				}
			}
		}

	}

}