From 8beab91730f3e3cd6b53e4d498f763ff9dc65ef5 Mon Sep 17 00:00:00 2001
From: kaiyu <404897439@qq.com>
Date: Wed, 18 Nov 2020 14:16:55 +0800
Subject: [PATCH] 修改污染溯源接口,从history表中获取改为从history天表获取

---
 src/main/java/com/moral/controller/MenuController.java |   64 ++++++++++++++++++++++++++++++++
 1 files changed, 64 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/moral/controller/MenuController.java b/src/main/java/com/moral/controller/MenuController.java
index b15b7fa..03bea43 100644
--- a/src/main/java/com/moral/controller/MenuController.java
+++ b/src/main/java/com/moral/controller/MenuController.java
@@ -1,6 +1,10 @@
 package com.moral.controller;
 
+import java.util.ArrayList;
+import java.util.LinkedHashMap;
+import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 
 import javax.annotation.Resource;
 
@@ -14,6 +18,8 @@
 import com.moral.common.bean.PageBean;
 import com.moral.common.bean.ResultBean;
 import com.moral.entity.Menu;
+import com.moral.mapper.ChannelMapper;
+import com.moral.mapper.RoleMenuMapper;
 import com.moral.service.AccountService;
 import com.moral.service.ChannelService;
 import com.moral.service.MenuService;
@@ -24,6 +30,12 @@
 public class MenuController {
     @Resource
     MenuService menuService;
+
+    @Resource
+    ChannelMapper channelMapper;
+
+    @Resource
+    RoleMenuMapper roleMenuMapper;
 
     @Resource
     ChannelService channelService;
@@ -79,4 +91,56 @@
         Map<String,Object> mapResult = accountService.getMenuListsByAccountName(accountName);
         return mapResult;
     }
+
+    @GetMapping("getZNodes")
+    public List<Map<String,Object>> getZNodes(Integer roleId) {
+        List<Map<String,Object>> zNodes= new ArrayList<>();
+        List<Integer> channelIds = channelMapper.getChannelId();
+        List<Integer> channels = roleMenuMapper.getChannels(roleId);
+        List<Integer> menus=roleMenuMapper.getMenus(roleId);
+        List<Menu> menuList=menuService.getZNodes();
+        for(Integer channelId:channelIds){
+            Map<String,Object> channelNodeMap=new LinkedHashMap<>();
+            channelNodeMap.put("id",channelId);
+            channelNodeMap.put("pId",0);
+            if (Objects.equals(channelId, 1)) {
+                channelNodeMap.put("name", "������");
+            } else if (Objects.equals(channelId, 2)) {
+                channelNodeMap.put("name", "web������");
+            }else if (Objects.equals(channelId, 3)) {
+                channelNodeMap.put("name", "������APP");
+            }else if(Objects.equals(channelId, 4)){
+                channelNodeMap.put("name", "���������������APP");
+            }
+            channelNodeMap.put("open",true);
+            for(Integer roleHasChannelId:channels){
+                if(roleHasChannelId.equals(channelId)){
+                    channelNodeMap.put("checked",true);
+                }
+            }
+            zNodes.add(channelNodeMap);
+        }
+        for(Menu menu:menuList){
+            Map<String,Object> childNodeMap=new LinkedHashMap<>();
+            childNodeMap.put("id",menu.getMenuOrder());
+            childNodeMap.put("pId",menu.getMenuParentId());
+            childNodeMap.put("name", menu.getMenuName());
+            if(menu.getMenuOrder().toString().length()==2){
+                childNodeMap.put("open",true);
+            }
+            for(Integer roleHasMenuId:menus){
+                if(roleHasMenuId.equals(menu.getId())){
+                    childNodeMap.put("checked", true);
+                }
+            }
+            zNodes.add(childNodeMap);
+        }
+        return zNodes;
+    }
+
+    @GetMapping("menuOrders")
+    public List<Integer> getMenuOrders(Integer roleId) {
+        List<Integer> menuOrders=menuService.getMenuOrders(roleId);
+        return menuOrders;
+    }
 }

--
Gitblit v1.8.0