前言
最近有一个解析大型xml的需求,xml大小7M,其中xml结构非常复杂,元素各种嵌套
不乏有元素下对象,元素下集合,集合下对象,集合下集合,兄弟节点,元素下对象下集合,同一元素下不同对象和集合...很复杂
注意:本方式无法解析递归元素
简介
本文使用了SAX解析XML,SAX对内存比较友好,但是对于编码比较复杂,但是本篇代码量不是很多,主要都集中在一个继承了DefaultHandler的处理类,该处理类中主要使用了两个方法,分别是startElement和endElement。
阅读必读
比较难理解的是xml节点复杂关系的呈现,大概设计思路是定义三个map和一个int类型的下标值,三个map的作用是第1个map存储结果集,第2个map记录元素下标对应的key,第3map个记录下标key所属的元素对象。当startElement开始读取节点元素时将该元素的节点名称存储至第2个map并将该对象以该元素的节点名称存储至第三个map中,然后将int类型的下标值手动+1,最后当元素节点读取结束时开始组装格式,下面是代码呈现
package com.syasuo.xml;
import com.alibaba.fastjson.JSON;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.CollectionUtils;
import org.xml.sax.Attributes;
import org.xml.sax.SAXException;
import org.xml.sax.helpers.DefaultHandler;
import java.util.*;
/**
* @author SYASUO
* @date 2023/6/19 13:46
*/
@Slf4j
public class MyDefaultHandler extends DefaultHandler {
//存储结果集
private Map<String,Object> resultMap = new HashMap<>();
//用下标标记key
private Map<Integer,String> indexKeyMap = new HashMap<>();
//下标标记的所属key的对象
private Map<String,Map<String,Object>> indexKeyMapObj = new HashMap<>();
//记录元素节点对应的下标,初始给0
private int index = 0;
@Override
public void startDocument() throws SAXException {
log.info("------------------------------XML读取开始------------------------------");
}
/**
* 开始读取XML元素,每次记录下标对应的kay和该key对应的对象
*/
@Override
public void startElement(String uri, String localName, String qName, Attributes attributes) throws SAXException {
Map<String,Object> node = new HashMap<>();
for (int i = 0; i < attributes.getLength(); i++) {
node.put(attributes.getQName(i),attributes.getValue(i));
}
indexKeyMap.put(index,qName);
indexKeyMapObj.put(qName,node);
index++;
}
/**
* 元素节点读取结束时,由于index在startElement中执行了+1操作,
* 因此获取对应节点的父级节点时需要先执行index-1操作得到当前节点下标,在当前节点下标位再次执行-1操作得到父级节点
* 取到父级节点时需要判断父级包不包含该节点,如果包含就说明该节点是一个List集合,否则就作为一个单独的对象放入父级对象中
* 最后给resultMap结果集对象赋值即可
*/
@Override
public void endElement(String uri, String localName, String qName) throws SAXException {
Map<String,Object> curObj = indexKeyMapObj.get(qName);
int j = --index;
if(j > 0){
j--;
String qname = indexKeyMap.get(j);
Map<String,Object> parent = indexKeyMapObj.get(qname);
if(parent.containsKey(qName)){
List<Map<String,Object>> list = new ArrayList<>();
//System.out.println(parent.get(qName).getClass().getTypeName());
if(parent.get(qName).getClass().getSimpleName().equals("HashMap")){
list.add((Map<String, Object>) parent.get(qName));
}else{
list.addAll((Collection<? extends Map<String, Object>>) parent.get(qName));
}
//将当前节点对象加入当前对象集合
list.add(curObj);
parent.put(qName,list);
}else{
parent.put(qName,curObj);
}
resultMap = parent;
}
}
@Override
public void characters(char[] ch, int start, int length) throws SAXException {
/*
String element = new String(ch,start,length);
log.info("该方法只有在【<ele>标签值</ele>】这种类型的标签下才能有效获取");
*/
}
/*
* 结束文档时调用
*/
@Override
public void endDocument() throws SAXException {
log.info("------------------------------XML读取结束------------------------------");
}
public Map<String,Object> getResultMap(){
return resultMap;
}
}
测试Main
注意:CprjInfo是xml对应格式的JavaBean,XML如何转JavaBean,可以看我这一篇
使用XJC将XML转换成JavaBean遇到的坑_辛丑年正月十五的博客-CSDN博客
package com.syasuo.xml;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.syasuo.entity.CprjInfo;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;
import org.xml.sax.XMLReader;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;
import java.io.File;
import java.io.IOException;
import java.util.Map;
/**
* @author SYASUO
* @date 2023/6/19 13:41
*/
public class Test {
public static void main(String[] args) throws ParserConfigurationException, SAXException, IOException {
Long startTime = System.currentTimeMillis();
SAXParser parser = SAXParserFactory.newInstance().newSAXParser();
MyDefaultHandler myDefaultHandler = new MyDefaultHandler();
parser.parse(new File("D://root.xml"), myDefaultHandler);
Map<String,Object> resultMap = myDefaultHandler.getResultMap();
//System.out.println(JSON.toJSONString(resultMap));
CprjInfo cprjInfo = JSONObject.parseObject(JSON.toJSONString(resultMap),CprjInfo.class);
System.out.println(JSON.toJSONString(cprjInfo));
System.out.println(System.currentTimeMillis()-startTime);
}
}
成功的截图,可以看见进度条多么小