From f22a631422f3045a06aad627e6e83c802be2760d Mon Sep 17 00:00:00 2001
From: 于紫祥_1901 <email@yuzixiang_1910>
Date: Tue, 01 Dec 2020 11:07:29 +0800
Subject: [PATCH] 站点平均值数据导出

---
 src/main/java/com/moral/service/impl/MenuServiceImpl.java |   14 ++++++++------
 1 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/src/main/java/com/moral/service/impl/MenuServiceImpl.java b/src/main/java/com/moral/service/impl/MenuServiceImpl.java
index 81ae2fa..ac8f2a4 100644
--- a/src/main/java/com/moral/service/impl/MenuServiceImpl.java
+++ b/src/main/java/com/moral/service/impl/MenuServiceImpl.java
@@ -6,8 +6,6 @@
 import java.util.Iterator;
 import java.util.List;
 
-import javax.annotation.Resource;
-
 import org.apache.commons.collections.CollectionUtils;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
@@ -22,7 +20,7 @@
 import com.moral.mapper.RoleMenuMapper;
 import com.moral.service.MenuService;
 import com.moral.util.TkMybatisUtils;
-
+import javax.annotation.Resource;
 import tk.mybatis.mapper.entity.Example;
 
 @Service
@@ -129,10 +127,14 @@
         List<Menu> menuList = menuMapper.getMenuListInfo();
         List<Menu> zNodes = new ArrayList<>();
         for (Menu menu : menuList) {
-            if (menu.getMenuOrder().toString().length() == 2) {
+            if (menu.getMenuOrder().toString().substring(0, 1).equals("1")) {
                 menu.setMenuParentId(Integer.valueOf(menu.getMenuOrder().toString().substring(0, 1)));
-            } else if (menu.getMenuOrder().toString().length() == 3) {
-                menu.setMenuParentId(Integer.valueOf(menu.getMenuOrder().toString().substring(0, 2)));
+            } else {
+                if (menu.getMenuOrder().toString().length() == 2) {
+                    menu.setMenuParentId(Integer.valueOf(menu.getMenuOrder().toString().substring(0, 1)));
+                } else if (menu.getMenuOrder().toString().length() == 3) {
+                    menu.setMenuParentId(Integer.valueOf(menu.getMenuOrder().toString().substring(0, 2)));
+                }
             }
             zNodes.add(menu);
         }

--
Gitblit v1.8.0