From 3dc1ca272294451ba48277e201a22d596d10e645 Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Tue, 12 Jul 2022 16:23:35 +0800 Subject: [PATCH] fix --- src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CompanyServiceImpl.java | 58 +++++++++++++++++++++++++++++++--------------------------- 1 files changed, 31 insertions(+), 27 deletions(-) diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CompanyServiceImpl.java b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CompanyServiceImpl.java index a9b6e90..9775390 100644 --- a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CompanyServiceImpl.java +++ b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/CompanyServiceImpl.java @@ -11,10 +11,7 @@ import com.xcong.farmer.cms.modules.system.dto.AdminBelongDto; import com.xcong.farmer.cms.modules.system.dto.AdminDeleteDto; import com.xcong.farmer.cms.modules.system.dto.AdminUpdateCompanyDto; -import com.xcong.farmer.cms.modules.system.entity.CompanyEntity; -import com.xcong.farmer.cms.modules.system.entity.MenuEntity; -import com.xcong.farmer.cms.modules.system.entity.RoleEntity; -import com.xcong.farmer.cms.modules.system.entity.UserEntity; +import com.xcong.farmer.cms.modules.system.entity.*; import com.xcong.farmer.cms.modules.system.mapper.*; import com.xcong.farmer.cms.modules.system.service.ICompanyService; import com.xcong.farmer.cms.modules.system.util.LoginUserUtil; @@ -25,6 +22,7 @@ import java.util.ArrayList; import java.util.List; + import cn.hutool.core.collection.CollUtil; import cn.hutool.core.util.ObjectUtil; @@ -46,6 +44,8 @@ private MenuMapper menuMapper; @Resource private UserMapper userMapper; + @Resource + private WebSetMapper webSetMapper; @Override public Result getBelongInPage(AdminBelongDto adminBelongDto) { @@ -54,10 +54,10 @@ Page<AdminCompanyVo> page = new Page<>(adminBelongDto.getPageNum(), adminBelongDto.getPageSize()); CompanyEntity companyEntity = new CompanyEntity(); String name = adminBelongDto.getName(); - if(StrUtil.isNotEmpty(name)){ + if (StrUtil.isNotEmpty(name)) { companyEntity.setName(name); } - if(belongId != UserEntity.USER_BELONG_TOP){ + if (belongId != UserEntity.USER_BELONG_TOP) { companyEntity.setId(belongId); } IPage<AdminCompanyVo> list = this.baseMapper.selectAdminBelongInPage(page, companyEntity); @@ -70,20 +70,24 @@ String name = adminAddCompanyDto.getName(); String code = adminAddCompanyDto.getCode(); QueryWrapper<CompanyEntity> objectQueryWrapper = new QueryWrapper<>(); - objectQueryWrapper.eq("code",code); + objectQueryWrapper.eq("code", code); List<CompanyEntity> belongEntities = this.baseMapper.selectList(objectQueryWrapper); - if(CollUtil.isNotEmpty(belongEntities)){ + if (CollUtil.isNotEmpty(belongEntities)) { return Result.fail("编码已存在,请重新输入"); } CompanyEntity companyEntity = new CompanyEntity(); companyEntity.setName(name); companyEntity.setCode(code); String webAddress = adminAddCompanyDto.getWebAddress(); - companyEntity.setWebAddress(webAddress); + companyEntity.setWebAddress(webAddress); String remark = adminAddCompanyDto.getRemark(); - companyEntity.setRemark(remark); + companyEntity.setRemark(remark); String pic = adminAddCompanyDto.getPic(); - companyEntity.setPic(pic); + companyEntity.setPic(pic); + + WebSettingEntity webSettingEntity = new WebSettingEntity(); + webSettingEntity.setCompanyId(companyEntity.getId()); + this.webSetMapper.insert(webSettingEntity); this.baseMapper.insert(companyEntity); return Result.ok("添加成功"); } @@ -91,7 +95,7 @@ @Override public Result seeBelongInfo(Long id) { CompanyEntity companyEntity = this.baseMapper.selectById(id); - if(ObjectUtil.isEmpty(companyEntity)){ + if (ObjectUtil.isEmpty(companyEntity)) { Result.fail("所属公司不存在"); } AdminSeeCompanyInfoVo adminSeeCompanyInfoVo = new AdminSeeCompanyInfoVo(); @@ -109,17 +113,17 @@ public Result updateBelong(AdminUpdateCompanyDto adminUpdateCompanyDto) { Long id = adminUpdateCompanyDto.getId(); CompanyEntity companyEntity = this.baseMapper.selectById(id); - if(ObjectUtil.isEmpty(companyEntity)){ + if (ObjectUtil.isEmpty(companyEntity)) { Result.fail("所属公司不存在"); } String name = adminUpdateCompanyDto.getName(); companyEntity.setName(name); String code = adminUpdateCompanyDto.getCode(); - if(!companyEntity.getCode().equals(code)){ + if (!companyEntity.getCode().equals(code)) { QueryWrapper<CompanyEntity> objectQueryWrapper = new QueryWrapper<>(); - objectQueryWrapper.eq("code",code); + objectQueryWrapper.eq("code", code); List<CompanyEntity> belongEntities = this.baseMapper.selectList(objectQueryWrapper); - if(CollUtil.isNotEmpty(belongEntities)){ + if (CollUtil.isNotEmpty(belongEntities)) { return Result.fail("编码已存在,请重新输入"); } companyEntity.setCode(code); @@ -136,13 +140,13 @@ UserEntity userlogin = LoginUserUtil.getLoginUser(); Long companyId = userlogin.getCompanyId(); QueryWrapper<CompanyEntity> objectQueryWrapper = new QueryWrapper<>(); - if(companyId != UserEntity.USER_BELONG_TOP){ - objectQueryWrapper.eq("id",companyId); + if (companyId != UserEntity.USER_BELONG_TOP) { + objectQueryWrapper.eq("id", companyId); } List<CompanyEntity> belongEntities = this.baseMapper.selectList(objectQueryWrapper); List<AdminCompanyVo> adminCompanyVos = new ArrayList<>(); - if(CollUtil.isNotEmpty(belongEntities)){ - for(CompanyEntity companyEntity : belongEntities){ + if (CollUtil.isNotEmpty(belongEntities)) { + for (CompanyEntity companyEntity : belongEntities) { AdminCompanyVo adminCompanyVo = new AdminCompanyVo(); adminCompanyVo.setId(companyEntity.getId()); adminCompanyVo.setName(companyEntity.getName()); @@ -158,23 +162,23 @@ public Result delObjs(AdminDeleteDto adminDeleteDto) { String ids = adminDeleteDto.getIds(); UserEntity userlogin = LoginUserUtil.getLoginUser(); - if(userlogin.getCompanyId() != UserEntity.USER_BELONG_TOP){ + if (userlogin.getCompanyId() != UserEntity.USER_BELONG_TOP) { return Result.fail("没有删除权限"); } - if(StrUtil.isNotEmpty(ids)){ + if (StrUtil.isNotEmpty(ids)) { String[] companyIds = ids.split(StringPool.COMMA); - for(String companyIdStr : companyIds){ + for (String companyIdStr : companyIds) { Long companyId = Long.valueOf(companyIdStr); QueryWrapper<UserEntity> userQueryWrapper = new QueryWrapper<>(); - userQueryWrapper.eq("company_id",companyId); + userQueryWrapper.eq("company_id", companyId); List<UserEntity> userEntities = userMapper.selectList(userQueryWrapper); - if(CollUtil.isNotEmpty(userEntities)){ + if (CollUtil.isNotEmpty(userEntities)) { return Result.fail("请先清空属于该公司的用户"); } QueryWrapper<RoleEntity> roleQueryWrapper = new QueryWrapper<>(); - roleQueryWrapper.eq("company_id",companyId); + roleQueryWrapper.eq("company_id", companyId); List<RoleEntity> roleEntities = roleMapper.selectList(roleQueryWrapper); - if(CollUtil.isNotEmpty(roleEntities)){ + if (CollUtil.isNotEmpty(roleEntities)) { return Result.fail("请先清空属于该公司的角色"); } this.baseMapper.deleteById(companyId); -- Gitblit v1.9.1