From 7fa0a141a3c88ca90c67c966260bf26f03f255da Mon Sep 17 00:00:00 2001
From: xiaoyong931011 <15274802129@163.com>
Date: Tue, 07 Jun 2022 16:00:09 +0800
Subject: [PATCH] 20220606

---
 src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/NavigationBarServiceImpl.java |   14 ++++++++++++--
 1 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/NavigationBarServiceImpl.java b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/NavigationBarServiceImpl.java
index f72cd93..795bc8c 100644
--- a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/NavigationBarServiceImpl.java
+++ b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/NavigationBarServiceImpl.java
@@ -12,9 +12,11 @@
 import com.xcong.farmer.cms.modules.system.dto.AdminNavigationBarDto;
 import com.xcong.farmer.cms.modules.system.dto.AdminUpdateBarDto;
 import com.xcong.farmer.cms.modules.system.entity.NavigationBarEntity;
+import com.xcong.farmer.cms.modules.system.entity.UserEntity;
 import com.xcong.farmer.cms.modules.system.mapper.NavigationBarMapper;
 import com.xcong.farmer.cms.modules.system.mapper.UserRoleMapper;
 import com.xcong.farmer.cms.modules.system.service.INavigationBarService;
+import com.xcong.farmer.cms.modules.system.util.LoginUserUtil;
 import com.xcong.farmer.cms.modules.system.vo.AdminNavigationBarVo;
 import com.xcong.farmer.cms.modules.system.vo.AdminSeeBarInfoVo;
 import lombok.extern.slf4j.Slf4j;
@@ -39,9 +41,12 @@
 
     @Override
     public Result getBarInPage(AdminNavigationBarDto adminNavigationBarDto) {
+        UserEntity userlogin = LoginUserUtil.getLoginUser();
+        long belongId = userlogin.getBelongId() == null ? 0L : userlogin.getBelongId();
         Page<AdminNavigationBarVo> page = new Page<>(adminNavigationBarDto.getPageNum(), adminNavigationBarDto.getPageSize());
         NavigationBarEntity navigationBarEntity = new NavigationBarEntity();
         navigationBarEntity.setParentId(NavigationBarEntity.PARENTID_DEFAULT);
+        navigationBarEntity.setBelongId(belongId);
         IPage<AdminNavigationBarVo> list = navigationBarMapper.selectAdminNavigationBarVoInPage(page,navigationBarEntity);
         List<AdminNavigationBarVo> records = list.getRecords();
         if(CollUtil.isNotEmpty(records)){
@@ -72,7 +77,10 @@
     @Override
     @Transactional
     public Result addBar(AdminAddBarDto adminAddBarDto) {
+        UserEntity userlogin = LoginUserUtil.getLoginUser();
+        long belongId = userlogin.getBelongId() == null ? 0L : userlogin.getBelongId();
         NavigationBarEntity navigationBarEntity = new NavigationBarEntity();
+        navigationBarEntity.setBelongId(belongId);
         String barName = adminAddBarDto.getBarName();
         if(StrUtil.isEmpty(barName)){
             return Result.fail("请输入导航栏标题");
@@ -182,11 +190,13 @@
 
     @Override
     public Result getBarInList() {
-        List<AdminNavigationBarVo> records = navigationBarMapper.selectAdminNavigationBarVoByParentId(NavigationBarEntity.PARENTID_DEFAULT);
+        UserEntity userlogin = LoginUserUtil.getLoginUser();
+        long belongId = userlogin.getBelongId() == null ? 0L : userlogin.getBelongId();
+        List<AdminNavigationBarVo> records = navigationBarMapper.selectAdminNavigationBarVoByParentId(NavigationBarEntity.PARENTID_DEFAULT,belongId);
         if(CollUtil.isNotEmpty(records)){
             for(AdminNavigationBarVo adminNavigationBarVo : records){
                 Long id = adminNavigationBarVo.getId();
-                List<NavigationBarEntity> navigationBarEntities = navigationBarMapper.selectListByParentId(id);
+                List<NavigationBarEntity> navigationBarEntities = navigationBarMapper.selectListByParentId(id,belongId);
                 List<AdminNavigationBarVo> adminNavigationBarVoChilds = new ArrayList<>();
                 if(CollUtil.isNotEmpty(navigationBarEntities)){
                     for(NavigationBarEntity navigationBarEntityChild : navigationBarEntities){

--
Gitblit v1.9.1