From bd4ab82f2e0c485f36a254e3cbf72f21dabfd628 Mon Sep 17 00:00:00 2001 From: xiaoyong931011 <15274802129@163.com> Date: Fri, 08 Jul 2022 09:57:00 +0800 Subject: [PATCH] 20220606 --- src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/WebSetServiceImpl.java | 25 +++++++++++++------------ 1 files changed, 13 insertions(+), 12 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..ed8d8bc 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,8 +24,12 @@ public class WebSetServiceImpl extends ServiceImpl<WebSetMapper, WebSetEntity> implements IWebSetService { @Override public Result seeWebSetInfo() { + UserEntity userlogin = LoginUserUtil.getLoginUser(); + long companyId = userlogin.getCompanyId() == null ? UserEntity.USER_BELONG_TOP : userlogin.getCompanyId(); AdminSeeWebSetInfoVo adminSeeWebSetInfoVo = new AdminSeeWebSetInfoVo(); - List<WebSetEntity> webSetEntities = this.baseMapper.selectList(new QueryWrapper<>()); + QueryWrapper<WebSetEntity> objectQueryWrapper = new QueryWrapper<>(); + objectQueryWrapper.eq("company_id",companyId); + List<WebSetEntity> webSetEntities = this.baseMapper.selectList(objectQueryWrapper); if(CollUtil.isNotEmpty(webSetEntities)){ WebSetEntity webSetEntity = webSetEntities.get(0); adminSeeWebSetInfoVo.setId(webSetEntity.getId()); @@ -37,16 +43,14 @@ @Override public Result updateWebSet(AdminUpdateWebSetDto adminUpdateWebSetDto) { + UserEntity userlogin = LoginUserUtil.getLoginUser(); + long companyId = userlogin.getCompanyId() == null ? UserEntity.USER_BELONG_TOP : userlogin.getCompanyId(); String webTitle = adminUpdateWebSetDto.getWebTitle(); - if(StrUtil.isEmpty(webTitle)){ - return Result.fail("请输入网页标题"); - } String webKeyword = adminUpdateWebSetDto.getWebKeyword(); - if(StrUtil.isEmpty(webKeyword)){ - return Result.fail("请输入网页关键字"); - } Long id = adminUpdateWebSetDto.getId() == null ? 0L : adminUpdateWebSetDto.getId(); - WebSetEntity webSetEntity = this.baseMapper.selectById(id); + QueryWrapper<WebSetEntity> objectQueryWrapper = new QueryWrapper<>(); + objectQueryWrapper.eq("company_id",companyId); + WebSetEntity webSetEntity = this.baseMapper.selectOne(objectQueryWrapper); if(ObjectUtil.isEmpty(webSetEntity)){ WebSetEntity webSetEntityAdd = new WebSetEntity(); webSetEntityAdd.setWebTitle(webTitle); @@ -59,19 +63,16 @@ if(StrUtil.isNotEmpty(webPic)){ webSetEntityAdd.setWebPic(webPic); } + webSetEntityAdd.setCompanyId(companyId); this.baseMapper.insert(webSetEntityAdd); return Result.ok("保存成功"); } webSetEntity.setWebTitle(webTitle); webSetEntity.setWebKeyword(webKeyword); String webRemark = adminUpdateWebSetDto.getWebRemark(); - if(StrUtil.isNotEmpty(webRemark)){ webSetEntity.setWebRemark(webRemark); - } String webPic = adminUpdateWebSetDto.getWebPic(); - if(StrUtil.isNotEmpty(webPic)){ webSetEntity.setWebPic(webPic); - } this.baseMapper.updateById(webSetEntity); return Result.ok("保存成功"); } -- Gitblit v1.9.1