package com.gkhy.safePlatform.targetDuty.service.impl; import com.baomidou.mybatisplus.core.metadata.IPage; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.gkhy.safePlatform.targetDuty.repository.TargetDutySummaryRepository; import com.gkhy.safePlatform.targetDuty.entity.TargetDutySummary; import com.gkhy.safePlatform.targetDuty.service.CommonService; import com.gkhy.safePlatform.targetDuty.service.TargetDutySummaryService; import org.springframework.stereotype.Service; import org.springframework.beans.factory.annotation.Autowired; import com.gkhy.safePlatform.commons.enums.ResultCodes; import com.gkhy.safePlatform.commons.query.PageQuery; import com.gkhy.safePlatform.commons.vo.ResultVO; import com.gkhy.safePlatform.commons.vo.SearchResultVO; import com.gkhy.safePlatform.targetDuty.model.dto.req.TargetDutySummaryQueryCriteria; import com.gkhy.safePlatform.targetDuty.model.dto.resp.TargetDutySummaryDto; import com.gkhy.safePlatform.targetDuty.utils.QueryHelpPlus; import com.gkhy.safePlatform.commons.utils.BeanCopyUtils; import javax.annotation.Resource; import java.util.HashSet; import java.util.List; import java.util.Map; import java.util.Set; /** * (TargetDutySummary)表服务实现类 * * @author xurui * @since 2022-07-21 15:35:14 */ @Service("targetDutySummaryService") public class TargetDutySummaryServiceImpl extends ServiceImpl implements TargetDutySummaryService { @Autowired private TargetDutySummaryRepository targetDutySummaryRepository; @Resource private CommonService commonService; @Override public ResultVO queryAll(PageQuery pageQuery) { Long pageIndex = pageQuery.getPageIndex(); Long pageSize = pageQuery.getPageSize(); IPage page = new Page<>(pageIndex, pageSize); page = baseMapper.selectPage(page, QueryHelpPlus.getPredicate(TargetDutySummary.class, pageQuery.getSearchParams())); List respList = BeanCopyUtils.copyBeanList(page.getRecords(), TargetDutySummaryDto.class); // --------------------------- 获取部门信息----------------------- //收集所用到的部门ID Set collectDepIdSet = new HashSet(); respList.forEach(f->{ collectDepIdSet.add(f.getDepartmentId()); }); //获取部门名集合 Map depNameMap = commonService.getDepName(collectDepIdSet); respList.forEach(f->{ f.setDepartmentName(depNameMap.get(f.getDepartmentId())); }); return new SearchResultVO<>( true, pageIndex, pageSize,page.getPages(), page.getTotal(), respList, ResultCodes.OK ); } @Override public List queryAll(TargetDutySummaryQueryCriteria criteria) { return baseMapper.selectList(QueryHelpPlus.getPredicate(TargetDutySummary.class, criteria)); } }