From e2f8fb1fbe26ba6d92f3e5dfcaeb0c69abe76b25 Mon Sep 17 00:00:00 2001 From: jiangping <jp@doumee.com> Date: 星期四, 14 十二月 2023 14:28:57 +0800 Subject: [PATCH] 海康接口对接开发 --- server/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java | 66 ++++++++++++++++++++++++++++++--- 1 files changed, 60 insertions(+), 6 deletions(-) diff --git a/server/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java b/server/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java index 3161c88..68aca87 100644 --- a/server/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java +++ b/server/dmvisit_service/src/main/java/com/doumee/service/business/impl/CompanyServiceImpl.java @@ -1,5 +1,7 @@ package com.doumee.service.business.impl; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.doumee.core.annotation.excel.ExcelColumn; import com.doumee.core.constants.ResponseStatus; import com.doumee.core.exception.BusinessException; import com.doumee.core.model.LoginUserInfo; @@ -7,10 +9,12 @@ import com.doumee.core.model.PageWrap; import com.doumee.core.utils.Constants; import com.doumee.core.utils.Utils; +import com.doumee.dao.admin.response.CompanyDTO; import com.doumee.dao.business.CompanyMapper; import com.doumee.dao.business.join.CompanyJoinMapper; import com.doumee.dao.business.model.Category; import com.doumee.dao.business.model.Company; +import com.doumee.dao.business.model.Member; import com.doumee.dao.system.model.SystemUser; import com.doumee.service.business.CompanyService; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; @@ -18,12 +22,18 @@ import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.github.yulichang.wrapper.MPJLambdaWrapper; +import io.swagger.annotations.ApiModelProperty; import org.apache.commons.collections.CollectionUtils; import org.apache.shiro.SecurityUtils; +import org.springframework.beans.BeanUtils; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; + +import java.util.ArrayList; import java.util.Date; import java.util.List; +import java.util.Objects; +import java.util.stream.Collectors; /** * 浼佷笟淇℃伅琛⊿ervice瀹炵幇 @@ -50,7 +60,6 @@ LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal(); checkUnique(company); - Company insert = new Company(); insert.setCreator(loginUserInfo.getId()); insert.setCreateDate(new Date()); @@ -135,6 +144,7 @@ IPage<Company> page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity()); MPJLambdaWrapper<Company> queryWrapper = new MPJLambdaWrapper<>(); Utils.MP.blankToNull(pageWrap.getModel()); + pageWrap.getModel().setIsdeleted(Constants.ZERO); if (pageWrap.getModel().getId() != null) { queryWrapper.eq(Company::getId, pageWrap.getModel().getId()); } @@ -230,10 +240,12 @@ } queryWrapper.orderByDesc(Company::getCreateDate); queryWrapper.leftJoin(Category.class,Category::getId,Company::getCategoryId) - .leftJoin(SystemUser.class,SystemUser::getId,Company::getCreator); + .leftJoin(Member.class, Member::getId,Company::getHeadId); queryWrapper.selectAll(Company.class) .selectAs(Category::getName,Company::getCategoryName) - .selectAs(SystemUser::getRealname,Company::getCreatorName); + .selectAs(Member::getName,Company::getHeadName) + .selectAs(Member::getPhone,Company::getHeadPhone) + .select("count(t1.id) as countNum"); IPage<Company> companyIPage = companyJoinMapper.selectJoinPage(page, Company.class, queryWrapper); return PageData.from(companyIPage); } @@ -243,6 +255,41 @@ QueryWrapper<Company> wrapper = new QueryWrapper<>(company); return companyMapper.selectCount(wrapper); } + + + @Override + public List<CompanyDTO> findCompanyTreePage(Integer type) { + + LambdaQueryWrapper<Company> wrapper = new LambdaQueryWrapper<>(); + + wrapper.eq(Company::getIsdeleted,Constants.ZERO) + .eq(Constants.equalsInteger(type,Constants.ONE),Company::getType,type); + List<Company> companies = companyMapper.selectList(wrapper); + return getCompanyTree( companies, null); + } + + + private List<CompanyDTO> getCompanyTree(List<Company> companies, Integer parentId){ + if (CollectionUtils.isEmpty(companies)){ + return null; + } + List<Company> collect = companies.stream() + .filter(s -> Constants.equalsInteger(s.getParentId(), parentId)) + .collect(Collectors.toList()); + if (CollectionUtils.isEmpty(companies)){ + return null; + } + List<CompanyDTO> companyDTOList = new ArrayList<>(); + for (int i = 0; i < collect.size(); i++) { + Company company = collect.get(i); + CompanyDTO companyDTO = new CompanyDTO(); + BeanUtils.copyProperties(company,companyDTO); + companyDTO.setCompanyDTOList(getCompanyTree(companies, company.getId())); + companyDTOList.add(companyDTO); + } + return companyDTOList; + } + /** * 鍔冲姟鍏徃鍞竴鎬ф牎楠� @@ -255,9 +302,16 @@ .eq(Company::getType,Constants.ZERO) .eq(Company::getName,company.getName()); List<Company> companies = companyMapper.selectList(wrapper); - - if (CollectionUtils.isNotEmpty(companies)){ - throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(),"浼佷笟淇℃伅宸插瓨鍦�"); + if (CollectionUtils.isEmpty(companies)){ + return; + } + //鏂板鐨勬儏鍐靛瓨鍦ㄨ褰� + if (Objects.isNull(company.getId())){ + throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(),"鍏徃鍚嶇О宸插瓨鍦�"); + } + //淇敼鐨勬儏鍐靛瓨鍦ㄨ褰� 鍖归厤涓嶅埌 own + if (!companies.stream().map(s -> s.getId()).anyMatch(s -> s.equals(company.getId()))){ + throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(),"鍏徃鍚嶇О宸插瓨鍦�"); } } } -- Gitblit v1.9.3