湖北水库项目
blame | history | patch | commit | commitdiff | ignore whitespace
shenyijian
2024-06-18 53e8b3ad59b48abf57321fd77c1924f4fe837b60 
 skjcmanager/skjcmanager-service/skjcmanager-sm/src/main/java/cn/gistack/sm/sg/service/impl/SgProjectInfoServiceImpl.java


@@ -11,6 +11,7 @@


import cn.gistack.sm.sg.service.SgProjectInfoService;


import com.baomidou.mybatisplus.core.metadata.IPage;


import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;


import com.baomidou.mybatisplus.core.toolkit.StringUtils;


import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;


import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;


import org.springframework.beans.factory.annotation.Autowired;


@@ -118,8 +119,14 @@


   public Map<String, Long> projectCount(ProjectSearchDTO searchDTO) {


      Map<String, Long> map  = new HashMap<>();


      LambdaQueryChainWrapper<SgProjectInfoDO> wrapper = new LambdaQueryChainWrapper<>(sgProjectInfoMapper);


      wrapper.like(SgProjectInfoDO::getProjectName,searchDTO.getProjectName());


      wrapper.eq(SgProjectInfoDO::getStatus,searchDTO.getStatus());


      String projectName = searchDTO.getProjectName();


      Integer status = searchDTO.getStatus();


      if (StringUtils.isNotBlank(projectName)){


         wrapper.like(SgProjectInfoDO::getProjectName,searchDTO.getProjectName());


      }


      if (status!=null){


         wrapper.eq(SgProjectInfoDO::getStatus,searchDTO.getStatus());


      }


      List<SgProjectInfoDO> list = wrapper.list();


      List<SgProjectInfoDO> pending = list.stream().filter(s -> s.getStatus() == 0).collect(Collectors.toList());


      List<SgProjectInfoDO> progress = list.stream().filter(s -> s.getStatus() == 1).collect(Collectors.toList());