Xml 转换成 Map

package work;

import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import org.dom4j.Document;
import org.dom4j.DocumentException;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;

public class XmlUtils {

    public static void main(String[] args) throws DocumentException, IOException {
        InputStream fis = XmlUtils.class.getClassLoader().getResourceAsStream("test.xml");
        byte[] b = new byte[fis.available()];
        fis.read(b);
        String str = new String(b);

        long beginTime = System.currentTimeMillis();

        Document doc = DocumentHelper.parseText(str);
        Map>> map = XmlUtils.xmlToMap(doc);

        System.out.println("Use time:"+(System.currentTimeMillis()-beginTime));
        System.out.println(map.toString());
    }

    public static Map>> xmlToMap (Document doc) {
        Map>> map = new HashMap>>();
        if(doc == null)
            return map;

        Element root = doc.getRootElement();

        List> elementMaps = new ArrayList();
        for (Iterator iterator = root.elementIterator(); iterator.hasNext();) {
            Map elementMap = new HashMap();
            Element e = (Element) iterator.next();
            List list = e.elements();
            if(list.size() > 0){
                elementMap.put(e.getName(), elementToMap(e));
            }else 
                elementMap.put(e.getName(), e.getText());

            elementMaps.add(elementMap);
        }

        map.put(root.getName(),elementMaps);
        return map;
    }

    public static Map elementToMap(Element e) {
        Map map = new HashMap();
        List list = e.elements();
        if (list.size() > 0) {
            for (int i = 0; i < list.size(); i++) {
                Element iter = (Element) list.get(i);

                if (iter.elements().size() > 0) {
                    manyIterToMap(map, iter);
                } else {
                    oneIterToMap(map, iter);
                } 
            } 
        } else {
            map.put(e.getName(), e.getText());
        }

        return map; 
    }

    private static void oneIterToMap(Map map, Element iter) {
        List mapList = new ArrayList();
        if (map.get(iter.getName()) != null) {
            Object obj = map.get(iter.getName());

            if(!obj.getClass().getName().equals("java.util.ArrayList")){
                mapList = new ArrayList();
                mapList.add(obj);
                mapList.add(iter.getText());
            }
            if(obj.getClass().getName().equals("java.util.ArrayList")){
                mapList = (List) obj;
                mapList.add(iter.getText());
            }

            map.put(iter.getName(), mapList);
        } else {
            map.put(iter.getName(), iter.getText());
        }
    }

    private static void manyIterToMap(Map map, Element iter) {
        Map m = elementToMap(iter);
        List mapList = new ArrayList();
        if (map.get(iter.getName()) != null){
            Object obj = map.get(iter.getName());

            if (!obj.getClass().getName().equals("java.util.ArrayList")) {
                mapList = new ArrayList();
                mapList.add(obj);
            }
            if (obj.getClass().getName().equals("java.util.ArrayList")) {
                mapList = (List) obj;
            }

            mapList.add(m);
            map.put(iter.getName(), mapList);
        } else {
            map.put(iter.getName(), m);
        }
    }
}

test.xml




Belgian Waffles
$5.95

two of our famous Belgian Waffles with plenty of real maple syrup

650


Strawberry Belgian Waffles
$7.95

light Belgian waffles covered with strawberries and whipped cream

900


Berry-Berry Belgian Waffles
$8.95

light Belgian waffles covered with an assortment of fresh berries and whipped cream

900


French Toast
$4.50

thick slices made from our homemade sourdough bread

600


Homestyle Breakfast
$6.95

two eggs, bacon or sausage, toast, and our ever-popular hash browns

950


你可能感兴趣的:(Xml 转换成 Map)