1 packagecom.juxtapose.xml.parser;2
3 importjava.io.StringReader;4 importjava.util.ArrayList;5 importjava.util.HashMap;6 importjava.util.List;7 importjava.util.Map;8 importjava.util.concurrent.atomic.AtomicInteger;9
10 importjavax.xml.stream.XMLInputFactory;11 importjavax.xml.stream.XMLStreamConstants;12 importjavax.xml.stream.XMLStreamReader;13
14 /**
15 * xml字符串解析器实现类.
16 * xml字符串转换为Map对象.
17 * 转换后的数据类型为Map、List、String三种数据类型.
18 *19 *@authorburceliu (mailto:jxta.liu@gmail.com)20 */
21
22 public class XMLParserStaxImpl implementsIXMLParser {23
24 public static final String NODE_ELEMENT_NAME = "root";25 public static final String NODE_DEFAULT_VALUE = "";26
27 private String rootName; //根节点
28 private String defaultNullValue; //节点没有值的情况下默认值
29
30 private static XMLInputFactory factory =XMLInputFactory.newInstance();31 static{32 factory.setProperty(XMLInputFactory.IS_NAMESPACE_AWARE, Boolean.FALSE);33 factory.setProperty(XMLInputFactory.IS_COALESCING, Boolean.TRUE);34 }35
36 /*(non-Javadoc)37 * @see com.juxtapose.xml.parser.IXMLParser#parse(java.lang.String)38 */
39 public Mapparse(String xml) {40 Map map = new HashMap();41 StringReader stringReader = null;42 try{43 stringReader = newStringReader(xml);44 XMLStreamReader reader =factory.createXMLStreamReader(stringReader);45 map =parse(reader);46 }catch(Throwable t){47 throw newRuntimeException(t);48 }finally{49 if(null !=stringReader){50 try{51 stringReader.close();52 } catch(Exception e) {53 throw newRuntimeException(e);54 }55 }56 }57 returnmap;58 }59
60 /*(non-Javadoc)61 * @see com.juxtapose.xml.parser.IXMLParser#init()62 */
63 public voidinit() {64 if(this.getRootName() == null){65 this.setRootName(NODE_ELEMENT_NAME);66 this.setDefaultNullValue(NODE_DEFAULT_VALUE);67 }68 }69
70 @SuppressWarnings({ "unchecked", "rawtypes"})71 private Map parse(XMLStreamReader reader) throwsThrowable{72 Map map = new HashMap();73 Map currentMap =map;74 int event =reader.getEventType();75 List names = new ArrayList();76 NodeAmount nodeAmount = newNodeAmount();77 int taglength = 0;78 String tagName = null;79 String tagValue = this.defaultNullValue;80 while(true){81 switch(event) {82 caseXMLStreamConstants.START_DOCUMENT:83 break;84 caseXMLStreamConstants.START_ELEMENT:85 tagValue = this.defaultNullValue;86 tagName =reader.getLocalName();87 if(this.rootName.equals(tagName)){88 break;89 }90 names.add(tagName);91 taglength++;92
93 currentMap =map;94 if(taglength > 1){95 for(int i=0;i< taglength-1;i++){96 Object object =currentMap.get(names.get(i));97 if(null ==object){98 object = new HashMap();99 currentMap.put(names.get(i), object);100 currentMap = (Map)object;101 }else{102 int currentTagNameSize = nodeAmount.getSize(i + 1 + "" +names.get(i));103 if( currentTagNameSize > 1){104 if(object instanceofMap){105 List parentList = newArrayList();106 parentList.add(object);107 Map tempMap = newHashMap();108 parentList.add(tempMap);109 currentMap.put(names.get(i), parentList);110 currentMap =tempMap;111 }else if(object instanceofList){112 List parentList =(List)object;113 int parentListSize =parentList.size();114 if(parentListSize !=currentTagNameSize){115 Map tempMap = newHashMap();116 parentList.add(tempMap);117 currentMap =tempMap;118 }else{119 Map tempMap = (Map) parentList.get(parentList.size()-1);120 currentMap =tempMap;121 }122 }123 }else{124 currentMap = (Map)object;125 }126 }127 }128 }129 nodeAmount.add(names.size() +tagName);130 break;131 caseXMLStreamConstants.CHARACTERS:132 tagValue =reader.getText();133 break;134 caseXMLStreamConstants.END_ELEMENT:135 tagName =reader.getLocalName();136 if(this.rootName.equals(tagName)){137 break;138 }139
140 currentMap =map;141 if(taglength > 1){142 for(int i=0;i< taglength-1;i++){143 Object object =currentMap.get(names.get(i));144 if(null ==object){145 //nothing to do
146 }else{147 if(object instanceofList){148 List list =(List)object;149 currentMap = (Map)list.get(list.size() -1);150 }else if(object instanceofMap){151 currentMap =(Map)object;152 }153 }154 }155 }156
157 Object oldValue =currentMap.get(tagName);158 if(!currentMap.containsKey(tagName)){159 currentMap.put(tagName, tagValue);160 nodeAmount.remove(names.size() +tagName);161 }else{162 if(oldValue instanceofList){163 List list =(List)oldValue;164 if(list.size() > 0){165 Object obj = list.get(0);166 if(obj instanceofString){167 ((List)oldValue).add(tagValue);168 nodeAmount.remove(names.size() +tagName);169 }170 }171 }else if(oldValue instanceofMap){172
173 }else{174 List tmpList = newArrayList();175 currentMap.put(tagName, tmpList);176 tmpList.add(oldValue);177 tmpList.add(tagValue);178 nodeAmount.remove(names.size() +tagName);179 }180 }181
182 tagValue = this.defaultNullValue;183 names.remove(names.size()-1);184 taglength--;185 break;186 caseXMLStreamConstants.END_DOCUMENT:187 break;188 }189
190 if (!reader.hasNext()) {191 break;192 }193 event =reader.next();194 }195 returnmap;196 }197
198 publicString getRootName() {199 returnrootName;200 }201
202 public voidsetRootName(String rootName) {203 this.rootName =rootName;204 }205
206 publicString getDefaultNullValue() {207 returndefaultNullValue;208 }209
210 public voidsetDefaultNullValue(String defaultNullValue) {211 this.defaultNullValue =defaultNullValue;212 }213
214 classNodeAmount{215 private Map map =new HashMap();216
217 public voidadd(String nodeName){218 AtomicInteger integer =map.get(nodeName);219 if(null ==integer){220 integer = new AtomicInteger(0);221 map.put(nodeName, integer);222 }223 integer.incrementAndGet();224 }225
226 public voidremove(String nodeName){227 AtomicInteger integer =map.get(nodeName);228 if(null !=integer){229 integer.decrementAndGet();230 }231 }232
233 public intgetSize(String nodeName){234 AtomicInteger integer =map.get(nodeName);235 if(null ==integer){236 integer = new AtomicInteger(0);237 map.put(nodeName, integer);238 }239 returninteger.intValue();240 }241 }242
243 }244
245 /*
246 * 修改历史247 * $Log$248 */