From 70830c5b2e96cc1c76d4adb88ddced55e7476547 Mon Sep 17 00:00:00 2001
From: ZhuDongming <773644075@qq.com>
Date: Thu, 28 Nov 2019 15:33:05 +0800
Subject: [PATCH] 大屏后台管理代码update

---
 src/main/java/com/moral/service/impl/MenuServiceImpl.java |   36 +++++++++++++++++++++++++++++++-----
 1 files changed, 31 insertions(+), 5 deletions(-)

diff --git a/src/main/java/com/moral/service/impl/MenuServiceImpl.java b/src/main/java/com/moral/service/impl/MenuServiceImpl.java
index 0c5da0c..07655dc 100644
--- a/src/main/java/com/moral/service/impl/MenuServiceImpl.java
+++ b/src/main/java/com/moral/service/impl/MenuServiceImpl.java
@@ -1,5 +1,6 @@
 package com.moral.service.impl;
 
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Date;
 import java.util.Iterator;
@@ -7,6 +8,7 @@
 
 import javax.annotation.Resource;
 
+import org.apache.commons.collections.CollectionUtils;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
@@ -16,9 +18,9 @@
 import com.moral.common.util.ExampleUtil;
 import com.moral.entity.Menu;
 import com.moral.mapper.MenuMapper;
+import com.moral.mapper.RoleMenuMapper;
 import com.moral.service.MenuService;
 import com.moral.util.TkMybatisUtils;
-import com.sun.scenario.effect.impl.sw.sse.SSEBlend_SRC_OUTPeer;
 
 import tk.mybatis.mapper.entity.Example;
 
@@ -30,11 +32,14 @@
     @Resource
     private MenuMapper menuMapper;
 
+    @Resource
+    private RoleMenuMapper roleMenuMapper;
+
     @Override
     @Transactional
     public void addOrModify(Menu menu) {
         try {
-            if(menu.getMenuParentId()==null){
+            if (menu.getMenuParentId() == null) {
                 menu.setMenuParentId(0);
             }
             menu.setUpdateTime(new Date());
@@ -116,8 +121,29 @@
     }
 
     @Override
-    public List<Menu> getMenuListByName(String menuName) {
-        List<Menu> menuList = menuMapper.getMenuListByName(menuName);
-        return menuList;
+    public List<Menu> getZNodes() {
+        List<Menu> menuList = menuMapper.getMenuListInfo();
+        List<Menu> zNodes = new ArrayList<>();
+        for (Menu menu : menuList) {
+            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);
+        }
+        return zNodes;
+    }
+
+    @Override
+    public List<Integer> getMenuOrders(Integer roleId) {
+        List<Integer> menuOrders = menuMapper.getMenuOrdersByRoleId(roleId);
+        List<Integer> channels = roleMenuMapper.getChannels(roleId);
+        if (CollectionUtils.isNotEmpty(channels)) {
+            for (Integer channel : channels) {
+                menuOrders.add(channel);
+            }
+        }
+        return menuOrders;
     }
 }

--
Gitblit v1.8.0