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/ColumnServiceImpl.java |   14 +++++++++++++-
 1 files changed, 13 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/ColumnServiceImpl.java b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/ColumnServiceImpl.java
index 934ac05..d5aeb3a 100644
--- a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/ColumnServiceImpl.java
+++ b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/ColumnServiceImpl.java
@@ -13,8 +13,10 @@
 import com.xcong.farmer.cms.modules.system.dto.AdminUpdateColumnDto;
 import com.xcong.farmer.cms.modules.system.entity.ColumnEntity;
 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.ColumnMapper;
 import com.xcong.farmer.cms.modules.system.service.IColumnService;
+import com.xcong.farmer.cms.modules.system.util.LoginUserUtil;
 import com.xcong.farmer.cms.modules.system.vo.AdminColumnVo;
 import com.xcong.farmer.cms.modules.system.vo.AdminSeeColumnInfoVo;
 import lombok.extern.slf4j.Slf4j;
@@ -32,8 +34,11 @@
 public class ColumnServiceImpl extends ServiceImpl<ColumnMapper, ColumnEntity> implements IColumnService {
     @Override
     public Result getColumnInPage(AdminColumnDto adminColumnDto) {
+        UserEntity userlogin = LoginUserUtil.getLoginUser();
+        long belongId = userlogin.getBelongId() == null ? 0L : userlogin.getBelongId();
         Page<AdminColumnVo> page = new Page<>(adminColumnDto.getPageNum(), adminColumnDto.getPageSize());
         ColumnEntity columnEntity = new ColumnEntity();
+        columnEntity.setBelongId(belongId);
         columnEntity.setParentId(ColumnEntity.PARENTID_DEFAULT);
         IPage<AdminColumnVo> list = this.baseMapper.selectAdminColumnVoInPage(page,columnEntity);
         List<AdminColumnVo> records = list.getRecords();
@@ -42,6 +47,7 @@
                 Long id = adminColumnVo.getId();
                 QueryWrapper<ColumnEntity> objectQueryWrapper = new QueryWrapper<>();
                 objectQueryWrapper.eq("parent_id",id);
+                objectQueryWrapper.eq("belong_id",belongId);
                 List<ColumnEntity> columnEntities = this.baseMapper.selectList(objectQueryWrapper);
                 List<AdminColumnVo> adminColumnVoChilds = new ArrayList<>();
                 if(CollUtil.isNotEmpty(columnEntities)){
@@ -65,7 +71,10 @@
     @Override
     @Transactional
     public Result addColumn(AdminAddColumnDto adminAddColumnDto) {
+        UserEntity userlogin = LoginUserUtil.getLoginUser();
+        long belongId = userlogin.getBelongId() == null ? 0L : userlogin.getBelongId();
         ColumnEntity columnEntity = new ColumnEntity();
+        columnEntity.setBelongId(belongId);
         String columnName = adminAddColumnDto.getColumnName();
         if(StrUtil.isEmpty(columnName)){
             return Result.fail("请输入栏目名称");
@@ -189,12 +198,15 @@
 
     @Override
     public Result getColumnInList() {
-        List<AdminColumnVo> records = this.baseMapper.selectColumnInListByParentId(ColumnEntity.PARENTID_DEFAULT);
+        UserEntity userlogin = LoginUserUtil.getLoginUser();
+        long belongId = userlogin.getBelongId() == null ? 0L : userlogin.getBelongId();
+        List<AdminColumnVo> records = this.baseMapper.selectColumnInListByParentId(ColumnEntity.PARENTID_DEFAULT,belongId);
         if(CollUtil.isNotEmpty(records)){
             for(AdminColumnVo adminColumnVo : records){
                 Long id = adminColumnVo.getId();
                 QueryWrapper<ColumnEntity> objectQueryWrapper = new QueryWrapper<>();
                 objectQueryWrapper.eq("parent_id",id);
+                objectQueryWrapper.eq("belong_id",belongId);
                 List<ColumnEntity> columnEntities = this.baseMapper.selectList(objectQueryWrapper);
                 List<AdminColumnVo> adminColumnVoChilds = new ArrayList<>();
                 if(CollUtil.isNotEmpty(columnEntities)){

--
Gitblit v1.9.1