搜索优化2

muyu-modules/muyu-system/src/main/java/com/muyu/system/controller/SysMenuController.java

@@ -30,8 +30,6 @@ public class SysMenuController extends BaseController {
     private SysMenuService menuService;
 
 
-
-
     /**
      * 获取菜单列表
      */
@@ -43,6 +41,14 @@ public class SysMenuController extends BaseController {
         return success(menus);
     }
 
+    @RequiresPermissions("system:menu:list")
+    @GetMapping("/listTwo")
+    public Result listTwo(SysMenu menu) {
+        Long userId = SecurityUtils.getUserId();
+        List<SysMenu> menusTwo = menuService.selectMenu(menu, userId);
+        return success(menusTwo);
+    }
+
     /**
      * 根据菜单编号获取详细信息
      */

muyu-modules/muyu-system/src/main/java/com/muyu/system/controller/SysUserOnlineController.java

@@ -38,20 +38,7 @@ public class SysUserOnlineController extends BaseController {
     @RequiresPermissions("monitor:online:list")
     @GetMapping("/list")
     public Result<TableDataInfo<SysUserOnline>> list (String ipaddr, String userName) {
-        Collection<String> keys = redisService.keys(CacheConstants.LOGIN_TOKEN_KEY + "*");
+        List<SysUserOnline> userOnlineList = getUserOnlineList(ipaddr, userName);
-        List<SysUserOnline> userOnlineList = new ArrayList<SysUserOnline>();
-        for (String key : keys) {
-            LoginUser user = redisService.getCacheObject(key);
-            if (StringUtils.isNotEmpty(ipaddr) && StringUtils.isNotEmpty(userName)) {
-                userOnlineList.add(userOnlineService.selectOnlineByInfo(ipaddr, userName, user));
-            } else if (StringUtils.isNotEmpty(ipaddr)) {
-                userOnlineList.add(userOnlineService.selectOnlineByIpaddr(ipaddr, user));
-            } else if (StringUtils.isNotEmpty(userName)) {
-                userOnlineList.add(userOnlineService.selectOnlineByUserName(userName, user));
-            } else {
-                userOnlineList.add(userOnlineService.loginUserToUserOnline(user));
-            }
-        }
         Collections.reverse(userOnlineList);
         userOnlineList.removeAll(Collections.singleton(null));
         return getDataTable(userOnlineList);
@@ -68,12 +55,21 @@ public class SysUserOnlineController extends BaseController {
         return success();
     }
 
+
+
     @PostMapping("/findTokenId")
     public Result<String> findTokenId(@RequestBody TokenResp tokenResp) {
-        String name = tokenResp.getName();
+        List<SysUserOnline> userOnlineList = getUserOnlineList(tokenResp.getIpaddr() ,tokenResp.getUserName()  );
-        String ipaddr = tokenResp.getIpaddr();
+        String id="";
-        String userName = tokenResp.getUserName();
+        for (SysUserOnline sysUserOnline : userOnlineList) {
+            if (sysUserOnline.getUserName().equals(tokenResp.getName())) {
+                 id = sysUserOnline.getTokenId();
+            }
+        }
+        return Result.success(id);
+    }
 
+    private List<SysUserOnline> getUserOnlineList(String ipaddr, String userName) {
         Collection<String> keys = redisService.keys(CacheConstants.LOGIN_TOKEN_KEY + "*");
         List<SysUserOnline> userOnlineList = new ArrayList<SysUserOnline>();
         for (String key : keys) {
@@ -88,13 +84,7 @@ public class SysUserOnlineController extends BaseController {
                 userOnlineList.add(userOnlineService.loginUserToUserOnline(user));
             }
         }
-        String id="";
+        return userOnlineList;
-        for (SysUserOnline sysUserOnline : userOnlineList) {
-            if (sysUserOnline.getUserName().equals(name)) {
-                 id = sysUserOnline.getTokenId();
-            }
-        }
-        return Result.success(id);
     }
 
 }

muyu-modules/muyu-system/src/main/java/com/muyu/system/domain/SysMenu.java

@@ -1,6 +1,7 @@
 package com.muyu.system.domain;
 
 import com.muyu.common.core.web.domain.BaseEntity;
+import com.muyu.system.domain.vo.Mean;
 import lombok.AllArgsConstructor;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
@@ -8,7 +9,6 @@ import lombok.NoArgsConstructor;
 import lombok.experimental.SuperBuilder;
 import org.apache.commons.lang3.builder.ToStringBuilder;
 import org.apache.commons.lang3.builder.ToStringStyle;
-
 import javax.validation.constraints.NotBlank;
 import javax.validation.constraints.NotNull;
 import javax.validation.constraints.Size;
@@ -108,6 +108,21 @@ public class SysMenu extends BaseEntity {
      */
     private List<SysMenu> children = new ArrayList<SysMenu>();
 
+    public static SysMenu meanBuild(Mean mean1, Long id) {
+        return SysMenu.builder()
+                .menuId(id)
+                .menuName(mean1.getMenuName())
+                .parentId(mean1.getParentId())
+                .orderNum(mean1.getOrderNum())
+                .path(mean1.getPath())
+                .component(mean1.getComponent())
+                .query(mean1.getQuery())
+                .isFrame(String.valueOf(mean1.getIsFrame()))
+                .isCache(String.valueOf(mean1.getIsCache()))
+                .menuType(mean1.getMenuType())
+                .build();
+    }
+
     public Long getMenuId() {
         return menuId;
     }

muyu-modules/muyu-system/src/main/java/com/muyu/system/service/SysMenuService.java

@@ -158,4 +158,6 @@ public interface SysMenuService extends IService<SysMenu> {
      * @return 结果
      */
     public boolean checkMenuNameUnique (SysMenu menu);
+
+    List<SysMenu> selectMenu(SysMenu menu, Long userId);
 }

muyu-modules/muyu-system/src/main/java/com/muyu/system/service/impl/SysMenuServiceImpl.java

@@ -24,6 +24,7 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.util.*;
+import java.util.function.Supplier;
 import java.util.stream.Collectors;
 
 /**
@@ -59,29 +60,6 @@ public class SysMenuServiceImpl extends ServiceImpl<SysMenuMapper, SysMenu> impl
     }
 
 
-    private void getParentIdList(List<Long> ids, Long parentId) {
-
-        if (parentId.equals(0L)) {
-            return;
-        }
-        Mean mean = this.meanService.getOne(new LambdaQueryWrapper<>() {{
-            eq(Mean::getMenuId, parentId);
-        }});
-
-        ids.add(mean.getMenuId());
-        getParentIdList(ids, mean.getParentId());
-
-    }
-
-    private void getSonIdList(List<Long> sonIds, Long menuId) {
-        Mean mean = this.meanService.getOne(new LambdaQueryWrapper<>() {{
-            eq(Mean::getParentId, menuId);
-        }});
-
-        sonIds.add(mean.getMenuId());
-        getParentIdList(sonIds, mean.getMenuId());
-    }
-
     /**
      * 查询系统菜单列表
      *
@@ -91,22 +69,6 @@ public class SysMenuServiceImpl extends ServiceImpl<SysMenuMapper, SysMenu> impl
     @Override
     public List<SysMenu> selectMenuList(SysMenu menu, Long userId) {
 
-
-//        List<Long> ids = new ArrayList<>();
-//        List<Long> sonIds = new ArrayList<>();
-//        if (menu != null) {
-//            Mean mean = this.meanService.getOne(new LambdaQueryWrapper<>() {{
-//                eq(Mean::getMenuName, menu.getMenuName());
-//            }});
-//            ids = new ArrayList<>();
-//            sonIds = new ArrayList<>();
-//            getParentIdList(ids, mean.getParentId());
-//            getSonIdList(sonIds, mean.getMenuId());
-//        }
-//
-//        System.out.println(ids);
-//        System.out.println(sonIds);
-
         List<SysMenu> menuList = null;
         // 管理员显示所有菜单信息
         if (SysUser.isAdmin(userId)) {
@@ -116,40 +78,19 @@ public class SysMenuServiceImpl extends ServiceImpl<SysMenuMapper, SysMenu> impl
             menuList = menuMapper.selectMenuListByUserId(menu);
         }
 
-//        List<SysMenu> finalMenuList = menuList;
-//        ids.forEach(id -> {
-//            Mean mean1 = this.meanService.getOne(new LambdaQueryWrapper<>() {{
-//                eq(Mean::getMenuId, id);
-//            }});
-//            finalMenuList.add(
-//                    SysMenu.builder()
-//                            .menuId(id)
-//                            .menuName(mean1.getMenuName())
-//                            .parentId(mean1.getParentId())
-//                            .orderNum(mean1.getOrderNum())
-//                            .path(mean1.getPath())
-//                            .component(mean1.getComponent())
-//                            .query(mean1.getQuery())
-//                            .isFrame(String.valueOf(mean1.getIsFrame()))
-//                            .isCache(String.valueOf(mean1.getIsCache()))
-//                            .menuType(mean1.getMenuType())
-//                            .build()
-//            );
-//        });
-
         if (menuList.size() == 1) {
             getParentList(menuList, menuList.get(0));
-
             List<SysMenu> childList = this.getChildList(menuMapper.selectMenuList(null), menuList.get(0));
             menuList.addAll(childList);
-
         }
 
         return menuList;
     }
 
+    /**
+     * 通过id 找儿子
+     */
     private void getParentList(List<SysMenu> menuList, SysMenu sysMenu) {
-
         SysMenu menu = this.selectMenuById(sysMenu.getParentId());
         if (menu != null) {
             menuList.add(menu);
@@ -158,6 +99,65 @@ public class SysMenuServiceImpl extends ServiceImpl<SysMenuMapper, SysMenu> impl
     }
 
 
+    /**
+     * 获取父节点 和子节点
+     */
+    private void getParentIdAndSonIdList(List<Long> ids, Long parentId, Long menuId) {
+
+        if (parentId.equals(0L)) {
+            return;
+        }
+        Mean mean = this.meanService.getOne(new LambdaQueryWrapper<>() {{
+            eq(Mean::getMenuId, parentId);
+        }});
+
+        List<Long> longList = meanService.list(new LambdaQueryWrapper<>() {{
+            eq(Mean::getParentId, menuId);
+        }}).stream().map(
+                Mean::getMenuId
+        ).toList();
+
+        ids.addAll(longList.stream().distinct().toList());
+        ids.add(mean.getMenuId());
+        getParentIdAndSonIdList(ids, mean.getParentId(), menuId);
+    }
+
+
+    @Override
+    public List<SysMenu> selectMenu(SysMenu menu, Long userId) {
+        List<Long> ids = new ArrayList<>();
+        if (menu != null) {
+            Mean mean = meanService.getOne(new LambdaQueryWrapper<Mean>() {{
+                eq(Mean::getMenuName, menu.getMenuName());
+            }});
+            if (mean != null) {
+                ids = new ArrayList<>();
+                getParentIdAndSonIdList(ids, mean.getParentId(), mean.getMenuId());
+            }
+        }
+        List<SysMenu> menuList = null;
+
+        // 管理员显示所有菜单信息
+        if (SysUser.isAdmin(userId)) {
+            menuList = menuMapper.selectMenuList(menu);
+        } else {
+            menu.getParams().put("userId", userId);
+            menuList = menuMapper.selectMenuListByUserId(menu);
+        }
+
+        List<SysMenu> finalMenuList = menuList;
+        ids.forEach(id -> {
+            finalMenuList.add(
+                    SysMenu.meanBuild(this.meanService.getOne(new LambdaQueryWrapper<>() {{
+                        eq(Mean::getMenuId, id);
+                    }}),id)
+            );
+        });
+
+        return finalMenuList;
+    }
+
+
     /**
      * 根据用户ID查询权限
      *
@@ -394,6 +394,7 @@ public class SysMenuServiceImpl extends ServiceImpl<SysMenuMapper, SysMenu> impl
         return UserConstants.UNIQUE;
     }
 
+
     /**
      * 获取路由名称
      *