添加历史影像

ruoyi-admin/src/main/java/com/ruoyi/web/controller/system/SysLoginController.java

@@ -1,7 +1,13 @@
 package com.ruoyi.web.controller.system;
 
 import java.util.List;
+import java.util.Objects;
 import java.util.Set;
+
+import com.ruoyi.common.core.domain.entity.SysDictData;
+import com.ruoyi.common.core.domain.model.LoginUser;
+import com.ruoyi.system.service.ISysDictDataService;
+import com.ruoyi.system.service.ISysUserService;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -34,6 +40,12 @@ public class SysLoginController
     @Autowired
     private SysPermissionService permissionService;
 
+    @Autowired
+    private ISysDictDataService dictDataService;
+
+    @Autowired
+    private ISysUserService userService;
+
     /**
      * 登录方法
      * 
@@ -81,6 +93,27 @@ public class SysLoginController
     {
         Long userId = SecurityUtils.getUserId();
         List<SysMenu> menus = menuService.selectMenuTreeByUserId(userId);
+        // 设置菜单不可见
+        SysUser user = userService.selectUserById(userId);
+        String taskLx = user.getTaskLx();
+        if(!taskLx.isEmpty()){
+            String[] taskLxList = taskLx.split(",");
+            SysDictData sysDictData = new SysDictData();
+            sysDictData.setDictType("task_lx");
+            List<SysDictData> sysDictDataList = dictDataService.selectDictDataList(sysDictData);
+            for (String item : taskLxList) {
+                for (SysDictData dictData : sysDictDataList) {
+                    if (Objects.equals(item, dictData.getDictValue())){
+                        for (SysMenu menu : menus) {
+                            System.out.println(dictData.getDictLabel()+"===="+menu.getMenuName());
+                            if (menu.getMenuName().equals(dictData.getDictLabel())){
+                                menu.setVisible("1");
+                            }
+                        }
+                    }
+                }
+            }
+        }
         return AjaxResult.success(menuService.buildMenus(menus));
     }
 }