mirror of https://github.com/halo-dev/halo
Fixed XmlTransferMapUtils.
ruibaby
parent
4b0b9d5245
commit
c52e295f9a
|
|
@ -1,101 +1,105 @@
|
||||||
package xyz.guqing.transfer.processor;
|
package run.halo.app.utils;
|
||||||
|
|
||||||
import org.dom4j.Document;
|
import org.dom4j.Document;
|
||||||
import org.dom4j.Element;
|
import org.dom4j.Element;
|
||||||
import org.dom4j.io.SAXReader;
|
import org.dom4j.io.SAXReader;
|
||||||
|
|
||||||
import java.io.*;
|
import java.io.File;
|
||||||
import java.util.*;
|
import java.io.FileInputStream;
|
||||||
|
import java.util.*;
|
||||||
/**
|
|
||||||
* 解析wordpress导出的xml文章数据为Map结果集
|
/**
|
||||||
* @uthor guqing
|
* 解析 WordPress 导出的 xml 文章数据为 Map 结果集
|
||||||
*/
|
*
|
||||||
public class TransferMapProcessor {
|
* @author guqing
|
||||||
// 存储在此集合中的节点名称都会被解析为一个List存储
|
*/
|
||||||
private static final List<String> ARRAY_PROPERTY = Arrays.asList(new String[]{"channel", "item", "category", "postmeta", "comment"});
|
public class XmlTransferMapUtils {
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* 根据xml文件对象获取xml的根节点rootElement对象
|
* 存储在此集合中的节点名称都会被解析为一个List存储
|
||||||
* @param file xml的文件对象
|
*/
|
||||||
* @return 返回根节点元素
|
private static final List<String> ARRAY_PROPERTY = Arrays.asList("channel", "item", "category", "postmeta", "comment");
|
||||||
*/
|
|
||||||
public Element getRootElement(File file) {
|
/**
|
||||||
try {
|
* 根据xml文件对象获取 xml 的根节点 rootElement 对象
|
||||||
SAXReader saxReader = new SAXReader();
|
*
|
||||||
FileInputStream fileInputStream = new FileInputStream(file);
|
* @param file xml的文件对象
|
||||||
Document document = saxReader.read(fileInputStream);
|
* @return 返回根节点元素
|
||||||
Element rootElement = document.getRootElement();
|
*/
|
||||||
|
public Element getRootElement(File file) {
|
||||||
return rootElement;
|
try {
|
||||||
} catch (Exception e) {
|
SAXReader saxReader = new SAXReader();
|
||||||
throw new RuntimeException("can not get root element");
|
FileInputStream fileInputStream = new FileInputStream(file);
|
||||||
}
|
Document document = saxReader.read(fileInputStream);
|
||||||
}
|
return document.getRootElement();
|
||||||
|
} catch (Exception e) {
|
||||||
|
throw new RuntimeException("can not get root element");
|
||||||
/**
|
}
|
||||||
* 根据根节点获取子节点元素集合递归遍历得到Map结果集
|
}
|
||||||
* @param root xml的根节点元素对象
|
|
||||||
* @return 返回解析xml得到的Map结果集
|
|
||||||
*/
|
/**
|
||||||
public Map<String, Object> getResultSetMapping(Element root) {
|
* 根据根节点获取子节点元素集合递归遍历得到Map结果集
|
||||||
Map<String, Object> result = new HashMap<String, Object>();
|
*
|
||||||
|
* @param root xml的根节点元素对象
|
||||||
try {
|
* @return 返回解析xml得到的Map结果集
|
||||||
// 获取根元素的所有子元素
|
*/
|
||||||
List<Element> children = root.elements();
|
public Map<String, Object> getResultSetMapping(Element root) {
|
||||||
|
Map<String, Object> result;
|
||||||
//递归遍历将xml节点数据解析为Map结果集
|
try {
|
||||||
result = transfer2Map(children,null);
|
// 获取根元素的所有子元素
|
||||||
} catch (Exception e) {
|
List<Element> children = root.elements();
|
||||||
throw new RuntimeException("can not transfer xml file to map." + e.getMessage());
|
|
||||||
}
|
//递归遍历将xml节点数据解析为Map结果集
|
||||||
|
result = transfer2Map(children, null);
|
||||||
return result;
|
} catch (Exception e) {
|
||||||
}
|
throw new RuntimeException("can not transfer xml file to map." + e.getMessage());
|
||||||
|
}
|
||||||
/**
|
|
||||||
* 递归解析xml，实现N层解析
|
return result;
|
||||||
* @param elements 所有子节点元素集，随着递归遍历而改变
|
}
|
||||||
* @param list 存储中间遍历结果的容器
|
|
||||||
* @return 返回递归完成后的Map结果集映射
|
/**
|
||||||
*/
|
* 递归解析xml，实现N层解析
|
||||||
private Map<String, Object> transfer2Map(List<Element> elements,List<Map<String,Object>> list){
|
*
|
||||||
Map<String, Object> map = new HashMap<String, Object>();
|
* @param elements 所有子节点元素集，随着递归遍历而改变
|
||||||
|
* @param list 存储中间遍历结果的容器
|
||||||
for(Element element : elements){
|
* @return 返回递归完成后的Map结果集映射
|
||||||
String name = element.getName();
|
*/
|
||||||
//如果是定义成数组
|
private Map<String, Object> transfer2Map(List<Element> elements, List<Map<String, Object>> list) {
|
||||||
if(ARRAY_PROPERTY.contains(name)) {
|
Map<String, Object> map = new HashMap<String, Object>();
|
||||||
//继续递归循环
|
|
||||||
List<Map<String,Object>> sublist = new ArrayList<Map<String,Object>>();
|
for (Element element : elements) {
|
||||||
|
String name = element.getName();
|
||||||
Map<String,Object> subMap = this.transfer2Map(element.elements(), sublist);
|
//如果是定义成数组
|
||||||
|
if (ARRAY_PROPERTY.contains(name)) {
|
||||||
//根据key获取是否已经存在
|
//继续递归循环
|
||||||
Object object = map.get(name);
|
List<Map<String, Object>> sublist = new ArrayList<Map<String, Object>>();
|
||||||
//如果存在,合并
|
|
||||||
if(object !=null ){
|
Map<String, Object> subMap = this.transfer2Map(element.elements(), sublist);
|
||||||
List<Map<String,Object>> olist = (List<Map<String,Object>>)object;
|
|
||||||
olist.add(subMap);//
|
//根据key获取是否已经存在
|
||||||
map.put(name, olist);
|
Object object = map.get(name);
|
||||||
}else{
|
//如果存在,合并
|
||||||
//否则直接存入map
|
if (object != null) {
|
||||||
map.put(name, sublist);
|
List<Map<String, Object>> olist = (List<Map<String, Object>>) object;
|
||||||
}
|
olist.add(subMap);
|
||||||
}else {
|
map.put(name, olist);
|
||||||
//单个值存入map
|
} else {
|
||||||
map.put(name, element.getTextTrim());
|
//否则直接存入map
|
||||||
}
|
map.put(name, sublist);
|
||||||
}
|
}
|
||||||
|
} else {
|
||||||
//存入list中
|
//单个值存入map
|
||||||
if(list != null) {
|
map.put(name, element.getTextTrim());
|
||||||
list.add(map);
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
//返回结果集合
|
//存入list中
|
||||||
return map;
|
if (list != null) {
|
||||||
}
|
list.add(map);
|
||||||
}
|
}
|
||||||
|
//返回结果集合
|
||||||
|
return map;
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue