DOM4j 拆分XML

目的

本文例子主要是将XML文件按某个节点拆分。至于DOM4j和SAX之间的区别也非常明显，在此我就不用赘述。我们通过一个实例来理解一下DOM4j的拆分功能。

    package shuai.study.dom4j.demo;  
      
    import java.io.File;  
    import java.io.FileInputStream;  
    import java.io.FileNotFoundException;  
    import java.io.FileOutputStream;  
    import java.io.IOException;  
    import java.io.OutputStream;  
    import java.io.UnsupportedEncodingException;  
    import java.util.ArrayList;  
    import java.util.Iterator;  
    import java.util.List;  
      
    import org.apache.commons.io.FileUtils;  
    import org.apache.commons.io.FilenameUtils;  
    import org.apache.commons.io.IOUtils;  
    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;  
      
    /** 
     * @author shengshu 
     *  
     */  
    public class Dom4jHandler {  
        private static volatile int pageNumber = -1;  
      
        private Document getDocument(File inputFile) {  
            SAXReader saxReader = new SAXReader();  
            Document document = null;  
      
            try {  
                document = saxReader.read(new FileInputStream(inputFile));  
            } catch (FileNotFoundException | DocumentException e) {  
                e.printStackTrace();  
            }  
      
            return document;  
        }  
      
        private Element getRootElement(Document document) {  
            return document.getRootElement();  
        }  
      
        public List<File> splitXml(File inputFile, String outputFileDirectory, String splitNodeString, int nodeLimitCount) {  
            Document document = this.getDocument(inputFile);  
            Element rootElement = this.getRootElement(document);  
      
            if (document == null || rootElement == null) {  
                throw new NullPointerException();  
            }  
      
            Document fragmentDocument = null;  
      
            List<File> exportFileList = new ArrayList<File>();  
            File exportFile = null;  
      
            Element tempRootElement = null;  
            int splitNodeIndex = 0;  
      
            Iterator<?> iterator = rootElement.elementIterator(splitNodeString);  
            while (iterator.hasNext()) {  
                if (splitNodeIndex == nodeLimitCount || splitNodeIndex == 0) {  
                    if (splitNodeIndex == nodeLimitCount) {  
                        exportFile = this.saveXml(fragmentDocument, this.getOutputFile(outputFileDirectory, inputFile, ++pageNumber));  
      
                        if (exportFile != null) {  
                            exportFileList.add(exportFile);  
                        }  
      
                        splitNodeIndex = 0;  
                    }  
      
                    fragmentDocument = DocumentHelper.createDocument();  
      
                    tempRootElement = fragmentDocument.addElement(rootElement.getName());  
                    tempRootElement.addAttribute("xmlns:xsi", "http://www.w3.org/2001/XMLSchema-instance");  
      
                    List<?> attributeList = rootElement.attributes();  
                    Iterator<?> attributeIterator = attributeList.iterator();  
                    while (attributeIterator.hasNext()) {  
                        Attribute attribute = (Attribute) attributeIterator.next();  
                        tempRootElement.addAttribute(attribute.getName(), attribute.getValue());  
                    }  
                }  
      
                Element splitElement = (Element) iterator.next();  
                tempRootElement.add((Element) splitElement.clone());  
                splitNodeIndex++;  
            }  
      
            if (fragmentDocument != null) {  
                exportFile = this.saveXml(fragmentDocument, this.getOutputFile(outputFileDirectory, inputFile, ++pageNumber));  
      
                if (exportFile != null) {  
                    exportFileList.add(exportFile);  
                }  
            }  
      
            return exportFileList;  
        }  
      
        private File saveXml(Document document, File outputFile) {  
            OutputStream fileOutputStream = null;  
            XMLWriter xmlWriter = null;  
      
            OutputFormat format = OutputFormat.createPrettyPrint();  
      
            try {  
                FileUtils.touch(outputFile);  
                fileOutputStream = new FileOutputStream(outputFile);  
      
                xmlWriter = new XMLWriter(fileOutputStream, format);  
                xmlWriter.write(document);  
                xmlWriter.flush();  
            } catch (FileNotFoundException fnfe) {  
                fnfe.printStackTrace();  
            } catch (UnsupportedEncodingException uee) {  
                uee.printStackTrace();  
            } catch (IOException ioe) {  
                ioe.printStackTrace();  
            } finally {  
                if (xmlWriter != null) {  
                    try {  
                        xmlWriter.close();  
                    } catch (IOException ioe) {  
                        ioe.printStackTrace();  
                    }  
                }  
      
                IOUtils.closeQuietly(fileOutputStream);  
            }  
      
            return outputFile;  
        }  
      
        private File getOutputFile(String outputFileDirectory, File inputFile, int pageNumber) {  
            String fileName = inputFile.getName();  
            String fileBaseName = FilenameUtils.getBaseName(fileName);  
            String fileNameExtension = FilenameUtils.getExtension(fileName);  
      
            String outputFilePath = outputFileDirectory + File.separator + fileBaseName + "-" + this.pagingFormat(pageNumber) + "." + fileNameExtension;  
            File outputFile = new File(outputFilePath);  
      
            return outputFile;  
        }  
      
        private synchronized String pagingFormat(int pageNumber) {  
            String pageNumberStr = "P00";  
      
            if (pageNumber < 0) {  
                throw new NumberFormatException("The page number should not be negative");  
            }  
      
            if (pageNumber < 10) {  
                pageNumberStr = String.format("P0%d", pageNumber);  
            } else {  
                pageNumberStr = String.format("P%d", pageNumber);  
            }  
      
            return pageNumberStr;  
        }  
    }  

    package shuai.study.dom4j.demo;  
      
    import java.io.File;  
      
    /** 
     * @author shengshu 
     *  
     */  
    public class Dom4jDemo {  
        public static void main(String[] args) {  
            String inputFilePath = Dom4jHandler.class.getResource("/file/input/company.xml").getPath();  
            File inputFile = new File(inputFilePath);  
      
            String outputFileDirectory = Dom4jHandler.class.getResource("/file/output").getPath();  
      
            String splitNodeString = "Employee";  
            int nodeLimitCount = 1;  
      
            Dom4jHandler dom4jHandler = new Dom4jHandler();  
            dom4jHandler.splitXml(inputFile, outputFileDirectory, splitNodeString, nodeLimitCount);  
        }  
    }  

    <?xml version = "1.0" encoding="UTF-8"?>  
      
    <Company xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"  
        xsi:noNamespaceSchemaLocation="CompanyFileFormat.xsd">  
        <Message>XXX Company Employee Message</Message>  
      
        <Employee ID="37">  
            <Name>  
                <FirstName>Zhou</FirstName>  
                <LastName>Shengshuai</LastName>  
            </Name>  
      
            <Sex>Male</Sex>  
      
            <Address>  
                <Country>China</Country>  
                <Province>ShanDong</Province>  
                <City>LinYi</City>  
                <Village>FengHuangYu</Village>  
      
                <Contact>  
                    <Mobile>18108***778</Mobile>  
                    <Mail>[email protected]</Mail>  
                    <QQ>254392398</QQ>  
                    <Postcode>276422</Postcode>  
                </Contact>  
            </Address>  
      
            <Profession>Software</Profession>  
        </Employee>  
      
        <Employee ID="66">  
            <Name>  
                <FirstName>Wang</FirstName>  
                <LastName>Eric</LastName>  
            </Name>  
      
            <Sex>Male</Sex>  
      
            <Address>  
                <Country>China</Country>  
                <Province>HeBei</Province>  
                <City>QinHuangDao</City>  
                <Village>hhh</Village>  
      
                <Contact>  
                    <Mobile>150*****955</Mobile>  
                    <Mail>[email protected]</Mail>  
                    <QQ>666666666</QQ>  
                    <Postcode>111666</Postcode>  
                </Contact>  
            </Address>  
      
            <Profession>Software</Profession>  
        </Employee>  
      
        <Employee ID="99">  
            <Name>  
                <FirstName>Shi</FirstName>  
                <LastName>Stone</LastName>  
            </Name>  
      
            <Sex>Male</Sex>  
      
            <Address>  
                <Country>China</Country>  
                <Province>HeNan</Province>  
                <City>PingDingShan</City>  
                <Village>nnn</Village>  
      
                <Contact>  
                    <Mobile>186*****015</Mobile>  
                    <Mail>[email protected]</Mail>  
                    <QQ>999999999</QQ>  
                    <Postcode>111999</Postcode>  
                </Contact>  
            </Address>  
      
            <Profession>Software</Profession>  
        </Employee>  
    </Company>