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<String, List<Map<String, Object>>> map = XmlUtils.xmlToMap(doc);
System.out.println("Use time:"+(System.currentTimeMillis()-beginTime));
System.out.println(map.toString());
}
public static Map<String, List<Map<String, Object>>> xmlToMap (Document doc) {
Map<String, List<Map<String, Object>>> map = new HashMap<String, List<Map<String, Object>>>();
if(doc == null)
return map;
Element root = doc.getRootElement();
List<Map<String, Object>> elementMaps = new ArrayList();
for (Iterator iterator = root.elementIterator(); iterator.hasNext();) {
Map<String, Object> elementMap = new HashMap<String, Object>();
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
<?xml version="1.0" encoding="UTF-8"?>
<breakfast_menu>
<food>
<name>Belgian Waffles</name>
<price>$5.95</price>
<description>
two of our famous Belgian Waffles with plenty of real maple syrup
</description>
<calories>650</calories>
</food>
<food>
<name>Strawberry Belgian Waffles</name>
<price>$7.95</price>
<description>
light Belgian waffles covered with strawberries and whipped cream
</description>
<calories>900</calories>
</food>
<food>
<name>Berry-Berry Belgian Waffles</name>
<price>$8.95</price>
<description>
light Belgian waffles covered with an assortment of fresh berries and whipped cream
</description>
<calories>900</calories>
</food>
<food>
<name>French Toast</name>
<price>$4.50</price>
<description>
thick slices made from our homemade sourdough bread
</description>
<calories>600</calories>
</food>
<food>
<name>Homestyle Breakfast</name>
<price>$6.95</price>
<description>
two eggs, bacon or sausage, toast, and our ever-popular hash browns
</description>
<calories>950</calories>
</food>
</breakfast_menu>