Fixed XmlTransferMapUtils.

src/main/java/run/halo/app/utils/XmlTransferMapUtils.java

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