From 8a66d1972d2e9245e2999d06d49d2c864492ab17 Mon Sep 17 00:00:00 2001
From: JinPengYong <812110275@qq.com>
Date: Sun, 21 Mar 2021 16:12:29 +0800
Subject: [PATCH] 获取用户层级菜单update

---
 screen-api/src/main/java/com/moral/api/controller/MenuController.java    |    3 +--
 screen-api/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java |    9 ++++++---
 screen-api/src/main/java/com/moral/api/service/UserService.java          |    2 ++
 screen-api/src/main/java/com/moral/api/service/impl/UserServiceImpl.java |   16 +++++++++-------
 4 files changed, 18 insertions(+), 12 deletions(-)

diff --git a/screen-api/src/main/java/com/moral/api/controller/MenuController.java b/screen-api/src/main/java/com/moral/api/controller/MenuController.java
index ecb36f9..94c4f07 100644
--- a/screen-api/src/main/java/com/moral/api/controller/MenuController.java
+++ b/screen-api/src/main/java/com/moral/api/controller/MenuController.java
@@ -68,7 +68,6 @@
         String token = request.getHeader("token");
         Map<String, Object> currentUserInfo = (Map<String, Object>) TokenUtils.getUserInfoByToken(token);
         Map<String, Object> result = menuService.selectMenusByOrgId((Integer) currentUserInfo.get("organizationId"));
-
-        return ResultMessage.ok();
+        return ResultMessage.ok(result);
     }
 }
diff --git a/screen-api/src/main/java/com/moral/api/service/UserService.java b/screen-api/src/main/java/com/moral/api/service/UserService.java
index 4ca4de2..2af3654 100644
--- a/screen-api/src/main/java/com/moral/api/service/UserService.java
+++ b/screen-api/src/main/java/com/moral/api/service/UserService.java
@@ -21,6 +21,8 @@
 
     Map<String, Object> login(Map<String, Object> parameters);
 
+    Map<String, Object> getMenus(Map<String, Object> parameters);
+
     Map<String, Object> addUser(User user, String token);
 
     void deleteUser(User user);
diff --git a/screen-api/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java b/screen-api/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java
index 337fc30..8d35511 100644
--- a/screen-api/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java
+++ b/screen-api/src/main/java/com/moral/api/service/impl/MenuServiceImpl.java
@@ -1,5 +1,6 @@
 package com.moral.api.service.impl;
 
+import java.util.HashMap;
 import java.util.Map;
 
 import com.moral.api.entity.Menu;
@@ -23,11 +24,13 @@
 public class MenuServiceImpl extends ServiceImpl<MenuMapper, Menu> implements MenuService {
 
     @Autowired
-    private UserServiceImpl userService;
+    private UserService userService;
 
     @Override
     public Map<String, Object> selectMenusByOrgId(Integer orgId) {
-        userService.getMenus(-1, orgId);
-        return null;
+        Map<String, Object> map = new HashMap<>();
+        map.put("orgId",orgId);
+        Map<String, Object> menus = userService.getMenus(map);
+        return menus;
     }
 }
diff --git a/screen-api/src/main/java/com/moral/api/service/impl/UserServiceImpl.java b/screen-api/src/main/java/com/moral/api/service/impl/UserServiceImpl.java
index d71bda7..100245f 100644
--- a/screen-api/src/main/java/com/moral/api/service/impl/UserServiceImpl.java
+++ b/screen-api/src/main/java/com/moral/api/service/impl/UserServiceImpl.java
@@ -114,7 +114,9 @@
         userInfo.put("organizationName", organization.getName());
         userInfo.put("locationLevel", organization.getLocationLevel());
         userInfo.put("groups", groups);
-        userInfo.putAll(getMenus(userBo.getId(), -1));
+        Map<String, Object> map = new HashMap<>();
+        map.put("userId",userBo.getId());
+        userInfo.putAll(getMenus(map));
         try {
             //������token������������redis
             String token = TokenUtils.getToken(userBo.getId().toString(), userInfo);
@@ -130,13 +132,13 @@
         return result;
     }
 
-    //������userId������������������������
-    public Map<String, Object> getMenus(int userId, int orgId) {
+    @Override
+    public Map<String, Object> getMenus(Map<String, Object> parameters) {
         List<Menu> allMenus;
-        if (orgId == -1) {
-            allMenus = menuMapper.selectUserMenu(userId);
+        if (parameters.containsKey("userId")) {
+            allMenus = menuMapper.selectUserMenu(Integer.parseInt(parameters.get("userId").toString()));
         } else {
-            allMenus = menuMapper.selectOrganizationMenu(orgId);
+            allMenus = menuMapper.selectOrganizationMenu(Integer.parseInt(parameters.get("orgId").toString()));
         }
 
         Map<String, Object> resultMap = new LinkedHashMap<>();
@@ -162,7 +164,7 @@
     }
 
     //������������������������������������
-    public List<Map<String, Object>> getMenusByRecursion(Menu menu, List<Menu> menus) {
+    private List<Map<String, Object>> getMenusByRecursion(Menu menu, List<Menu> menus) {
         List<List<Map<String, Object>>> resultList = new ArrayList();
         Menu newMenu = new Menu();
         newMenu.setParentId(menu.getId());

--
Gitblit v1.8.0