From 93e281b3cc60c8d24a0ed5547d3d399c03f9bab6 Mon Sep 17 00:00:00 2001
From: xiaoyong931011 <15274802129@163.com>
Date: Fri, 10 Jun 2022 18:03:52 +0800
Subject: [PATCH] 20220606

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

diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/WebSetServiceImpl.java b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/WebSetServiceImpl.java
index 3ba0015..8904e67 100644
--- a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/WebSetServiceImpl.java
+++ b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/WebSetServiceImpl.java
@@ -4,9 +4,11 @@
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.xcong.farmer.cms.common.response.Result;
 import com.xcong.farmer.cms.modules.system.dto.AdminUpdateWebSetDto;
+import com.xcong.farmer.cms.modules.system.entity.UserEntity;
 import com.xcong.farmer.cms.modules.system.entity.WebSetEntity;
 import com.xcong.farmer.cms.modules.system.mapper.WebSetMapper;
 import com.xcong.farmer.cms.modules.system.service.IWebSetService;
+import com.xcong.farmer.cms.modules.system.util.LoginUserUtil;
 import com.xcong.farmer.cms.modules.system.vo.AdminSeeWebSetInfoVo;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.stereotype.Service;
@@ -22,7 +24,11 @@
 public class WebSetServiceImpl extends ServiceImpl<WebSetMapper, WebSetEntity> implements IWebSetService {
     @Override
     public Result seeWebSetInfo() {
+        UserEntity userlogin = LoginUserUtil.getLoginUser();
+        long belongId = userlogin.getBelongId() == null ? UserEntity.USER_BELONG_TOP : userlogin.getBelongId();
         AdminSeeWebSetInfoVo adminSeeWebSetInfoVo = new AdminSeeWebSetInfoVo();
+        QueryWrapper<WebSetEntity> objectQueryWrapper = new QueryWrapper<>();
+        objectQueryWrapper.eq("belong_id",belongId);
         List<WebSetEntity> webSetEntities = this.baseMapper.selectList(new QueryWrapper<>());
         if(CollUtil.isNotEmpty(webSetEntities)){
             WebSetEntity webSetEntity = webSetEntities.get(0);
@@ -37,6 +43,8 @@
 
     @Override
     public Result updateWebSet(AdminUpdateWebSetDto adminUpdateWebSetDto) {
+        UserEntity userlogin = LoginUserUtil.getLoginUser();
+        long belongId = userlogin.getBelongId() == null ? UserEntity.USER_BELONG_TOP : userlogin.getBelongId();
         String webTitle = adminUpdateWebSetDto.getWebTitle();
         if(StrUtil.isEmpty(webTitle)){
             return Result.fail("请输入网页标题");
@@ -46,7 +54,10 @@
             return Result.fail("请输入网页关键字");
         }
         Long id = adminUpdateWebSetDto.getId() == null ? 0L : adminUpdateWebSetDto.getId();
-        WebSetEntity webSetEntity = this.baseMapper.selectById(id);
+        QueryWrapper<WebSetEntity> objectQueryWrapper = new QueryWrapper<>();
+        objectQueryWrapper.eq("belong_id",belongId);
+        WebSetEntity webSetEntity = this.baseMapper.selectOne(objectQueryWrapper);
+//        WebSetEntity webSetEntity = this.baseMapper.selectById(id);
         if(ObjectUtil.isEmpty(webSetEntity)){
             WebSetEntity webSetEntityAdd = new WebSetEntity();
             webSetEntityAdd.setWebTitle(webTitle);
@@ -59,6 +70,7 @@
             if(StrUtil.isNotEmpty(webPic)){
                 webSetEntityAdd.setWebPic(webPic);
             }
+            webSetEntityAdd.setBelongId(belongId);
             this.baseMapper.insert(webSetEntityAdd);
             return Result.ok("保存成功");
         }

--
Gitblit v1.9.1