Java操作XML的工具类

 

 

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.StringWriter;
import java.io.Writer;

import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.stream.StreamResult;
import javax.xml.transform.stream.StreamSource;

import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.ProcessingInstruction;
import org.dom4j.io.DocumentSource;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.SAXReader;
import org.dom4j.io.XMLWriter;

public class XMLUtil {
    /**
     * 以编码为UTF-8的方式返回xml
     * @param doc Document
     * @return String
     */
    public static String toString(Document doc) {
        return toString(doc, "UTF-8");
    }

    /**
     * 以指定编码格式返回xml
     * @param doc  Document
     * @param encoding String
     * @return String
     */
    public static String toString(Document doc, String encoding) {
        if (null != doc) {
            OutputFormat outputFormat = new OutputFormat();
            outputFormat.setEncoding(encoding);
            StringWriter stringWriter = new StringWriter();
            XMLWriter xmlWriter = new XMLWriter(stringWriter, outputFormat);
            try {
                xmlWriter.write(doc);
                return stringWriter.toString();
            } catch (IOException ex) {
                return "";
            }
        } else {
            return "";
        }
    }
    
    private static String converPath( String path ){
        if(OSUtil.LIUNX.equals(System.getProperty("os.name"))){
            path = OSUtil.convert2linuxPath(path) ;
            System.out.println( "system is Linux , path conver to :" + path );
        }
        return path ;
    }
    
    
    /**
     * 创建xml格式的文件
     * @param doc
     * @param encoding
     * @param strFileName
     * @return
     */
    public static boolean saveXMLDocumentToFile(Document doc, String encoding, String strFileName) {
        boolean flag = false;
        // 创建路径
        strFileName = converPath(strFileName);
        String strDir = FileNameUtil.extractFilePath(strFileName);
        DirectoryUtil.forceDirectory(strDir);

        if (encoding == null || encoding.length() == 0) {
            encoding = "UTF-8";
        }
        OutputFormat outputFormat = new OutputFormat();
        outputFormat.setEncoding(encoding);
        FileOutputStream fos = null;
        XMLWriter xmlWriter = null;
        try {
            // FileWriter fileWriter = new FileWriter(strFileName);
            // XMLWriter xmlWriter = new XMLWriter(fileWriter, outputFormat);//
            // 不能解决UTF-8编码问题
            fos = new FileOutputStream(strFileName);// 可解决UTF-8编码问题
            xmlWriter = new XMLWriter(fos, outputFormat);
            xmlWriter.write(doc);
            flag = true;
        } catch (IOException e) {
            flag = false;
            System.out.println("保存xml文件出错：" + e.getMessage());
            e.printStackTrace();
        } finally {
            try {
                if (xmlWriter != null) {
                    xmlWriter.flush();
                }
                if (fos != null) {
                    fos.flush();
                }
                if (xmlWriter != null) {
                    xmlWriter.close();
                }
                if (fos != null) {
                    fos.close();
                }
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        return flag;
    }
    
    /**
     * 创建格式化过的xml格式的文件
     * @param doc
     * @param encoding
     * @param strFileName
     * @return
     */
    public static boolean saveFormatXMLDocumentToFile(Document doc, String encoding,
            String strFileName)  {
        boolean flag = false;
        
        // 创建路径
        strFileName = converPath(strFileName);
        String strDir = FileNameUtil.extractFilePath(strFileName) ;
        DirectoryUtil.forceDirectory(strDir);

        if (encoding == null || encoding.length() == 0) {
            encoding = "UTF-8";
        }
        OutputFormat outputFormat = OutputFormat.createPrettyPrint();
        outputFormat.setEncoding(encoding);
        FileOutputStream fos = null;
        XMLWriter xmlWriter  = null;
        try{
            fos = new FileOutputStream(strFileName);// 可解决UTF-8编码问题
            xmlWriter = new XMLWriter(fos, outputFormat);
            xmlWriter.write(doc);
            flag = true;
        }catch(IOException e){
            flag = false;
        }finally{
            try {
                xmlWriter.flush();
                fos.flush();
                xmlWriter.close();
                fos.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
            
        }
        return flag;
    }
    
    public static void saveXMLDocumentToOutputStream(Document doc,
            String encoding, OutputStream outputstream) throws IOException {
        if (encoding == null || encoding.length() == 0) {
            encoding = "UTF-8";
        }
        OutputFormat outputFormat = new OutputFormat();
        outputFormat.setEncoding(encoding);
        XMLWriter xmlWriter = new XMLWriter(outputstream, outputFormat);
        xmlWriter.write(doc);
        xmlWriter.close();
        outputstream.close();
    }

    public static Document loadXMLFile(String strFileName)
            throws DocumentException {
        SAXReader saxReader = new SAXReader();
        saxReader.setValidation(false);
        saxReader.setEntityResolver(new IgnoreDTDEntityResolver()); 
        return saxReader.read(new File(OSUtil.convert2linuxPath(strFileName)));
    }

    public static Document loadXMLInputstream(InputStream in){
        SAXReader reader = new SAXReader();
        try {
            return reader.read(in);
        } catch (DocumentException e) {
            return null;
        }
    }
    
    /**
     * 用于xml 与 xsl 的归并输出含处理指令的xml到out
     * 处理指令指定了浏览器渲染的时候使用的xsl文件相对路径
     * 
     * @author sun
     */
    @SuppressWarnings("unchecked")
    public static void outputXML(Document xmldoc, String xslname,
            Writer out) throws Exception {
        if (xslname != null) {
            ProcessingInstruction pi = DocumentHelper
                    .createProcessingInstruction("xml-stylesheet", "href=\""
                            + xslname + "\" type=\"text/xsl\"");
            xmldoc.content().add(0, pi);
        }
        TransformerFactory factory = TransformerFactory.newInstance();
        Transformer transformer = factory.newTransformer();
        transformer.setOutputProperty("encoding", "UTF-8");

        transformer
                .transform(new DocumentSource(xmldoc), new StreamResult(out));
        out.flush();
        out.close();
    }

    /**
     * 用于xml 与 xsl 的归并输出xml或html到out
     * 输出html时，xsl名称不能为null
     * @author
     * @throws TransformerException 
     */
    public static void transformXml(Document xmldoc, String xslname,
            Writer out) throws TransformerException{
        TransformerFactory factory = TransformerFactory.newInstance();
        Transformer transformer = xslname == null ? factory.newTransformer()
                : factory.newTransformer(new StreamSource(xslname));
        transformer.setOutputProperty("encoding", "UTF-8");

        transformer.transform(new DocumentSource(xmldoc),
                new StreamResult(out));
    }
}