XML讀寫工具

import java.io.ByteArrayOutputStream; import java.io.File; import java.io.FileOutputStream; import java.io.InputStream; import java.util.List;   import javax.xml.transform.Transformer; import javax.xml.transform.TransformerFactory; import javax.xml.transform.stream.StreamSource;   import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.dom4j.Attribute; import org.dom4j.Document; import org.dom4j.DocumentHelper; import org.dom4j.Element; import org.dom4j.io.DocumentResult; import org.dom4j.io.DocumentSource; import org.dom4j.io.OutputFormat; import org.dom4j.io.SAXReader; import org.dom4j.io.XMLWriter;   /**  * XML讀寫工具  * @author csx  *  */ public class XmlUtil {                   private static final Log logger=LogFactory.getLog(XmlUtil.class);          public static final String CONFIG_FILE_NAME = "hurong.xml";// 系統配置文件名稱                            public static Document getSystemConfigXML(){                                       try{                       Document document = null;                       File configFile = null;                       String configFilePath = Thread.currentThread().getContextClassLoader().getResource("").toURI().getPath()+"hurong.xml";                       configFile = new File(configFilePath);                       SAXReader saxReader = new SAXReader();                       document = saxReader.read(configFile);                       return document;}                    catch (Exception e) {                             e.printStackTrace();                             return null;                    }          }          /**           *           * 獲取第三方短信、身份證驗證接口           *           *           * **/          public  static Document getThridInterface(){                    try{                             Document document = null;                             File configFile = null;                             String configFilePath = Thread.currentThread().getContextClassLoader().getResource("").toURI().getPath()+"thridInterface.xml";                             configFile = new File(configFilePath);                             SAXReader saxReader = new SAXReader();                             document = saxReader.read(configFile);                             return document;}                    catch (Exception e) {                             e.printStackTrace();                             return null;                    }          }                   /**           * 把Document對象轉成XML String           * @param document           * @return           */          public static String docToString(Document document) {                    String s = "";                    try {                             ByteArrayOutputStream out = new ByteArrayOutputStream();                               OutputFormat format = new OutputFormat("  ", true, "UTF-8");                             XMLWriter writer = new XMLWriter(out, format);                             writer.write(document);                             s = out.toString("UTF-8");                    } catch (Exception ex) {                             logger.error("docToString error:"+ex.getMessage());                    }                    return s;          }          /**           * 把XML String轉成Document對象           * @param s           * @return           */          public static Document stringToDocument(String s) {                    Document doc = null;                    try {                             doc = DocumentHelper.parseText(s);                    } catch (Exception ex) {                             logger.error("stringToDocument error:"+ex.getMessage());                    }                    return doc;          }                   /**           * 把Document對象轉成XML對象           * @param document           * @param filename           * @return           */          public static boolean docToXmlFile(Document document, String filename) {                    boolean flag = true;                    try {                             OutputFormat format = OutputFormat.createPrettyPrint();                             format.setEncoding("UTF-8");                             XMLWriter writer = new XMLWriter(                                                 new FileOutputStream(new File(filename)), format);                             writer.write(document);                             writer.close();                    } catch (Exception ex) {                             flag = false;                             logger.error("docToXmlFile error:"+ex.getMessage());                    }                    return flag;          }          /**           * 把String XML轉成XML文件           * @param str           * @param filename           * @return           */          public static boolean stringToXmlFile(String str, String filename) {                    boolean flag = true;                    try {                             Document doc = DocumentHelper.parseText(str);                             flag = docToXmlFile(doc, filename);                    } catch (Exception ex) {                             flag = false;                             logger.error("stringToXmlFile error:"+ex.getMessage());                    }                    return flag;          }          /**           * 加載一個XML文件轉成Document對象           * @param filename           * @return           */          public static Document load(String filename) {                    return load(new File(filename));          }                   public static Document load(File file){                    Document document = null;                    try {                             SAXReader saxReader = new SAXReader();                             saxReader.setEncoding("UTF-8");                             document = saxReader.read(file);                    } catch (Exception ex) {                             logger.error("load XML File error:"+ex.getMessage());                    }                    return document;          }                   /**           * 加載一個XML文件轉成Document對象           * @param filename           * @return           */          public static Document load(String filename,String encode) {                    Document document = null;                    try {                             SAXReader saxReader = new SAXReader();                             saxReader.setEncoding("encode");                             document = saxReader.read(new File(filename));                    } catch (Exception ex) {                             logger.error("load XML File error:"+ex.getMessage());                    }                    return document;          }          /**           * 經過流加載一個XML文檔對象           * @param is           * @return           */          public static Document load(InputStream is){                    Document document = null;                    try {                             SAXReader saxReader = new SAXReader();                             //System.out.println("code:" + System.getProperty("file.encoding"));                             saxReader.setEncoding("UTF-8");                             document = saxReader.read(is);                    } catch (Exception ex) {                             logger.error("load XML File error:"+ex.getMessage());                    }                    return document;          }                     /**           * 經過流加載一個XML文檔對象           * @param is           * @return           */          public static Document load(InputStream is,String encode){                    Document document = null;                    try {                             SAXReader saxReader = new SAXReader();                             saxReader.setEncoding(encode);                             document = saxReader.read(is);                    } catch (Exception ex) {                             logger.error("load XML File error:"+ex.getMessage());                    }                    return document;          }                   public static Document styleDocument(                  Document document,                  String stylesheet              ) throws Exception {                  // load the transformer using JAXP                  TransformerFactory factory = TransformerFactory.newInstance();                  Transformer transformer = factory.newTransformer(                      new StreamSource( stylesheet )                  );                  // now lets style the given document                  DocumentSource source = new DocumentSource( document );                  DocumentResult result = new DocumentResult();                  transformer.transform( source, result );                  // return the transformed document                  Document transformedDoc = result.getDocument();                  return transformedDoc;              }                       public static void main(String[]args){                    String filePath="L:/devtools/workspace/eoffice/web/js/menu.xml";                    String style="L:/devtools/workspace/eoffice/web/js/menu-public.xsl";                    Document doc=XmlUtil.load(filePath);                    try{                             Document another=styleDocument(doc,style);                             System.out.println("xml:" + another.asXML());                             //Set idSet=new HashSet();                              Document publicDoc=another;                              Element rootEl=publicDoc.getRootElement();                              List idNodes=rootEl.selectNodes("/Menus//*");                              System.out.println("size:" + idNodes.size());                              for(int i=0;i<idNodes.size();i++){                                       Element el=(Element)idNodes.get(i);                                       Attribute attr= el.attribute("id");                                       if(attr!=null){                                                System.out.println("attr:" + attr.getValue());                                                 //idSet.add(attr.getValue());                                       }                              }                         }catch(Exception ex){                             ex.printStackTrace();                    }                 } }